1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
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}
|
