import sqlite3 from fastapi import FastAPI from enum import Enum from pathlib import Path app = FastAPI() class ModelName(str, Enum): alexnet = "normal" resnet = "nearR18" lenet = "r18" @app.get("/") def index(): return {"我只是一个API": "使用方法:https://blog.lolihub.icu/#/api"} @app.get("/setu/{h_type}") def get_setu(h_type: str): if h_type == ModelName.alexnet: con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'normal.db') cur = con.cursor() msg = cur.execute('SELECT * FROM normal ORDER BY RANDOM() limit 1;') for i in msg: pid = i[0] title = i[1] tags = i[2] account = i[3] name = i[4] u_id = i[5] user_link = i[6] img = i[7] print('success!') con.commit() con.close() return {"Pid": pid, 0:{"title": title,"tags": tags,"account": account, "name": name,"u_id": u_id, "user_link": user_link, "img": img}} elif h_type == ModelName.resnet: con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'nearR18.db') cur = con.cursor() msg = cur.execute('SELECT * FROM nearR18 ORDER BY RANDOM() limit 1;') for i in msg: pid = i[0] title = i[1] tags = i[2] account = i[3] name = i[4] u_id = i[5] user_link = i[6] img = i[7] print('success!') con.commit() con.close() return {"Pid": pid, 0:{"title": title,"tags": tags,"account": account, "name": name,"u_id": u_id, "user_link": user_link, "img": img}} elif h_type == ModelName.lenet: con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'setu' / 'r18.db') cur = con.cursor() msg = cur.execute('SELECT * FROM r18 ORDER BY RANDOM() limit 1;') for i in msg: pid = i[0] title = i[1] tags = i[2] account = i[3] name = i[4] u_id = i[5] user_link = i[6] img = i[7] con.commit() con.close() return {"Pid": pid, 0:{"title": title,"tags": tags,"account": account, "name": name,"u_id": u_id, "user_link": user_link, "img": img}} @app.get("/cloudmusic") async def Depression(): con = sqlite3.connect(Path('.') / 'ATRI' / 'data' / 'sqlite' / 'cloudmusic' / 'cloudmusic.db') cur = con.cursor() msg = cur.execute('SELECT * FROM cloudmusic ORDER BY RANDOM() limit 1;') for i in msg: msg = i[0] con.commit() con.close() return {"msg": msg}