-
Notifications
You must be signed in to change notification settings - Fork 0
/
models.py
136 lines (118 loc) · 6.73 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
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
from flask_sqlalchemy import SQLAlchemy
from datetime import datetime, timedelta
from random import randrange
db = SQLAlchemy()
class User(db.Model):
id = db.Column(db.Integer, primary_key=True, nullable=False)
username = db.Column(db.String(80), unique=True, nullable=False)
password = db.Column(db.String(64), unique=False, nullable=True)
email = db.Column(db.String(120), unique=False, nullable=True)
currentroom = db.Column(db.Integer, db.ForeignKey('room.id'), nullable=True)
def __init__(self, username, password, email, currentroom=None):
self.username = username
self.password = password
self.email = email
self.currentroom = 0
def __repr__(self):
return "<User {} {}>".format(repr(self.id), repr(self.username))
def Everything():
txt = "\t" + str(User.__table__) + "\n"
cols = User.__table__.columns.keys()
txt += (str(cols) + "\n")
resultSet = User.query.order_by(User.id.asc()).all()
for item in resultSet:
txt += ' '.join([str(getattr(item, col)) for col in cols]) + "\n"
return txt
class Room(db.Model):
id = db.Column(db.Integer, primary_key=True, nullable=False)
roomname = db.Column(db.String(80), unique=True, nullable=False)
creator = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
created = db.Column(db.DateTime, nullable=False, default=datetime.utcnow())
lastmessage = db.Column(db.DateTime, nullable=False)
def __init__(self, roomname, creator=None, created=None, lastmessage=None):
self.roomname = roomname
self.creator = 1 if creator == None else creator
self.created = datetime.utcnow() if created == None else created
self.lastmessage = datetime.utcnow() if lastmessage == None else created
def __repr__(self):
return "<Room {} {}>".format(repr(self.id), repr(self.roomname))
def Everything():
txt = "\t" + str(Room.__table__) + "\n"
cols = Room.__table__.columns.keys()
txt += (str(cols) + "\n")
resultSet = Room.query.order_by(Room.id.asc()).all()
for item in resultSet:
txt += ' '.join([str(getattr(item, col)) for col in cols]) + "\n"
return txt
class Chat(db.Model):
id = db.Column(db.Integer, primary_key=True, nullable=False)
room = db.Column(db.Integer, db.ForeignKey('room.id'), nullable=False)
creator = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
created = db.Column(db.DateTime, nullable=False, default=datetime.utcnow())
message = db.Column(db.String(200), unique=False, nullable=False)
def __init__(self, room, creator=None, created=None, message=None):
self.room = room
self.creator = 0 if creator == None else creator
self.created = datetime.utcnow() if created == None else created
self.message = "😶" if message == None else message
def __repr__(self):
return "<Chat {} {} {}>".format(repr(self.id), repr(self.room), repr(self.message))
def as_json(room=None):
json = dict()
cols = Chat.__table__.columns.keys()
if room == None:
resultSet = Chat.query.order_by(Chat.id.asc()).all()
else:
resultSet = Chat.query.filter(Chat.room == int(room)).order_by(Chat.id.asc()).all()
for item in resultSet:
json[item.id] = {col : getattr(item, col) for col in cols}
return json
def as_jsonUpdates(room, count):
json = dict()
cols = Chat.__table__.columns.keys()
if room == None or count == None:
resultSet = Chat.query.order_by(Chat.id.asc()).all()
else:
resultSet = Chat.query.filter(Chat.room == int(room)).order_by(Chat.id.desc()).limit(count).all()[::-1]
for item in resultSet:
json[item.id] = {col : getattr(item, col) for col in cols}
json[item.id]['name'] = User.query.filter(User.id == item.creator).first().username;
return json
def Everything():
txt = "\t" + str(Chat.__table__) + "\n"
cols = Chat.__table__.columns.keys()
txt += (str(cols) + "\n")
resultSet = Chat.query.order_by(Chat.id.asc()).all()
for item in resultSet:
txt += ' '.join([str(getattr(item, col)) for col in cols]) + "\n"
return txt
def populateDB():
db.session.add(User(username="owner", password="pass", email="N@A", currentroom=None))
db.session.add(User(username="FirstUser", password="pass", email="N@A", currentroom=None))
db.session.add(User(username="🦖", password="pass", email="N@A", currentroom=1))
db.session.add(User(username="User1", password="pass", email="N@A", currentroom=1))
db.session.add(User(username="User2", password="pass", email="N@A", currentroom=1))
db.session.add(User(username="User3", password="pass", email="N@A", currentroom=2))
db.session.add(User(username="User4", password="pass", email="N@A", currentroom=3))
db.session.add(User(username="User5", password="pass", email="N@A", currentroom=4))
db.session.add(Room(roomname="Welcome Room", creator=None, created=datetime.utcnow(), lastmessage=datetime.utcnow()))
db.session.add(Room(roomname="Room 2", creator=1, created=datetime(2017, 10, 31, 20, 0), lastmessage=None))
db.session.add(Room(roomname="Room 3", creator=1, created=datetime(2017, 10, 31, 20, 0), lastmessage=None))
db.session.add(Room(roomname="Room 4", creator=2, created=datetime(2017, 10, 31, 20, 0), lastmessage=None))
db.session.add(Room(roomname="Room 5", creator=2, created=datetime(2017, 10, 31, 20, 0), lastmessage=None))
db.session.add(Room(roomname="Room 6", creator=3, created=datetime(2017, 10, 31, 20, 0), lastmessage=None))
db.session.add(Chat(room=1, creator=1, created=None, message="Test Message"))
db.session.add(Chat(room=1, creator=2, created=None, message="First Message"))
db.session.add(Chat(room=1, creator=3, created=None, message="rar"))
db.session.add(Chat(room=1, creator=3, created=None, message="rar"))
db.session.add(Chat(room=2, creator=3, created=None, message="rar"))
db.session.add(Chat(room=3, creator=3, created=None, message="rar"))
db.session.add(Chat(room=4, creator=3, created=None, message="rar"))
db.session.add(Chat(room=2, creator=3, created=None, message="rar"))
db.session.add(Chat(room=3, creator=3, created=None, message="rar"))
db.session.add(Chat(room=1, creator=3, created=None, message="rar"))
db.session.add(Chat(room=1, creator=3, created=None, message="rar"))
db.session.commit()
print('DB Populated...')
return True
print('Model loaded...')