src/models/draft.py

from src.extensions import db
import datetime

class Draft(db.Model):
    __tablename__ = 'drafts'
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String, nullable=False)
    content = db.Column(db.Text)
    policy_intent = db.Column(db.Text)
    status = db.Column(db.String, default='Draft') # e.g., 'Draft', 'Review', 'Approved', 'Rejected'
    author_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    related_legislation_id = db.Column(db.Integer, db.ForeignKey('legislation.id')) # Optional
    created_at = db.Column(db.DateTime, default=datetime.datetime.utcnow)
    last_updated = db.Column(db.DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow)

    def __repr__(self):
        return f'<Draft {self.id}: {self.title[:50]}>'