from fastapi import FastAPI, Form, Request
from fastapi.encoders import jsonable_encoder
from fastapi.middleware.cors import CORSMiddleware
import dataset
import pandas as pd
from starlette.responses import FileResponse, StreamingResponse
from fastapi.responses import HTMLResponse
from fastapi.staticfiles import StaticFiles
import io
from starlette.responses import RedirectResponse
import codecs
import random
from starlette.status import HTTP_302_FOUND,HTTP_303_SEE_OTHER
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=['*'],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
def get_db():
db = dataset.connect('mysql://choozmo:pAssw0rd@139.162.121.30:33306/hhh?charset=utf8mb4')
step_questions_table = db['step_questions']
return step_questions_table
@app.post("/step_questions/submit", response_class=RedirectResponse)
async def submit(request: Request):
form_data = await request.form()
# form_data = request.form()
result = {
'sex': form_data.get('q1'),
'phone': form_data.get('q2'),
'email': form_data.get('q3'),
'building_case_name': form_data.get('q4'),
'building_case_type': form_data.get('q5'),
'decoration_style': ','.join(form_data.getlist('q6')),
'decoration_budget': ','.join(form_data.getlist('q7')),
'decoration_size': ','.join(form_data.getlist('q8')),
}
print(result)
get_db().insert(result)
# fr=codecs.open('static/index_complete.html','r','utf-8')
# lines=fr.read()
# fr.close()
# response = RedirectResponse(url='/a1/index_complete.html')
# lines="""
# """
# return RedirectResponse(url='/a1/index_complete.html')
# return HTMLResponse(content=lines.format(str(random.randint(100000,999999))), status_code=200)
return RedirectResponse(url="/a1/index_complete.html", status_code=HTTP_302_FOUND)
# return result
app.mount("/a1", StaticFiles(directory="static"), name="static")
@app.get("/a1")
async def redirect():
# url = app.url_path_for("/a1/index.html")
response = RedirectResponse(url='/a1/index.html')
return response
#@app.get("/a1", response_class=HTMLResponse)
#@app.get("/a1", response_class=HTMLResponse)
#async def read_items():
# return """
#
#
# Some HTML in here
#
#
# Look ma! HTML!
#
#
# """
@app.get("/step_questions")
async def get_step_question(request: Request):
db_data = list(get_db().find())
df = pd.DataFrame(db_data, columns=db_data[0].keys())
stream = io.StringIO()
df.to_csv(stream, index = False)
# response = StreamingResponse(data_frame.to_csv(index=False), media_type="text/csv")
response = StreamingResponse(iter([stream.getvalue()]), media_type="text/csv")
response.headers["Content-Disposition"] = "attachment; filename=export.csv"
return response