|
from src.extensions import db |
|
from werkzeug.security import generate_password_hash, check_password_hash |
|
import datetime |
|
|
|
class User(db.Model): |
|
__tablename__ = 'users' |
|
id = db.Column(db.Integer, primary_key=True) |
|
username = db.Column(db.String(80), unique=True, nullable=False) |
|
password_hash = db.Column(db.String(128), nullable=False) |
|
full_name = db.Column(db.String(120)) |
|
role = db.Column(db.String(80)) |
|
created_at = db.Column(db.DateTime, default=datetime.datetime.utcnow) |
|
|
|
|
|
drafts = db.relationship('Draft', backref='author', lazy=True) |
|
amendments = db.relationship('Amendment', backref='author', lazy=True) |
|
analyses = db.relationship('ImpactAnalysis', backref='generator', lazy=True) |
|
|
|
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 __repr__(self): |
|
return f'<User {self.username}>' |
|
|
|
|