choozmo
/
hhh_step_question


			
							123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
							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
import smtplib
import traceback
import os
from email.mime.text import MIMEText
from email.mime.image import MIMEImage
from email.mime.multipart import MIMEMultipart

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')
    db = dataset.connect('mysql://choozmo:pAssw0rd@db.ptt.cx:3306/hhh?charset=utf8mb4')
    step_questions_table = db['step_questions']
    return step_questions_table


def mail_to_user(umail):
    print('mail_to_user')
    gmail_user = 'edm@choozmo.com'
    gmail_password='wqdsyqwvppmubitv'
    sent_from = gmail_user
    to = [umail]
    img_data=None
    with open('/tmp/coupon.png', 'rb') as f:
        img_data = f.read()

    msg = MIMEMultipart()
    msg['Subject'] = '幸福空間五萬裝修折價券'
    msg['From'] = 'edm@choozmo.com'
    msg['To'] = 'jeweiliang@gmail.com'
    print('MIMEMultipart')

    text = MIMEText("感謝您填寫問卷，敬送您五萬元裝修折價券。")
    msg.attach(text)
    image = MIMEImage(img_data, name=os.path.basename("c:/tmp/coupon.png"))
    msg.attach(image)

    try:
        print('begin...')

        server = smtplib.SMTP_SSL('smtp.gmail.com', 465)
        server.ehlo()
        server.login(gmail_user, gmail_password)
        server.sendmail(sent_from, to,  msg.as_string())
        server.close()

        print ('Email sent!')
    except:
        traceback.print_exc()
        print ('Something went wrong...')


@app.post("/step_questions/submit", response_class=RedirectResponse)
async def submit(request: Request):
    form_data = await request.form()
#    form_data = request.form()
    email=form_data.get('q3')
    result = {
        'sex': form_data.get('q1'),
        'phone': form_data.get('q2'),
        'email': email,
        '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="""<html><body onload="document.location.replace( '/a1/index_complete.html?par={0}');">
#    </body></html>"""
#    return RedirectResponse(url='/a1/index_complete.html')
#    return HTMLResponse(content=lines.format(str(random.randint(100000,999999))), status_code=200)
    mail_to_user(email)
    return RedirectResponse(url="/a1/index_complete_msg.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 """
#    <html>
#        <head>
#            <title>Some HTML in here</title>
#        </head>
#        <body>
#            <h1>Look ma! HTML!</h1>
#        </body>
#    </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