36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
from flask import request, jsonify, session, url_for, render_template
|
|
from .config import db, slideshow_images
|
|
import pymysql
|
|
|
|
|
|
def connect(mobileNo, encrypted_password):
|
|
conn = pymysql.connect(**db)
|
|
cursor = conn.cursor(pymysql.cursors.DictCursor)
|
|
args = (mobileNo, encrypted_password)
|
|
verify_sql = "SELECT Phone_number FROM Users WHERE Phone_number = %s AND `Password` = %s;"
|
|
cursor.execute(verify_sql, args)
|
|
user = cursor.fetchone()
|
|
cursor.close()
|
|
conn.close()
|
|
return user
|
|
|
|
def login():
|
|
if request.method == 'GET':
|
|
images = slideshow_images
|
|
return render_template('login.html', images=images)
|
|
|
|
if request.method == 'POST':
|
|
session.pop('user_id', None)
|
|
mobileNo = request.json.get('username')
|
|
encrypted_password = request.json.get('password')
|
|
try:
|
|
user = connect(mobileNo, encrypted_password)
|
|
if not user:
|
|
return jsonify({'message': '用户不存在，请点击注册按钮注册'}), 401
|
|
session['user_id'] = mobileNo
|
|
session.modified = True
|
|
return jsonify({'redirect': url_for('index')})
|
|
except Exception as e:
|
|
print(e)
|
|
return jsonify({'message': '数据库错误，请稍后再试'}), 500
|