class Hour(db.Model):
id = db.Column(db.Integer, primary_key=True)
amount = db.Column(db.Integer, nullable=False)
work_date = db.Column(db.String(50), default=func.current_date())
project_shortcut = db.Column(db.String(50),
db.ForeignKey("project.shortcut"),
nullable=False)
user_id = db.Column(db.Integer,
db.ForeignKey("user.id", ondelete="CASCADE"),
nullable=False)
def __repr__(self):
return f"{self.amount} hours"
class Vacation(db.Model):
id = db.Column(db.Integer, primary_key=True)
vacation_date = db.Column(db.String(50), default=func.current_date())
user_id = db.Column(
db.Integer, db.ForeignKey("user.id", ondelete="CASCADE"), nullable=False
)
def __repr__(self):
return f"Vacation day in {self.vacation_date}"
class Todo(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(256), nullable=False)
description = db.Column(db.Text, nullable=False)
date_created = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
date_completed = db.Column(db.DateTime, nullable=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
user = db.relationship('User', backref=db.backref('todos', lazy=True))
def __repr__(self):
return f"Description: '{self.description}', Date: '{self.date_created}'"
class Timesheet(db.Model):
id = db.Column(db.Integer, primary_key=True)
date = db.Column(db.DateTime, nullable=False, default=datetime.now)
workhours = db.Column(db.Numeric(5, 1), nullable=False)
description = db.Column(db.Text, nullable=True)
date_created = db.Column(db.DateTime, nullable=False, default=datetime.now)
date_updated = db.Column(db.DateTime, nullable=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
user = db.relationship('User', backref=db.backref('timesheets', lazy=True))
subject_id = db.Column(db.Integer,
db.ForeignKey('subject.id'),
nullable=False)
subject = db.relationship('Subject',
backref=db.backref('timesheets', lazy=True))
task_id = db.Column(db.Integer, db.ForeignKey('task.id'), nullable=True)
task = db.relationship('Task', backref=db.backref('timesheets', lazy=True))
def __repr__(self):
if self.task:
return f"Timesheet('{self.subject.name}', '{self.task.name}', '{self.date_created}', '{self.hours}')"
else:
return f"Timesheet('{self.subject.name}', '{self.date_created}', '{self.hours}')"
class Subject(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(126), nullable=False)
description = db.Column(db.Text, nullable=False)
date_created = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
active = db.Column(db.Boolean, nullable=False, default=True)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'), nullable=False)
user = db.relationship('User', backref=db.backref('subjects', lazy=True))
def __repr__(self):
return f"Name: '{self.name}', Date: '{self.date_created}'"