remake了所有的models:(

models.py

@@ -1,51 +1,41 @@
 # -*- coding: utf-8 -*-
-# @Time    : 2024/11/18 下午7:00
+# @Time    : 2024/11/19 下午6:09
 # @Author  : 河瞬
-# @FileName: models.py
+# @FileName: new_models.py
 # @Software: PyCharm
-from sqlalchemy import create_engine, Column, Integer, String, DateTime, ForeignKey, Text
-from sqlalchemy.orm import relationship, sessionmaker, declarative_base
 from datetime import datetime
+from typing import List, Optional
 
-# 创建基类
-Base = declarative_base()
+from sqlmodel import Field, Relationship, SQLModel, DateTime
 
 
-class Tenant(Base):
+class Tenant(SQLModel, table=True):
     __tablename__ = 'Tenant'
-    id = Column(Integer, primary_key=True)
-    name = Column(String(100), nullable=False)
-    users = relationship('User', back_populates='tenant', cascade='all, delete-orphan')
-    projects = relationship('Project', back_populates='owner', cascade='all, delete-orphan')
+    id: Optional[int] = Field(default=None, primary_key=True)
+    name: str = Field(index=True)
+    users: List["User"] = Relationship(back_populates="tenant")
+    projects: List["Project"] = Relationship(back_populates="owner")
 
 
-class User(Base):
+class User(SQLModel, table=True):
     __tablename__ = 'User'
-    id = Column(Integer, primary_key=True)
-    username = Column(String(100), nullable=False)
-    password = Column(String(100), nullable=False)
-    role = Column(Integer, nullable=False)  # 0: 超级管理员 1：租户管理员 2：评估员 3：审核员
-    tenant_id = Column(Integer, ForeignKey('Tenant.id'), nullable=False)
-    tenant = relationship('Tenant', back_populates='users')
+    id: Optional[int] = Field(default=None, primary_key=True)
+    username: str = Field(index=True)
+    password: str
+    role: int
+    tenant_id: int = Field(default=None, foreign_key="Tenant.id")
+    tenant: Tenant = Relationship(back_populates="users")
+    projects: List["Project"] = Relationship(back_populates="estimators")
 
 
-class Project(Base):
+class Project(SQLModel, table=True):
     __tablename__ = 'Project'
-    id = Column(Integer, primary_key=True)
-    name = Column(String(100), nullable=False)
-    requirement = Column(String(200), nullable=True)
-    start_time = Column(DateTime, nullable=False, default=datetime.utcnow)
-    deadline = Column(DateTime, nullable=False)
-    owner_id = Column(Integer, ForeignKey('Tenant.id'), nullable=False)
-    owner = relationship('Tenant', back_populates='projects')
-
-
-if __name__ == "__main__":
-    # 创建数据库引擎
-    engine = create_engine('sqlite:///test.db')
-
-    # 创建所有表
-    Base.metadata.create_all(engine)
-
-    # 创建会话
-    SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+    id: Optional[int] = Field(default=None, primary_key=True)
+    name: str = Field(index=True)
+    requirement: str
+    owner_id: int = Field(default=None, foreign_key="Tenant.id")
+    owner: Tenant = Relationship(back_populates="projects")
+    start_time: DateTime = Field(default=datetime.utcnow)
+    deadline: DateTime
+    estimators: List["User"] = Relationship(back_populates="projects")
+    auditors: List["User"] = Relationship(back_populates="projects")