于小丘
51b6de921b
- Added user authentication and registration endpoints with JWT support. - Created tag management routes for creating and deleting tags. - Implemented value-added service (VAS) endpoints for managing storage packs and orders. - Developed WebDAV account management routes for creating, updating, and deleting accounts. - Introduced slave router for handling file uploads, downloads, and aria2 task management. - Enhanced JWT utility functions for token creation and secret key management. - Established lifespan management for FastAPI application startup and shutdown processes. - Integrated password handling utilities with Argon2 hashing and two-factor authentication support.
39 lines
982 B
Python
39 lines
982 B
Python
from fastapi import FastAPI
|
|
from contextlib import asynccontextmanager
|
|
|
|
__on_startup: list[callable] = []
|
|
__on_shutdown: list[callable] = []
|
|
|
|
def add_startup(func: callable):
|
|
"""
|
|
注册一个函数，在应用启动时调用。
|
|
|
|
:param func: 需要注册的函数。它应该是一个异步函数。
|
|
"""
|
|
__on_startup.append(func)
|
|
|
|
def add_shutdown(func: callable):
|
|
"""
|
|
注册一个函数，在应用关闭时调用。
|
|
|
|
:param func: 需要注册的函数。
|
|
"""
|
|
__on_shutdown.append(func)
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""
|
|
应用程序的生命周期管理器。
|
|
|
|
此函数在应用启动时执行所有注册的启动函数，
|
|
并在应用关闭时执行所有注册的关闭函数。
|
|
"""
|
|
# Execute all startup functions
|
|
for func in __on_startup:
|
|
await func()
|
|
|
|
yield
|
|
|
|
# Execute all shutdown functions
|
|
for func in __on_shutdown:
|
|
await func() |