models.py
4.46 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
# coding= utf8
from web import *
from datetime import datetime
user_task_table = db.Table('user_task',
db.Column('user_id', db.Integer, db.ForeignKey('user.id')),
db.Column('task_id', db.Integer, db.ForeignKey('task.id'))
)
class User(db.Model):
"""An admin user capable of viewing reports.
"""
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(80), unique=True, nullable=False)
password = db.Column(db.String(256), nullable=False)
authenticated = db.Column(db.Boolean, default=False)
tasks = db.relationship('Task', secondary=user_task_table,
backref=db.backref('users', lazy='dynamic'))
def __init__(self, username, password):
self.username = username
self.password = bcrypt.generate_password_hash(password)
def is_active(self):
"""True, as all users are active."""
return True
def get_id(self):
"""Return the email address to satisfy Flask-Login's requirements."""
return self.id
def is_authenticated(self):
"""Return True if the user is authenticated."""
return self.authenticated
def is_anonymous(self):
"""False, as anonymous users aren't supported."""
return False
def __repr__(self):
return '<models.User %r>' % self.username
class Task(db.Model):
enum_priority = ['COMMOM', 'IMPORTANT', 'CRITICAL', 'SERIOUS']
enum_status = ['OPEN', 'CLOSED', 'SECRET']
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(80), unique=True, nullable=False)
pubdate = db.Column(db.DateTime, nullable=True)
duedate = db.Column(db.DateTime, nullable=False)
attachfile = db.Column(db.String(80), nullable=True)
description = db.Column(db.String(512), nullable=True)
priority = db.Column(db.Integer, nullable=False)
status = db.Column(db.Integer, nullable=False)
def __init__(self, name, duedate, priority, status, pubdate=None, attachfile=None,
description=None):
self.name = name
self.pubdate = pubdate
self.duedate = duedate
self.priority = priority
self.attachfile = attachfile
self.description = description
self.status = status
def __repr__(self):
return '<models.Task %r>' % self.name
def db_test():
user1 = User('admin', 'admin')
user2 = User('nonadmin', 'nonadmin')
user3 = User('Amy', 'amy')
user4 = User('Bob', 'bob')
user5 = User('Cindy', 'cindy')
user6 = User('David', 'david')
db.session.add(user1)
db.session.add(user2)
db.session.add(user3)
db.session.add(user4)
db.session.add(user5)
db.session.add(user6)
task1 = Task("Setting the Database", datetime.strptime('12/07/2013', '%m/%d/%Y'), 0, 1,
datetime.today())
task2 = Task("another task", datetime.strptime('12/07/2013', '%m/%d/%Y'), 2, 0,
datetime.today())
task3 = Task("Emergency!!", datetime.strptime('12/6/2016', '%m/%d/%Y'), 1, 0,
datetime.today())
task4 = Task("hhhhhhhhh", datetime.strptime('4/07/2015', '%m/%d/%Y'), 3, 0,
datetime.today())
task5 = Task("OMG-AKA", datetime.strptime('3/12/2016', '%m/%d/%Y'), 0, 1,
datetime.today())
task6 = Task("赛高！", datetime.strptime('11/11/2015', '%m/%d/%Y'), 2, 0,
datetime.today())
task7 = Task("asdfghj", datetime.strptime('12/07/2015', '%m/%d/%Y'), 2, 1,
datetime.today())
task8 = Task("sdfadfadsfasd", datetime.strptime('12/07/2015', '%m/%d/%Y'), 2, 0,
datetime.today())
task9 = Task("a secret task example", datetime.strptime('12/07/2015', '%m/%d/%Y'), 0, -1,
datetime.today(), description='hahahahahaha\nhahahah')
db.session.add(task1)
db.session.add(task2)
db.session.add(task3)
db.session.add(task4)
db.session.add(task5)
db.session.add(task6)
db.session.add(task7)
db.session.add(task8)
db.session.add(task9)
task1.users = [user1, user5]
task3.users = [user3, user5]
task4.users = [user2, user3, user4]
task5.users = [user5, user6]
task8.users = [user1]
print task8,user2
task8.users.append(user2)
db.session.commit()
if __name__ == '__main__':
db.drop_all()
db.create_all()
db_test()
list_task = Task.query.filter_by(status=0).all()
print list_task