File size: 1,358 Bytes
61a1270 38a6743 61a1270 7d0b12c 61a1270 36c0d01 61a1270 d7effdb 61a1270 38a6743 61a1270 12657d7 61a1270 d558da0 61a1270 d558da0 d7effdb d558da0 d7effdb d558da0 d7effdb d558da0 d7effdb d558da0 |
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 |
import os
import time
import sqlite3
import threading
from pydantic import BaseModel
from fastapi import FastAPI
from fastapi.responses import HTMLResponse
from huggingface_hub import HfApi
from filesystem import upload, download
app = FastAPI()
class sql(BaseModel):
sql: str
def Upload():
download()
while True:
try:
upload()
except:
pass
time.sleep(30)
def Connect():
conn = sqlite3.connect("database.db")
cursor = conn.cursor()
return conn, cursor
def Close(conn: sqlite3.Connection):
conn.commit()
conn.close()
thread = threading.Thread(target=Upload)
thread.start()
@app.get("/", response_class=HTMLResponse)
def HuggingFaceSpaceRoot():
return open("usage.html").read()
@app.head("/")
def UptimeRobot():
return "OK"
def execute_sql(sql: str):
conn, cursor = Connect()
cursor.execute(sql)
Close(conn)
@app.post("/sql")
def execute_sql_post(param: sql):
return execute_sql(param.sql)
@app.get("/sql")
def execute_sql_get(sql: str):
return execute_sql(sql)
def select(sql: str):
conn, cursor = Connect()
result = cursor.execute(sql).fetchall()
Close(conn)
return result
@app.post("/select")
def select_post(param: sql):
return select(param.sql)
@app.get("/select")
def select_get(sql: str):
return select(sql) |