-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
34 lines (27 loc) · 995 Bytes
/
main.py
File metadata and controls
34 lines (27 loc) · 995 Bytes
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
import io
import json
from generator import gen_sql
from fastapi import FastAPI,HTTPException,Query
from fastapi.responses import StreamingResponse,FileResponse
from fastapi.staticfiles import StaticFiles
app = FastAPI()
app.mount("/static", StaticFiles(directory="static"), name="static")
@app.get('/')
def root():
return FileResponse('index.html')
@app.get('/gensql/{table}')
def generate_sql(table:str,qty:int,columns:str=Query(...)):
columns = columns.strip()
columns = [field for field in columns.split(',') if field != '']
# ===========================
with open('static/types.json') as file:
available = json.load(file)
for field in columns:
if not field in available:
raise HTTPException(status_code=400,detail=f"unrecognized key:'{field}'")
# =============================
sql = gen_sql(table,qty,columns)
file = io.StringIO()
file.write(sql)
file.seek(0)
return StreamingResponse(file)