class UserBookings(db.Model):
id = db.Column(db.Integer(), primary_key=True)
user_id = db.Column(db.Integer(),
db.ForeignKey('user.id', ondelete='CASCADE'))
event_id = db.Column(db.Integer(), nullable=False)
ticket_id = db.Column(db.Integer(), nullable=False)
ticket_type = db.Column(db.String(100), nullable=False)
number_booked = db.Column(db.Integer(), nullable=False)
number_scanned = db.Column(db.Integer(), nullable=False)
payment_status = db.Column(db.String(100), nullable=False)
image_file = db.Column(db.String(50),
nullable=False,
default='default_qr.jpeg')
class Ticket(db.Model):
id = db.Column(db.Integer, primary_key=True)
event_id = db.Column(db.Integer,
db.ForeignKey('event.id', ondelete='CASCADE'),
nullable=False)
ticket_type = db.Column(db.String(30), nullable=False)
# number of tickets per type
num_tickets = db.Column(db.Integer, nullable=False)
num_bought = db.Column(db.Integer, nullable=False, server_default="0")
price = db.Column(db.Float, nullable=False)
class UserRoles(db.Model):
id = db.Column(db.Integer(), primary_key=True)
user_id = db.Column(db.Integer(),
db.ForeignKey('user.id', ondelete='CASCADE'))
role_id = db.Column(db.Integer(),
db.ForeignKey('role.id', ondelete='CASCADE'))
class Role(db.Model):
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(50), unique=False)
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
# username and email are the only ones decided by the user upon registration
username = db.Column(db.String(20), unique=True, nullable=False)
email = db.Column(db.String(120), unique=True, nullable=False)
# image file is automatically done by default
image_file = db.Column(db.String(20),
nullable=False,
default='default.jpg')
password = db.Column(db.String(60), nullable=False)
first_name = db.Column(db.String(100), nullable=False, server_default='')
last_name = db.Column(db.String(100), nullable=False, server_default='')
# backref is like adding another column to an event
# what it does is when we have an event we can use the manager attribute to see who made the event
events = db.relationship('Event', backref='manager', lazy=True)
bookings = db.relationship('UserBookings', backref='customer', lazy=True)
roles = db.relationship('Role',
secondary='user_roles',
backref=db.backref('users', lazy='dynamic'))
def get_reset_token(self, expires_sec=1800):
s = Serializer(current_app.config['SECRET_KEY'], expires_sec)
return s.dumps({'user_id': self.id}).decode('utf-8')
def generate_auth_token(self, expires_sec=600):
s = Serializer(current_app.config['SECRET_KEY'], expires_sec)
return s.dumps({'user_id': self.id}).decode('utf-8')
@staticmethod
def verify_reset_token(token):
s = Serializer(current_app.config['SECRET_KEY'])
try:
user_id = s.loads(token)['user_id']
except:
return None
return User.query.get(user_id)
@staticmethod
def verify_auth_token(token):
s = Serializer(current_app.config['SECRET_KEY'])
try:
user_id = s.loads(token)['user_id']
# data = s.loads(token)
except SignatureExpired:
return None # valid token, but expired
except BadSignature:
return None # invalid token
# user = User.query.get(data['user_id'])
user = User.query.get(user_id)
return user
def confirm(self, token):
s = Serializer(current_app.config['SECRET_KEY'])
try:
data = s.loads(token.encode('utf-8'))
except:
return False
if data.get('confirm') != self.id:
return False
self.confirmed = True
db.session.add(self)
return True
def __repr__(self): # how our object is printed
return f"User('{self.username}','{self.email}','{self.image_file}')"
class EventStaff(db.Model):
id = db.Column(db.Integer(), primary_key=True)
user_id = db.Column(db.Integer(),
db.ForeignKey('user.id', ondelete='CASCADE'))
event_id = db.Column(db.Integer(),
db.ForeignKey('event.id', ondelete='CASCADE'))
class Event(db.Model):
id = db.Column(db.Integer, primary_key=True)
# controllable items start here
title = db.Column(db.String(100), nullable=False)
event_type = db.Column(db.Text, nullable=False)
# date and time will be the event date and time
event_date = db.Column(db.Date, nullable=False)
time_from = db.Column(db.Time, nullable=False)
time_to = db.Column(db.Time, nullable=False)
# num_tickets = db.Column(db.Integer, nullable=False)
content = db.Column(db.Text, nullable=False)
content_eng = db.Column(db.Text, nullable=True)
address = db.Column(db.String(100), nullable=False)
city = db.Column(db.String(100), nullable=False)
location = db.Column(db.String(100), nullable=False)
tickets = db.relationship('Ticket',
backref='event',
lazy=True,
cascade='all,delete-orphan')
image_file = db.Column(db.String(20),
nullable=False,
default='default_event.jpg')
# automatically set items start here
# date_posted will be for when the event is created
date_posted = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
# lower case, table and column name is referenced
# for backref/relationship purposes
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
staffers = db.relationship('User',
secondary='event_staff',
backref=db.backref('users', lazy='dynamic'))
# automatically set items stop here
def __repr__(self): # how our object is printed
return f"Event('{self.title}','{self.date_posted}','{self.content}')"