43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
|
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())
|