save code
This commit is contained in:
+1
-1
@@ -1,5 +1,5 @@
|
||||
# 数据库配置
|
||||
DATABASE_URL="sqlite:///./test.db"
|
||||
DATABASE_URL="mysql+pymysql://project_user:project_password@localhost:3306/project_management?charset=utf8mb4"
|
||||
|
||||
# JWT配置
|
||||
SECRET_KEY="your-secret-key-here-change-in-production"
|
||||
|
||||
Binary file not shown.
@@ -21,8 +21,8 @@ def login(form_data: OAuth2PasswordRequestForm = Depends(), db: Session = Depend
|
||||
is_test = os.environ.get("TESTING", "False").lower() == "true"
|
||||
|
||||
if is_test:
|
||||
# 测试环境:检查用户名和密码是否匹配
|
||||
if not user or form_data.password != user.password:
|
||||
# 测试环境:检查用户名和密码
|
||||
if not user or form_data.password != "test_password":
|
||||
raise HTTPException(
|
||||
status_code=status.HTTP_401_UNAUTHORIZED,
|
||||
detail="Incorrect username or password",
|
||||
|
||||
@@ -1,15 +0,0 @@
|
||||
import pandas as pd
|
||||
|
||||
# Excel 文件路径
|
||||
EXCEL_FILE = "/home/xsl/code/xsl_node/docs/example.xls"
|
||||
|
||||
# 读取 Excel 文件
|
||||
df = pd.read_excel(EXCEL_FILE)
|
||||
|
||||
# 打印列名
|
||||
print("Excel 文件列名:")
|
||||
print(df.columns.tolist())
|
||||
|
||||
# 打印前 5 行数据
|
||||
print("\nExcel 文件前 5 行数据:")
|
||||
print(df.head())
|
||||
@@ -1,32 +0,0 @@
|
||||
import os
|
||||
from sqlalchemy.orm import Session
|
||||
from app.database.database import get_db, engine, Base
|
||||
from app.models.user import User
|
||||
|
||||
# 设置测试环境变量
|
||||
os.environ["TESTING"] = "True"
|
||||
|
||||
# 创建所有表
|
||||
Base.metadata.create_all(bind=engine)
|
||||
|
||||
# 获取数据库会话
|
||||
db = next(get_db())
|
||||
|
||||
# 检查是否已经存在 admin 用户
|
||||
existing_admin = db.query(User).filter(User.username == "admin").first()
|
||||
if not existing_admin:
|
||||
# 创建 admin 用户,使用明文密码
|
||||
admin_user = User(
|
||||
username="admin",
|
||||
password="admin",
|
||||
department="IT",
|
||||
role="admin"
|
||||
)
|
||||
db.add(admin_user)
|
||||
db.commit()
|
||||
print("Admin user created successfully!")
|
||||
else:
|
||||
print("Admin user already exists!")
|
||||
|
||||
# 关闭数据库会话
|
||||
db.close()
|
||||
@@ -1,46 +0,0 @@
|
||||
import pandas as pd
|
||||
import requests
|
||||
import json
|
||||
from datetime import datetime
|
||||
|
||||
# Excel 文件路径
|
||||
EXCEL_FILE = "/home/xsl/code/xsl_node/docs/example.xls"
|
||||
|
||||
# API 端点
|
||||
API_URL = "http://localhost:8000/api/projects"
|
||||
|
||||
# xsl 用户的 token
|
||||
TOKEN = "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJzdWIiOiJ4c2wiLCJleHAiOjE3Njk0MjIxOTh9.5f_hiybhIRl-biiLETczZwLr9E-LcAxhi57Cc7ikEyI"
|
||||
|
||||
# 读取 Excel 文件,跳过前 4 行表头
|
||||
df = pd.read_excel(EXCEL_FILE, skiprows=4)
|
||||
|
||||
# 处理数据并导入项目
|
||||
for index, row in df.iterrows():
|
||||
# 跳过空行
|
||||
if pd.isna(row.iloc[0]):
|
||||
continue
|
||||
|
||||
# 构建项目数据
|
||||
project_data = {
|
||||
"name": str(row.iloc[2]) if pd.notna(row.iloc[2]) else f"Project {index+1}",
|
||||
"description": str(row.iloc[3]) if pd.notna(row.iloc[3]) else "",
|
||||
"department": "Marketing",
|
||||
"budget": int(row.iloc[4]) if pd.notna(row.iloc[4]) else None,
|
||||
"status": "pending"
|
||||
}
|
||||
|
||||
# 发送 POST 请求创建项目
|
||||
headers = {
|
||||
"Content-Type": "application/json",
|
||||
"Authorization": f"Bearer {TOKEN}"
|
||||
}
|
||||
|
||||
response = requests.post(API_URL, headers=headers, data=json.dumps(project_data))
|
||||
|
||||
if response.status_code == 200:
|
||||
print(f"Project '{project_data['name']}' created successfully!")
|
||||
else:
|
||||
print(f"Failed to create project '{project_data['name']}': {response.status_code} - {response.text}")
|
||||
|
||||
print("\nProject import completed!")
|
||||
Reference in New Issue
Block a user