from fastapi import FastAPI, Request, HTTPException, Depends from fastapi.responses import HTMLResponse, FileResponse, Response import markdown from fastapi.staticfiles import StaticFiles from fastapi.templating import Jinja2Templates import json, yaml import os import uvicorn from datetime import datetime from xml.etree import ElementTree as ET from sqlalchemy.orm import Session from database import get_db, RawResponse app = FastAPI() templates = Jinja2Templates(directory="templates") app.mount("/static", StaticFiles(directory="static"), name="static") # 加载所有问卷数据 def load_all_scales(): scales = {} tags = [] for root, dirs, files in os.walk(os.path.realpath('scales')): for filename in files: if filename.endswith(('.yaml', '.yml')): try: with open(os.path.join(root, filename), 'r', encoding='utf-8') as f: scale = yaml.safe_load(f) scale['instructions']=markdown.markdown(scale['instructions'], extensions=['fenced_code','tables','mdx_math']) scale['descriptions']=markdown.markdown(scale['descriptions'], extensions=['fenced_code','tables','mdx_math']) scale['abstract']=markdown.markdown(scale['abstract'], extensions=['fenced_code','tables','mdx_math']) if 'tag' not in scale: scale['tag']='其他' if scale['tag'] not in tags: tags.append(scale['tag']) scale_id = os.path.splitext(filename)[0] # 使用文件名作为标识 scales[scale_id] = scale except Exception as e: print(f"Error loading scale {filename}: {e}") return tags, scales @app.get("/", response_class=HTMLResponse) async def index(request: Request): tags, _ = load_all_scales() # 新增读取README.md的逻辑 readme_content = "" try: with open("README.md", "r", encoding="utf-8") as f: readme_content = markdown.markdown(f.read()) except FileNotFoundError: pass # 如果README不存在则静默失败 return templates.TemplateResponse("index.html", { "request": request, "tags": tags, "readme_content": readme_content # 新增模板变量 }) @app.get("/tag/{tag}", response_class=HTMLResponse) async def list(request: Request, tag: str): tags, scales = load_all_scales() return templates.TemplateResponse("list.html", { "request": request, "tags": tags, "scales": scales, "tag": tag }) @app.get("/scales/{scale_id}", response_class=HTMLResponse) async def scale(request: Request, scale_id: str): tags, scales = load_all_scales() scale = scales.get(scale_id) if scale: return templates.TemplateResponse("scale.html", { "request": request, "scale_id": scale_id, "scale": scale, "tags":tags }) raise HTTPException(status_code=404, detail="问卷未找到") @app.post("/scales/{scale_id}", response_class=HTMLResponse) async def result(request: Request, scale_id: str, db: Session = Depends(get_db)): form_data = await request.form() tags, scales = load_all_scales() scale = scales.get(scale_id) if scale: # Save response to database db_response = RawResponse( scale_id=scale_id, user_agent=request.headers.get("user-agent", "Unknown"), ip_address=request.client.host, response=dict(form_data) ) db.add(db_response) db.commit() responses = {} average = {} options = {} for subscale, qids in scale['subscales'].items(): responses[subscale] = 0 min_val = min(scale['options'].keys()) max_val = max(scale['options'].keys()) options[subscale] = [min_val*len(qids),max_val*len(qids)] for qid in qids: if qid<0: responses[subscale] += min_val + max_val - int(form_data[str(-qid)]) else: responses[subscale] += int(form_data[str(qid)]) average[subscale] = round(responses[subscale]/len(qids),2) return templates.TemplateResponse("result.html", { "request": request, "responses": responses, "average": average, "options": options, "scale": scale, "tags":tags }) raise HTTPException(status_code=404, detail="问卷未找到") def generate_sitemap(): # Create the root element urlset = ET.Element("urlset", xmlns="http://www.sitemaps.org/schemas/sitemap/0.9") latest_mtime = max( os.path.getmtime(os.path.join(root, f)) for root, _, files in os.walk('scales') for f in files if f.endswith(('.yaml', '.yml')) ) # Add static routes static_routes = ["/"] # Add your static routes here for route in static_routes: url = ET.SubElement(urlset, "url") ET.SubElement(url, "loc").text = f"https://psychoscales.org{route}" ET.SubElement(url, "lastmod").text = datetime.fromtimestamp(latest_mtime).strftime("%Y-%m-%d") ET.SubElement(url, "changefreq").text = "monthly" ET.SubElement(url, "priority").text = "0.8" # Add dynamic tag routes tags, scales = load_all_scales() for tag in tags: url = ET.SubElement(urlset, "url") ET.SubElement(url, "loc").text = f"https://psychoscales.org/tag/{tag}" ET.SubElement(url, "lastmod").text = datetime.fromtimestamp(latest_mtime).strftime("%Y-%m-%d") ET.SubElement(url, "changefreq").text = "weekly" ET.SubElement(url, "priority").text = "0.6" # Add dynamic scale routes for scale_id in scales.keys(): url = ET.SubElement(urlset, "url") ET.SubElement(url, "loc").text = f"https://psychoscales.org/scales/{scale_id}" # For individual scale pages, use the actual file modification time scale_file = os.path.join('scales', f"{scale_id}.yaml") if os.path.exists(scale_file): mtime = os.path.getmtime(scale_file) ET.SubElement(url, "lastmod").text = datetime.fromtimestamp(mtime).strftime("%Y-%m-%d") ET.SubElement(url, "changefreq").text = "monthly" ET.SubElement(url, "priority").text = "0.6" # Convert to string return ET.tostring(urlset, encoding='unicode', method='xml') # Mount all files from public directory to root @app.get("/{filename}") async def get_public_file(filename: str): public_path = os.path.join("public", filename) if filename == "sitemap.xml": return Response( content=generate_sitemap(), media_type="application/xml" ) if os.path.isfile(public_path): return FileResponse(public_path) raise HTTPException(status_code=404, detail="File not found") if __name__ == '__main__': uvicorn.run(app,host='0.0.0.0',port=8000)