From 5299fdd3095a0738ca40b2cdbb136a1ea4f47607 Mon Sep 17 00:00:00 2001 From: benaryorg Date: Thu, 22 May 2014 16:41:37 +0200 Subject: [PATCH] Changed User class, added Diary and Entry class --- models.py | 26 +++++++++++++++++++++++++- 1 file changed, 25 insertions(+), 1 deletion(-) diff --git a/models.py b/models.py index 7ccc51c..287e242 100755 --- a/models.py +++ b/models.py @@ -1,6 +1,8 @@ #!/usr/bin/env python2.7 -from sqlalchemy import Column,Integer,String +import datetime + +from sqlalchemy import Column,Integer,String,Date from database import Base class User(Base): @@ -12,3 +14,25 @@ class User(Base): def __init__(self,username,password): self.username=username self.password=password + +class Diary(Base): + __tablename__='diaries' + id=Column(Integer(),primary_key=True,unique=True) + owner=Column(Integer()) + name=Column(String(256),unique=True) + + def __init__(self,owner,name): + self.owner=owner + self.name=name + +class Entry(Base): + __tablename__='entries' + id=Column(Integer(),primary_key=True,unique=True) + diary=Column(Integer()) + date=Column(Date()) + text=Column(String(8192)) + + def __init__(self,diary,text): + self.diary=diary + self.text=text + self.date=datetime.datetime.now().strftime("%Y-%m-%d")