from sqlalchemy import Column, Integer, String, Boolean, DateTime, ForeignKey, Text from sqlalchemy.sql import func from app.core.database import Base class Role(Base): __tablename__ = "roles" id = Column(Integer, primary_key=True, autoincrement=True) name = Column(String(50), unique=True, nullable=False) # admin, energy_manager, area_manager, operator, analyst, visitor display_name = Column(String(100), nullable=False) description = Column(Text) permissions = Column(Text) # JSON string of permission list created_at = Column(DateTime(timezone=True), server_default=func.now()) class User(Base): __tablename__ = "users" id = Column(Integer, primary_key=True, autoincrement=True) username = Column(String(50), unique=True, nullable=False, index=True) email = Column(String(100), unique=True) hashed_password = Column(String(200), nullable=False) full_name = Column(String(100)) phone = Column(String(20)) role = Column(String(50), ForeignKey("roles.name"), default="visitor") is_active = Column(Boolean, default=True) last_login = Column(DateTime(timezone=True)) created_at = Column(DateTime(timezone=True), server_default=func.now()) updated_at = Column(DateTime(timezone=True), server_default=func.now(), onupdate=func.now()) class AuditLog(Base): __tablename__ = "audit_logs" id = Column(Integer, primary_key=True, autoincrement=True) user_id = Column(Integer, ForeignKey("users.id")) action = Column(String(50), nullable=False) resource = Column(String(100)) detail = Column(Text) ip_address = Column(String(50)) created_at = Column(DateTime(timezone=True), server_default=func.now())