-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathdb.py
More file actions
75 lines (60 loc) · 2.19 KB
/
db.py
File metadata and controls
75 lines (60 loc) · 2.19 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
import sqlalchemy.orm as orm
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy import Column, Integer, String, DateTime, Boolean, ForeignKey, Date
from sqlalchemy.orm import relationship
import bcrypt
from datetime import datetime
Base = declarative_base()
Session = orm.scoped_session(orm.sessionmaker())
class User(Base):
__tablename__ = 'users'
id = Column(Integer, primary_key=True)
name = Column(String)
email = Column(String, unique=True)
pw_hash = Column(String)
salt = Column(String)
signedin = Column(Boolean)
scheduleverified=Column(Boolean)
schedule=Column(String)
finalschedule=Column(String)
code=Column(Integer)
emailVerified=Column(Boolean)
registerTime=Column(DateTime)
signins = relationship("Signin", back_populates="user")
events = relationship("Event", back_populates="user")
def as_dict(self):
return {c.name: getattr(self, c.name) for c in self.__table__.columns}
def check_password(self, password):
return self.pw_hash == bcrypt.hashpw(password.encode('utf-8'), self.salt)
def is_admin(self):
if self.email == 'daynb@guilford.edu':
return True
elif self.email == 'kiddlm@guilford.edu':
return True
elif self.email=='kerrj@guilford.edu':
return True
elif self.email=='justin.g.kerr@gmail.com':
return True
else:
return False
def __repr__(self):
return "<User(name='%s')>" % (self.name)
class Signin(Base):
__tablename__ = 'signins'
id = Column(Integer, primary_key=True)
date_in = Column(DateTime)
user_id = Column(Integer, ForeignKey('users.id'))
user = relationship("User", back_populates="signins")
def __repr__(self):
return "<Signin(id='%s')>" % (self.id)
class Event(Base):
__tablename__ = 'events'
id = Column(Integer, primary_key=True)
from_date = Column(Date)
to_date = Column(Date)
title = Column(String)
description = Column(String)
user_id = Column(Integer, ForeignKey('users.id'))
user = relationship("User", back_populates="events")
def __repr__(self):
return "<Event(id='%s')>" % (self.id)