82 lines
3.1 KiB
Python
82 lines
3.1 KiB
Python
from sqlalchemy import Column, Integer, String, DateTime, ForeignKey, func, Table
|
||
from sqlalchemy.orm import relationship
|
||
from sqlalchemy.ext.declarative import declarative_base
|
||
from datetime import datetime
|
||
|
||
Base = declarative_base()
|
||
|
||
|
||
class Role(Base):
|
||
"""角色数据库模型"""
|
||
|
||
__tablename__ = "role"
|
||
id = Column(Integer, primary_key=True, index=True, comment="角色ID")
|
||
name = Column(String(50), nullable=False, unique=True, comment="角色名称")
|
||
createtime = Column(DateTime, default=datetime.utcnow, comment="创建时间")
|
||
updatetime = Column(
|
||
DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, comment="更新时间"
|
||
)
|
||
creator = Column(String(50), nullable=True, comment="创建人")
|
||
|
||
# 关联到User
|
||
users = relationship("User", back_populates="role_rel")
|
||
# 关联到Permission（多对多）
|
||
permissions = relationship(
|
||
"Permission",
|
||
secondary="role_permission",
|
||
back_populates="roles"
|
||
)
|
||
|
||
|
||
class User(Base):
|
||
"""用户数据库模型"""
|
||
|
||
__tablename__ = "users"
|
||
id = Column(Integer, primary_key=True, index=True)
|
||
username = Column(String(50), unique=True, nullable=False, comment="登录名")
|
||
nickname = Column(String(50), nullable=False, comment="昵称")
|
||
email = Column(String(50), nullable=True, comment="电子邮箱")
|
||
phone = Column(String(11), nullable=True, comment="手机号码")
|
||
wx_openid = Column(String(100), nullable=True, comment="微信OpenID")
|
||
password_hash = Column(String(128), nullable=False, comment="密码哈希值")
|
||
avatar = Column(String(255), nullable=True, comment="头像路径")
|
||
role_id = Column(Integer, ForeignKey("role.id"), default=2, comment="角色ID")
|
||
createtime = Column(DateTime, default=datetime.utcnow, comment="创建时间")
|
||
updatetime = Column(
|
||
DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, comment="更新时间"
|
||
)
|
||
lastlogintime = Column(DateTime, comment="最后登录时间")
|
||
|
||
# 关联到Role
|
||
role_rel = relationship("Role", back_populates="users")
|
||
|
||
|
||
class Permission(Base):
|
||
"""权限数据库模型"""
|
||
|
||
__tablename__ = "permission"
|
||
id = Column(Integer, primary_key=True, index=True, comment="权限ID")
|
||
name = Column(String(50), nullable=False, comment="权限名称")
|
||
code = Column(String(50), nullable=False, unique=True, comment="权限编码")
|
||
description = Column(String(255), nullable=True, comment="权限描述")
|
||
createtime = Column(DateTime, default=datetime.utcnow, comment="创建时间")
|
||
updatetime = Column(
|
||
DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, comment="更新时间"
|
||
)
|
||
|
||
# 关联到Role（多对多）
|
||
roles = relationship(
|
||
"Role",
|
||
secondary="role_permission",
|
||
back_populates="permissions"
|
||
)
|
||
|
||
|
||
# 角色权限关联表（多对多）
|
||
role_permission = Table(
|
||
"role_permission",
|
||
Base.metadata,
|
||
Column("role_id", Integer, ForeignKey("role.id"), primary_key=True, comment="角色ID"),
|
||
Column("permission_id", Integer, ForeignKey("permission.id"), primary_key=True, comment="权限ID")
|
||
)
|