class ArgumentRole(db.Model):
__tablename__ = 'argument_roles'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
role = db.Column(db.String(32))
version = db.Column(db.Integer, db.ForeignKey(Version.id))
def __repr__(self):
return "<Argument Role(role='%s', start='%s', end='%s', text='%s')>" % self.role
class Document(db.Model):
__tablename__ = 'documents'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
document_id = db.Column(db.String(64))
text = db.Column(db.Text)
def __repr__(self):
return "<Document(id='%s', text='%s')>" % self.doc_id, self.text
class EventCategory(db.Model):
__tablename__ = 'event_category'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
category = db.Column(db.String(32))
version = db.Column(db.Integer, db.ForeignKey(Version.id))
def __repr__(self):
return "<Event Category(category='%s')>" % self.category
class ArgumentEntity(db.Model):
__tablename__ = 'arguments_entity'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
role = db.Column(db.Integer, db.ForeignKey(ArgumentRole.id), nullable=False)
event = db.Column(db.Integer, db.ForeignKey(Event.id), nullable=False)
argument = db.Column(db.Integer, db.ForeignKey(Entity.id), nullable=False)
version = db.Column(db.Integer, db.ForeignKey(Version.id))
def __repr__(self):
return "<Entity(category='%s', start='%s', end='%s', text='%s')>" % (
self.category, self.start, self.end, self.text)
class EntityProperty(db.Model):
"""key-value property pairs for entity or event
"""
__tablename__ = 'entity_properties'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
label = db.Column(db.String(64))
value = db.Column(db.String(64))
entity = db.Column(db.ForeignKey(Entity.id), nullable=False)
def __repr__(self):
return "<Entity Property(key='%s', value='%s')>" % self.key, self.value
class User(db.Model):
__tablename__ = 'users'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(32), unique=True)
password = db.Column(db.String(32))
def __init__(self, username, password):
self.username = username
self.password = password
def __repr__(self):
return "<User(name='%s')>" % self.username
class Event(db.Model):
__tablename__ = 'events'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
document = db.Column(db.Integer, db.ForeignKey(Document.id))
category = db.Column(db.Integer, db.ForeignKey(EventCategory.id))
arguments_entity = db.relationship('ArgumentEntity')
arguments_event = db.relationship('ArgumentEvent',
foreign_keys="ArgumentEvent.event")
properties = db.relationship('EventProperty')
version = db.Column(db.Integer, db.ForeignKey(Version.id))
def __repr__(self):
return "<Entity(category='%s', start='%s', end='%s', text='%s')>" % (
self.category, self.start, self.end, self.text)
class Version(db.Model):
__tablename__ = 'versions'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
# TODO: how to get the user from a version object, currently it is just user id not user object
user = db.Column(db.Integer, db.ForeignKey(User.id))
datetime = db.Column(db.DateTime, default=datetime.datetime.utcnow())
version = db.Column(db.String(64))
entities = db.relationship('Entity')
events = db.relationship('Event')
entity_categories = db.relationship('EntityCategory')
event_categories = db.relationship('EventCategory')
def __init__(self, user, version):
self.user = user
self.version = version
def __repr__(self):
return "<Version(user='******', version='%s')>" % self.user.username, self.version
class Entity(db.Model):
__tablename__ = 'entities'
__table_args__ = {'mysql_engine': 'InnoDB'}
id = db.Column(db.Integer, primary_key=True)
document = db.Column(db.Integer, db.ForeignKey(Document.id))
category = db.Column(db.Integer, db.ForeignKey(EntityCategory.id))
start = db.Column(db.Integer)
end = db.Column(db.Integer)
text = db.Column(db.Text)
properties = db.relationship('EntityProperty')
version = db.Column(db.Integer, db.ForeignKey(Version.id))
def __repr__(self):
return "<Entity(category='%s', start='%s', end='%s', text='%s')>" % (
self.category, self.start, self.end, self.text)
