72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
from .extensions import db
|
|
from datetime import datetime, date
|
|
|
|
|
|
class Project(db.Model):
|
|
__tablename__ = 'projects'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
name = db.Column(db.String(200), nullable=False)
|
|
color = db.Column(db.String(7), nullable=False, default='#C9A84C')
|
|
description = db.Column(db.Text)
|
|
drive_url = db.Column(db.String(500))
|
|
archived_at = db.Column(db.DateTime, nullable=True)
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
deliverables = db.relationship(
|
|
'Deliverable',
|
|
backref='project',
|
|
cascade='all, delete-orphan',
|
|
lazy=True,
|
|
)
|
|
|
|
def to_dict(self, include_deliverables=True):
|
|
data = {
|
|
'id': self.id,
|
|
'name': self.name,
|
|
'color': self.color,
|
|
'description': self.description,
|
|
'drive_url': self.drive_url,
|
|
'archived_at': self.archived_at.isoformat() if self.archived_at else None,
|
|
'created_at': self.created_at.isoformat() if self.created_at else None,
|
|
}
|
|
if include_deliverables:
|
|
data['deliverables'] = [
|
|
d.to_dict() for d in sorted(self.deliverables, key=lambda x: x.due_date)
|
|
]
|
|
return data
|
|
|
|
|
|
class Deliverable(db.Model):
|
|
__tablename__ = 'deliverables'
|
|
|
|
id = db.Column(db.Integer, primary_key=True)
|
|
project_id = db.Column(
|
|
db.Integer,
|
|
db.ForeignKey('projects.id', ondelete='CASCADE'),
|
|
nullable=False,
|
|
)
|
|
title = db.Column(db.String(300), nullable=False)
|
|
due_date = db.Column(db.Date, nullable=False)
|
|
status = db.Column(db.String(20), nullable=False, default='upcoming')
|
|
created_at = db.Column(db.DateTime, default=datetime.utcnow)
|
|
|
|
def effective_status(self):
|
|
"""
|
|
Returns 'overdue' if the due date has passed and the deliverable has not been
|
|
marked completed. Completed deliverables are never auto-downgraded.
|
|
"""
|
|
if self.status != 'completed' and self.due_date < date.today():
|
|
return 'overdue'
|
|
return self.status
|
|
|
|
def to_dict(self):
|
|
return {
|
|
'id': self.id,
|
|
'project_id': self.project_id,
|
|
'title': self.title,
|
|
'due_date': self.due_date.isoformat() if self.due_date else None,
|
|
'status': self.effective_status(),
|
|
'created_at': self.created_at.isoformat() if self.created_at else None,
|
|
}
|