""" Modèle User - GMAO Collège Gestion des utilisateurs et authentification Compatible avec la base de données existante """ from flask_login import UserMixin from werkzeug.security import generate_password_hash, check_password_hash from datetime import datetime, timezone from ...extensions import db class User(UserMixin, db.Model): """Utilisateur de l'application GMAO.""" __tablename__ = "users" id = db.Column(db.Integer, primary_key=True) username = db.Column(db.String(80), unique=True, nullable=False) email = db.Column(db.String(120), unique=True, nullable=True) password_hash = db.Column(db.String(256), nullable=False) full_name = db.Column(db.String(150), nullable=False, default="") role = db.Column(db.String(20), default="user") # admin, user, viewer is_active = db.Column(db.Boolean, default=True) created_at = db.Column(db.DateTime, default=lambda: datetime.now(timezone.utc)) last_login = db.Column(db.DateTime, nullable=True) def set_password(self, password): """Hash et stocke le mot de passe.""" self.password_hash = generate_password_hash(password) def check_password(self, password): """Vérifie le mot de passe.""" return check_password_hash(self.password_hash, password) def is_admin(self): """Vérifie si l'utilisateur est admin.""" return self.role == "admin" def __repr__(self): return f"" class Staff(db.Model): """Personnel du collège.""" __tablename__ = "staff" id = db.Column(db.Integer, primary_key=True) first_name = db.Column(db.String(100), nullable=False) last_name = db.Column(db.String(100), nullable=False) function = db.Column(db.String(200), nullable=True) # Fonction/Poste department = db.Column(db.String(200), nullable=True) email = db.Column(db.String(200), nullable=True) phone = db.Column(db.String(20), nullable=True) is_active = db.Column(db.Boolean, default=True) created_at = db.Column(db.DateTime, default=lambda: datetime.now(timezone.utc)) @property def full_name(self): """Retourne le nom complet.""" return f"{self.first_name} {self.last_name}" def __repr__(self): return f""