class UserRoles(db.Model):
__tablename__ = "user_roles"
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 StartDoc(db.Model):
__tablename__ = "start_docs"
id = db.Column(db.Integer(), primary_key=True)
notes = db.Column(db.String(300), default="")
delivered = db.Column(db.Boolean(), default=False)
sent_to_hr = db.Column(db.Boolean(), default=False)
sent_date = db.Column(db.Date(), index=True)
start_doc_type = db.Column(db.Integer, db.ForeignKey("start_doc_type.id"))
worker_id = db.Column(db.Integer(), db.ForeignKey("worker.id"))
def __repr__(self):
return "{}".format(
StartDocType.query.filter_by(id=self.start_doc_type).first())
class Role(db.Model):
__tablename__ = "role"
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(50), unique=True)
def __repr__(self):
return "{}".format(self.name)
class Event(db.Model):
__tablename__ = "event"
id = db.Column(db.Integer(), primary_key=True)
begin = db.Column(db.DateTime(),
index=True,
nullable=False,
default=datetime.now())
end = db.Column(db.DateTime(), default=datetime.now())
event_kind_id = db.Column(db.Integer(), db.ForeignKey("event_kind.id"))
worker_id = db.Column(db.Integer(), db.ForeignKey("worker.id"))
notes = db.Column(db.String(300), default="")
delivered = db.Column(db.Boolean(), default=False)
sent_to_hr = db.Column(db.Boolean(), default=False)
sent_date = db.Column(db.Date(), index=True)
def __repr__(self):
return "{}\t{}\t{}\t{}".format(
self.event_kind, self.begin, self.end,
User.query.filter_by(id=self.worker_id).first())
class StartDocType(db.Model):
__tabname__ = "start_doc_type"
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(50), index=True, unique=True)
docs = db.relationship("StartDoc", backref="doc_type", cascade="all")
def __repr__(self):
return "{}".format(self.name)
def assign_document(self, document):
self.docs.append(document)
class Workplace(db.Model):
__tablename__ = "workplace"
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(50), index=True, unique=True)
workers = db.relationship("Worker", backref="workplace", cascade="all")
def __repr__(self):
return "{}".format(self.name)
def set_workplace(self, worker):
self.workers.append(worker)
class EventKind(db.Model):
__tablename__ = "event_kind"
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(50), index=True, nullable=False, unique=True)
events = db.relationship("Event", backref="event_kind", cascade="all")
def __repr__(self):
return "{}".format(self.name)
def set_kind(self, event):
self.events.append(event)
class Function(db.Model):
__tablename__ = "function"
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(120), index=True, unique=True)
salary = db.Column(db.Float(), index=True)
workers = db.relationship("Worker", backref="function", cascade="all")
def __repr__(self):
return "{}".format(self.name)
def set_function(self, worker):
self.workers.append(worker)
class User(db.Model, UserMixin):
__tablename__ = "user"
id = db.Column(db.Integer(), primary_key=True)
username = db.Column(db.String(120), index=True, unique=True)
password_hash = db.Column(db.String(128))
roles = db.relationship("Role", secondary="user_roles")
def __repr__(self):
return "{}".format(self.username)
def set_password(self, password):
self.password_hash = generate_password_hash(password)
def check_password(self, password):
return check_password_hash(self.password_hash, password)
def assign_role(self, role):
self.roles.append(role)
class Worker(db.Model, UserMixin):
__tablename__ = "worker"
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(120), index=True, unique=True)
contract_begin = db.Column(db.Date(), index=True)
work_end = db.Column(db.Date(), index=True)
contract_end = db.Column(db.Date(), index=True)
working = db.Column(db.Boolean(), default=True)
events = db.relationship("Event", backref="worker", cascade="all")
function_id = db.Column(db.Integer, db.ForeignKey("function.id"))
workplace_id = db.Column(db.Integer, db.ForeignKey("workplace.id"))
start_docs = db.relationship("StartDoc", backref="worker", cascade="all")
def __repr__(self):
return "{}".format(self.name)
def assign_event(self, event):
self.events.append(event)
def assign_start_doc(self, doc):
self.start_docs.append(doc)