import pymysql as pymysql
from flask import Flask, render_template, request, json

app = Flask(__name__)


@app.route("/", methods=["GET"])
def default():
    return render_template('login.html')


# 表单登录
@app.route("/login_form", methods=["POST", "GET"])
def login_form():
    # 获取页面传递的手机号
    mobile = request.values.get('mobile')
    # 获取页面传递的密码
    password = request.values.get('password')
    params = {
        'mobile': mobile,
        'password': password
    }
    print(json.dumps(params))
    return login_dao(mobile, password)

# json登录
@app.route("/login_json", methods=["POST", "GET"])
def login_json():
    # 获取原始表单数据
    data = request.get_json()
    mobile = data['mobile']
    password = data['password']
    # 获取页面传递的密码
    return login_dao(mobile, password)


# 数据库操作
def login_dao(mobile, password):
    # 连接数据库
    conn = pymysql.Connect(
        host='47.94.130.233',
        port=3306,
        user='root',
        passwd='123456',
        db='training',
        charset='utf8',
    )
    # print(conn)
    # 获取游标
    my_cursor = conn.cursor()
    # 准备SQL
    sql = 'select * from emp where mobile = %s and password = %s'
    # 准备插入的数据
    values = (mobile, password)
    # 执行SQL
    my_cursor.execute(sql, values)
    # 获取所有记录
    rs = my_cursor.fetchall()
    # 关闭连接
    conn.close()

    if len(rs) > 0:
        response_result = {
            'code': 200,
            'msg': '',
            'data': {
                'id': rs[0][0],
                'nickname': rs[0][3]
            }
        }
        return json.dumps(response_result, ensure_ascii=False)
    else:
        response_result = {
            'code': 10086,
            'msg': '账号或者密码错误',
            'data': {}
        }
        return json.dumps(response_result, ensure_ascii=False)


if __name__ == '__main__':
    app.run()