database / app.py
xcx0902's picture
Upload folder using huggingface_hub
7d0b12c verified
raw
history blame
1.04 kB
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 ""
@app.head("/")
def UptimeRobot():
return "OK"
@app.post("/sql")
def execute_sql(param: sql):
conn, cursor = Connect()
cursor.execute(param.sql)
Close(conn)
@app.get("/sql")
def execute_sql_get(sql: str):
conn, cursor = Connect()
cursor.execute(sql)
Close(conn)