summaryrefslogtreecommitdiff
path: root/API.py
diff options
context:
space:
mode:
Diffstat (limited to 'API.py')
-rw-r--r--API.py87
1 files changed, 0 insertions, 87 deletions
diff --git a/API.py b/API.py
deleted file mode 100644
index f85c052..0000000
--- a/API.py
+++ /dev/null
@@ -1,87 +0,0 @@
-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"
-
-
-def index():
- return {"我只是一个API": "使用方法:https://blog.lolihub.icu/#/api"}
-
-
[email protected]("/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}}
-
-
[email protected]("/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} \ No newline at end of file