初次提交
This commit is contained in:
314
backend/start_8080.py
Normal file
314
backend/start_8080.py
Normal file
@@ -0,0 +1,314 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
云盘应用启动脚本 - 端口8080版本
|
||||
包含完整的API测试功能
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
# 添加当前目录到Python路径
|
||||
current_dir = Path(__file__).parent
|
||||
sys.path.insert(0, str(current_dir))
|
||||
|
||||
from fastapi import FastAPI
|
||||
from fastapi.middleware.cors import CORSMiddleware
|
||||
import uvicorn
|
||||
import subprocess
|
||||
import time
|
||||
import requests
|
||||
from typing import Dict, List
|
||||
|
||||
# 导入应用模块
|
||||
try:
|
||||
from app.core.config import settings
|
||||
from app.api.v1.endpoints import health, auth, files
|
||||
APP_AVAILABLE = True
|
||||
except ImportError as e:
|
||||
print(f"⚠️ App模块导入失败: {e}")
|
||||
print("🔧 使用简化模式启动...")
|
||||
APP_AVAILABLE = False
|
||||
|
||||
def create_fastapi_app():
|
||||
"""创建FastAPI应用"""
|
||||
if APP_AVAILABLE:
|
||||
# 使用完整的应用
|
||||
app = FastAPI(
|
||||
title="云盘应用 API",
|
||||
description="现代化的云存储Web应用后端API",
|
||||
version="1.0.0",
|
||||
docs_url="/docs",
|
||||
redoc_url="/redoc"
|
||||
)
|
||||
|
||||
# CORS中间件
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=settings.ALLOWED_HOSTS,
|
||||
allow_credentials=True,
|
||||
allow_methods=["*"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
|
||||
# 包含路由
|
||||
app.include_router(health.router, prefix="/api/v1", tags=["health"])
|
||||
app.include_router(auth.router, prefix="/api/v1/auth", tags=["authentication"])
|
||||
app.include_router(files.router, prefix="/api/v1/files", tags=["files"])
|
||||
|
||||
@app.get("/")
|
||||
async def root():
|
||||
return {
|
||||
"message": "云盘应用 API",
|
||||
"version": "1.0.1",
|
||||
"docs": "/docs",
|
||||
"health": "/api/v1/health"
|
||||
}
|
||||
|
||||
return app
|
||||
else:
|
||||
# 创建简化版本的应用
|
||||
app = FastAPI(
|
||||
title="云盘应用 API (简化版)",
|
||||
description="云存储Web应用后端API - 简化版本",
|
||||
version="1.0.0",
|
||||
docs_url="/docs",
|
||||
redoc_url="/redoc"
|
||||
)
|
||||
|
||||
# CORS中间件
|
||||
app.add_middleware(
|
||||
CORSMiddleware,
|
||||
allow_origins=["*"],
|
||||
allow_credentials=True,
|
||||
allow_methods=["*"],
|
||||
allow_headers=["*"],
|
||||
)
|
||||
|
||||
@app.get("/")
|
||||
async def root():
|
||||
return {
|
||||
"message": "云盘应用 API (简化版)",
|
||||
"version": "1.0.0",
|
||||
"docs": "/docs",
|
||||
"health": "/health"
|
||||
}
|
||||
|
||||
@app.get("/health")
|
||||
async def health_check():
|
||||
return {
|
||||
"status": "healthy",
|
||||
"message": "服务运行正常",
|
||||
"mode": "simplified"
|
||||
}
|
||||
|
||||
@app.get("/api/v1/health")
|
||||
async def api_health():
|
||||
return {
|
||||
"status": "healthy",
|
||||
"timestamp": time.time(),
|
||||
"version": "1.0.0"
|
||||
}
|
||||
|
||||
return app
|
||||
|
||||
def test_api_endpoints(base_url: str = "http://localhost:8080") -> Dict[str, any]:
|
||||
"""测试API端点"""
|
||||
print("\n🧪 开始测试API端点...")
|
||||
print("=" * 50)
|
||||
|
||||
results = {}
|
||||
|
||||
# 测试端点列表
|
||||
endpoints = [
|
||||
("/", "根路径"),
|
||||
("/health", "健康检查"),
|
||||
("/api/v1/health", "API健康检查"),
|
||||
("/docs", "API文档"),
|
||||
("/redoc", "ReDoc文档"),
|
||||
("/openapi.json", "OpenAPI规范")
|
||||
]
|
||||
|
||||
for endpoint, description in endpoints:
|
||||
try:
|
||||
print(f"📍 测试 {description}: {endpoint}")
|
||||
|
||||
if endpoint in ["/docs", "/redoc"]:
|
||||
# 对于文档端点，只检查HTTP状态
|
||||
response = requests.get(f"{base_url}{endpoint}", timeout=5)
|
||||
if response.status_code == 200:
|
||||
print(f" ✅ {description} - 状态码: {response.status_code}")
|
||||
results[endpoint] = {"status": "success", "code": response.status_code}
|
||||
else:
|
||||
print(f" ❌ {description} - 状态码: {response.status_code}")
|
||||
results[endpoint] = {"status": "failed", "code": response.status_code}
|
||||
else:
|
||||
# 对于API端点，检查响应内容
|
||||
response = requests.get(f"{base_url}{endpoint}", timeout=5)
|
||||
if response.status_code == 200:
|
||||
data = response.json()
|
||||
print(f" ✅ {description} - 响应: {data}")
|
||||
results[endpoint] = {"status": "success", "data": data}
|
||||
else:
|
||||
print(f" ❌ {description} - 状态码: {response.status_code}")
|
||||
results[endpoint] = {"status": "failed", "code": response.status_code}
|
||||
|
||||
except requests.exceptions.ConnectionError:
|
||||
print(f" ❌ {description} - 连接失败")
|
||||
results[endpoint] = {"status": "connection_failed"}
|
||||
except requests.exceptions.Timeout:
|
||||
print(f" ❌ {description} - 请求超时")
|
||||
results[endpoint] = {"status": "timeout"}
|
||||
except Exception as e:
|
||||
print(f" ❌ {description} - 错误: {e}")
|
||||
results[endpoint] = {"status": "error", "error": str(e)}
|
||||
|
||||
return results
|
||||
|
||||
def start_server():
|
||||
"""启动服务器"""
|
||||
print("🚀 启动云盘应用服务...")
|
||||
print("=" * 50)
|
||||
|
||||
# 创建必要目录
|
||||
os.makedirs("logs", exist_ok=True)
|
||||
os.makedirs("uploads", exist_ok=True)
|
||||
|
||||
# 创建FastAPI应用
|
||||
app = create_fastapi_app()
|
||||
|
||||
# 获取本机IP
|
||||
import socket
|
||||
try:
|
||||
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
||||
s.connect(("8.8.8.8", 80))
|
||||
local_ip = s.getsockname()[0]
|
||||
s.close()
|
||||
except:
|
||||
local_ip = "127.0.0.1"
|
||||
|
||||
# 显示启动信息
|
||||
print(f"📍 本地访问:")
|
||||
print(f" 根路径: http://localhost:8080")
|
||||
print(f" API文档: http://localhost:8080/docs")
|
||||
print(f" ReDoc: http://localhost:8080/redoc")
|
||||
print(f" 健康检查: http://localhost:8080/api/v1/health")
|
||||
print("")
|
||||
print(f"🌐 网络访问:")
|
||||
print(f" 根路径: http://{local_ip}:8080")
|
||||
print(f" API文档: http://{local_ip}:8080/docs")
|
||||
print(f" ReDoc: http://{local_ip}:8080/redoc")
|
||||
print("")
|
||||
print(f"🔧 API测试:")
|
||||
print(f" 自动测试将在服务启动后执行")
|
||||
print("=" * 50)
|
||||
print("⏹️ 按 Ctrl+C 停止服务")
|
||||
print("")
|
||||
|
||||
# 启动服务器
|
||||
try:
|
||||
uvicorn.run(
|
||||
app,
|
||||
host="0.0.0.0",
|
||||
port=8080,
|
||||
reload=False,
|
||||
access_log=True,
|
||||
log_level="info"
|
||||
)
|
||||
except KeyboardInterrupt:
|
||||
print("\n🛑 服务已停止")
|
||||
except Exception as e:
|
||||
print(f"❌ 启动失败: {e}")
|
||||
return False
|
||||
|
||||
return True
|
||||
|
||||
def run_with_auto_test():
|
||||
"""启动服务器并自动测试"""
|
||||
print("🤖 自动模式：启动服务器并进行API测试")
|
||||
|
||||
# 启动服务器（在后台）
|
||||
import subprocess
|
||||
import signal
|
||||
import threading
|
||||
|
||||
# 启动服务器进程
|
||||
server_process = subprocess.Popen([
|
||||
sys.executable, __file__
|
||||
], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
||||
|
||||
# 等待服务器启动
|
||||
print("⏳ 等待服务器启动...")
|
||||
time.sleep(5)
|
||||
|
||||
try:
|
||||
# 测试API端点
|
||||
results = test_api_endpoints()
|
||||
|
||||
# 显示测试结果
|
||||
print("\n📊 测试结果汇总:")
|
||||
print("=" * 50)
|
||||
|
||||
success_count = sum(1 for r in results.values() if r.get("status") == "success")
|
||||
total_count = len(results)
|
||||
|
||||
print(f"成功: {success_count}/{total_count}")
|
||||
|
||||
for endpoint, result in results.items():
|
||||
status = result.get("status", "unknown")
|
||||
if status == "success":
|
||||
print(f" ✅ {endpoint}")
|
||||
else:
|
||||
print(f" ❌ {endpoint} - {status}")
|
||||
|
||||
if success_count == total_count:
|
||||
print("\n🎉 所有API端点测试通过！")
|
||||
else:
|
||||
print(f"\n⚠️ {total_count - success_count} 个端点测试失败")
|
||||
|
||||
finally:
|
||||
# 停止服务器
|
||||
print("\n🛑 停止服务器...")
|
||||
server_process.terminate()
|
||||
server_process.wait()
|
||||
|
||||
if __name__ == "__main__":
|
||||
import argparse
|
||||
|
||||
parser = argparse.ArgumentParser(description="云盘应用启动器 - 端口8080")
|
||||
parser.add_argument("--test", action="store_true", help="启动后自动测试API端点")
|
||||
parser.add_argument("--auto", action="store_true", help="自动模式：启动并测试")
|
||||
args = parser.parse_args()
|
||||
|
||||
if args.auto:
|
||||
run_with_auto_test()
|
||||
elif args.test:
|
||||
# 先启动，再测试
|
||||
print("🚀 启动服务器...")
|
||||
app = create_fastapi_app()
|
||||
|
||||
# 在单独线程中启动服务器
|
||||
import threading
|
||||
|
||||
def run_server():
|
||||
uvicorn.run(app, host="0.0.0.0", port=8080, reload=False, log_level="warning")
|
||||
|
||||
server_thread = threading.Thread(target=run_server, daemon=True)
|
||||
server_thread.start()
|
||||
|
||||
# 等待服务器启动
|
||||
time.sleep(3)
|
||||
|
||||
# 测试API
|
||||
test_api_endpoints()
|
||||
|
||||
print("\n服务器继续运行，按 Ctrl+C 停止...")
|
||||
try:
|
||||
while True:
|
||||
time.sleep(1)
|
||||
except KeyboardInterrupt:
|
||||
print("\n🛑 服务已停止")
|
||||
else:
|
||||
# 正常启动
|
||||
start_server()
|
||||
Reference in New Issue
Block a user