commit 13c317e80c297505c26ce0d55e94989299ab9f15 Author: heshunme Date: Mon Nov 18 19:47:19 2024 +0800 初始化项目：models+fastapi登录示例 diff --git a/.idea/.gitignore b/.idea/.gitignore new file mode 100644 index 0000000..35410ca --- /dev/null +++ b/.idea/.gitignore @@ -0,0 +1,8 @@ +# 默认忽略的文件 +/shelf/ +/workspace.xml +# 基于编辑器的 HTTP 客户端请求 +/httpRequests/ +# Datasource local storage ignored files +/dataSources/ +/dataSources.local.xml diff --git a/.idea/CostEvalPlatform.iml b/.idea/CostEvalPlatform.iml new file mode 100644 index 0000000..d0876a7 --- /dev/null +++ b/.idea/CostEvalPlatform.iml @@ -0,0 +1,8 @@ + + + + + + + + \ No newline at end of file diff --git a/.idea/dataSources.xml b/.idea/dataSources.xml new file mode 100644 index 0000000..95b90ca --- /dev/null +++ b/.idea/dataSources.xml @@ -0,0 +1,20 @@ + + + + + sqlite.xerial + true + org.sqlite.JDBC + jdbc:sqlite:E:\pyprojects\CostEvalPlatform\test.db + $ProjectFileDir$ + + + file://$APPLICATION_CONFIG_DIR$/jdbc-drivers/Xerial SQLiteJDBC/3.45.1/org/xerial/sqlite-jdbc/3.45.1.0/sqlite-jdbc-3.45.1.0.jar + + + file://$APPLICATION_CONFIG_DIR$/jdbc-drivers/Xerial SQLiteJDBC/3.45.1/org/slf4j/slf4j-api/1.7.36/slf4j-api-1.7.36.jar + + + + + \ No newline at end of file diff --git a/.idea/inspectionProfiles/Project_Default.xml b/.idea/inspectionProfiles/Project_Default.xml new file mode 100644 index 0000000..1c665c8 --- /dev/null +++ b/.idea/inspectionProfiles/Project_Default.xml @@ -0,0 +1,53 @@ + + + + \ No newline at end of file diff --git a/.idea/inspectionProfiles/profiles_settings.xml b/.idea/inspectionProfiles/profiles_settings.xml new file mode 100644 index 0000000..105ce2d --- /dev/null +++ b/.idea/inspectionProfiles/profiles_settings.xml @@ -0,0 +1,6 @@ + + + + \ No newline at end of file diff --git a/.idea/misc.xml b/.idea/misc.xml new file mode 100644 index 0000000..63337c1 --- /dev/null +++ b/.idea/misc.xml @@ -0,0 +1,32 @@ + + + + + + + + + + EditorConfig + + + GitHub 操作 + + + 正则表达式 + + + 版本控制 + + + + + 用户定义 + + + + + + + \ No newline at end of file diff --git a/.idea/modules.xml b/.idea/modules.xml new file mode 100644 index 0000000..719a6e7 --- /dev/null +++ b/.idea/modules.xml @@ -0,0 +1,8 @@ + + + + + + + + \ No newline at end of file diff --git a/.idea/vcs.xml b/.idea/vcs.xml new file mode 100644 index 0000000..94a25f7 --- /dev/null +++ b/.idea/vcs.xml @@ -0,0 +1,6 @@ + + + + + + \ No newline at end of file diff --git a/main.py b/main.py new file mode 100644 index 0000000..ba1a1fe --- /dev/null +++ b/main.py @@ -0,0 +1,55 @@ +from fastapi import FastAPI, HTTPException, Response, Depends +from typing import Optional +from datetime import datetime, timedelta +from jose import JWTError, jwt +from models import * + +app = FastAPI() + +# 创建数据库引擎 +engine = create_engine('sqlite:///test.db') + +# 创建所有表 +Base.metadata.create_all(engine) + +# 创建会话 +SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) +session = SessionLocal() + +# 用于生成和验证JWT的密钥 +SECRET_KEY = "your_secret_key" +ALGORITHM = "HS256" + + +# 生成JWT token +def create_access_token(data: dict, expires_delta: Optional[timedelta] = None): + to_encode = data.copy() + if expires_delta: + expire = datetime.utcnow() + expires_delta + else: + expire = datetime.utcnow() + timedelta(minutes=15) + to_encode.update({"exp": expire}) + encoded_jwt = jwt.encode(to_encode, SECRET_KEY, algorithm=ALGORITHM) + return encoded_jwt + + +# 登录路由 +@app.post("/api/s1/login") +async def login(response: Response, user_data: dict): + # 查询用户 + user = session.query(User).filter(User.name == user_data['username']).first() + + # 验证用户名和密码 + if not user or user.password != user_data['password']: + raise HTTPException(status_code=401, detail="Login failed") + + # 生成JWT token + token = create_access_token(data={"sub": user.name}) + + # 设置cookie + response.set_cookie(key="session_token", value=token, httponly=True) + + # 关闭数据库会话 + session.close() + + return {"message": "Login successful"} diff --git a/models.py b/models.py new file mode 100644 index 0000000..1a9c4b7 --- /dev/null +++ b/models.py @@ -0,0 +1,50 @@ +# -*- coding: utf-8 -*- +# @Time : 2024/11/18 下午7:00 +# @Author : 河瞬 +# @FileName: models.py +# @Software: PyCharm +from sqlalchemy import create_engine, Column, Integer, String, DateTime, Enum, ForeignKey, Text +from sqlalchemy.orm import relationship, sessionmaker, declarative_base +from datetime import datetime + +# 创建基类 +Base = declarative_base() + + +class Tenant(Base): + __tablename__ = 'Tenant' + id = Column(Integer, primary_key=True) + name = Column(String(100), nullable=False) + users = relationship('User', backref='Tenant', cascade='all, delete-orphan') + projects = relationship('Project', backref='Tenant', cascade='all, delete-orphan') + + +class User(Base): + __tablename__ = 'User' + id = Column(Integer, primary_key=True) + name = 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) + + +class Project(Base): + __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', backref='projects') + + +if __name__ == "__main__": + # 创建数据库引擎 + engine = create_engine('sqlite:///test.db') + + # 创建所有表 + Base.metadata.create_all(engine) + + # 创建会话 + SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) \ No newline at end of file diff --git a/test.db b/test.db new file mode 100644 index 0000000..5d2c592 Binary files /dev/null and b/test.db differ