DidaAPI/main.py at main · LifeArchiveProject/DidaAPI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
"""滴答清单API主应用"""
import os
from contextlib import asynccontextmanager

import uvicorn
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles

from core import config, db
from routers import auth, tasks, system, projects, statistics, pomodoros, habits, users, export
from services import wechat_service
from utils import app_logger


@asynccontextmanager
async def lifespan(app: FastAPI):
    """应用生命周期管理"""
    # 启动时执行
    app_logger.info("滴答清单API服务启动中...")

    # 初始化数据库
    db.init_database()
    app_logger.info("数据库初始化完成")

    yield

    # 关闭时执行
    app_logger.info("滴答清单API服务关闭中...")
    await wechat_service.close()
    app_logger.info("服务已关闭")


# 创建FastAPI应用
app = FastAPI(
    title=config.app.get('name', '滴答清单API'),
    version=config.app.get('version', '1.0.0'),
    description="""
## 滴答清单API接口文档

这是一个滴答清单的Web端API接口项目，**在原始滴答清单API基础上进行了封装**，提供更简单易用的接口。

### 🚀 快速开始

#### 微信扫码登录
**[📱 点击这里体验微信扫码登录](/auth/wechat/login)**
    """,
    docs_url="/docs",
    redoc_url="/redoc",
    lifespan=lifespan
)

# 添加CORS中间件
app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],  # 在生产环境中应该限制具体的域名
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

# 创建静态文件目录
static_dir = "static"
if not os.path.exists(static_dir):
    os.makedirs(static_dir)

# 挂载静态文件
app.mount("/static", StaticFiles(directory=static_dir), name="static")

# 注册路由
app.include_router(auth.router)
app.include_router(tasks.router)
app.include_router(projects.router)
app.include_router(statistics.router)
app.include_router(pomodoros.router)
app.include_router(pomodoros.timer_router)
app.include_router(habits.router)
app.include_router(users.router)
app.include_router(export.router)
app.include_router(system.router)


@app.get("/", summary="根路径", description="API服务根路径，返回基本信息")
async def root():
    """根路径接口"""
    return {
        "message": "欢迎使用滴答清单API",
        "version": config.app.get('version', '1.0.0'),
        "docs": "/docs",
        "redoc": "/redoc",
        "health": "/auth/health",
        "wechat_login": "/auth/wechat/login",  # 添加微信登录页面
        "auth_status": "/tasks/status",
        "url_management": "/system/urls",
        "system_info": "/system/info",
        "api_modules": {
            "authentication": "/auth/",
            "tasks": "/tasks/",
            "projects": "/projects/",
            "statistics": "/statistics/",
            "pomodoros": "/pomodoros/",
            "habits": "/habits/",
            "users": "/users/",
            "custom": "/custom/",
            "system": "/system/"
        }
    }


def main():
    """主函数，启动应用"""
    app_config = config.app

    uvicorn.run(
        "main:app",
        host=app_config.get('host', '127.0.0.1'),
        port=app_config.get('port', 8000),
        reload=app_config.get('debug', True),
        log_level="info"
    )


if __name__ == "__main__":
    main()