open-douyin/main.py

import socket
import time
from fastapi import FastAPI, Request  
from fastapi.responses import HTMLResponse  
from fastapi.staticfiles import StaticFiles  
from fastapi.templating import Jinja2Templates
import requests  
import uvicorn
from fastapi.responses import FileResponse,JSONResponse
from fastapi import FastAPI, Depends, HTTPException, Form  
import httpx
import os
# from db.user import UserOAuthToken
from config import *
from fastapi.middleware.cors import CORSMiddleware 
from db.user import DatabaseManager
from api.login import login_router
from api.redis import RedisSession
from starlette.middleware.sessions import SessionMiddleware
from contextlib import asynccontextmanager

app = FastAPI()  
app.add_middleware(  
    CORSMiddleware,  
    allow_origins=["*"],  
    allow_credentials=True,  
    allow_methods=["*"],  
    allow_headers=["*"], 
     
) 
app.include_router(login_router)  

# 使用 Redis 会话  
redis_session = RedisSession("redis://localhost:8001")  
app.add_middleware(SessionMiddleware, secret_key=JWT_SECRET_KEY)  

@asynccontextmanager
async def lifespan(app: FastAPI):
    # 在应用启动前运行的代码
    await redis_session.connect()  
    yield
    # 在应用关闭后运行的代码
    await redis_session.disconnect()

def get_session(request: Request):  
    return request.session  
    
@app.get("/")  
async def read_root(request: Request):  
    return {"message": "ok", "time":time.strftime("%Y-%m-%d_%H-%M-%S", time.localtime())}  


def main():
    print(f"https://open-douyin-cf.magong.site  公网代理地址，cloudflare dns proxy ,由 caddy 转发到 8600 端口")
    print(f"https://open-douyin-wk.magong.site  公网代理地址，cloudflare workers 转发到 8600 端口")
    print(f"http://sv-v2.magong.site:{PORT}  ⭐ 推荐，仅支持 ipv6 ，直连、满速、无延迟。缺点是不支持 https 协议，因为不经过 Caddy 代理，直达 Fastapi 没有配置 https")
    print(f"https://open-douyin.magong.site  内网穿透隧道，cloudflare tunnel ，经常访问不了")
    print(f"https://swl-8l9.pages.dev/  访问前端网站")
    uvicorn.run(app, host=None, port=PORT, log_level="info")

if __name__ == "__main__":
    main()