-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodels.py
75 lines (54 loc) · 2 KB
/
models.py
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
from flask_sqlalchemy import SQLAlchemy
import uuid
#create a new sqlalchemy object
db = SQLAlchemy()
# Base model with common fields
class Base(db.Model):
__abstract__ = True
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
date_created = db.Column(db.DateTime, default=db.func.current_timestamp())
date_modified = db.Column(db.DateTime, default=db.func.current_timestamp(),
onupdate=db.func.current_timestamp())
class Topics(Base, db.Model):
title = db.Column(db.String(500))
status = db.Column(db.Boolean, default=1)
def __repr__(self):
return self.title
def to_json(self):
return {
'title': self.title,
'options': [
{ 'name': option.option.name,
'vote_count': option.vote_count
}
for option in self.options.all()
],
'status': self.status,
'total_vote': self.total_vote()
}
def total_vote(self,total=0):
for option in self.options.all():
total+=option.vote_count
return total
class Options(Base, db.Model):
name = db.Column(db.String(200),unique=True)
def to_json(self):
return {
'id': uuid.uuid4(),
'name': self.name
}
class Polls(Base, db.Model):
topic_id = db.Column(db.Integer,db.ForeignKey('topics.id'))
option_id = db.Column(db.Integer,db.ForeignKey('options.id'))
vote_count = db.Column(db.Integer,default=0)
# To define relationship b/w diff models
topic = db.relationship('Topics', foreign_keys=[topic_id], backref=db.backref('options', lazy='dynamic'))
option = db.relationship('Options',foreign_keys=[option_id])
def __repr__(self):
return self.option.name
class Users(Base):
email = db.Column(db.String(100),unique=True)
username = db.Column(db.String(50),unique=True)
password = db.Column(db.String(200))
def __repr__(self):
return self.username