class Event(DB.Model):
'''Event Table/Model for database'''
event_id = DB.Column(DB.Integer, primary_key=True)
host = DB.Column(DB.String(80), nullable=False)
event_name = DB.Column(DB.String(300), nullable=False)
event_description = DB.Column(DB.String(300), nullable=False)
restaurant = DB.Column(DB.String(150), nullable=False)
location = DB.Column(DB.String(80), nullable=False)
event_date = DB.Column(DB.String(80), nullable=False)
event_time = DB.Column(DB.String(80), nullable=False)
attendees = DB.Column(DB.ARRAY(DB.String(50)), nullable=False)
def __repr__(self):
return '<Event %r>' % self.event_name
class Dataset(DB.Model):# pylint: disable=R0903
"""
Subsets model for SQL database
"""
__tablename__ = 'dataset'
id = DB.Column(DB.Integer, primary_key=True, autoincrement=True)
included_rows = DB.Column(DB.ARRAY(DB.String), nullable=True)
date = DB.Column(DB.DateTime, nullable=False, default=datetime.utcnow)
file_id = DB.Column(DB.Integer, DB.ForeignKey('file.id', ondelete="CASCADE"))
user_id = DB.Column(DB.Integer, DB.ForeignKey('users.id'))
filter_id = DB.Column(DB.Integer, DB.ForeignKey('filter.id'), nullable=True)
def __init__(self, file_id, user_id, included_rows=None, filter_id=None):
self.date = datetime.utcnow()
self.included_rows = included_rows
self.file_id = file_id
self.filter_id = filter_id
self.user_id = user_id
