Merge branch 'main' of https://github.com/666ghj/Weibo_PublicOpinion_AnalysisSystem
Showing
2 changed files
with
52 additions
and
0 deletions
utils/errResp.py
0 → 100644
| 1 | +import time | ||
| 2 | +from flask import Blueprint, redirect, render_template, request,Flask, session | ||
| 3 | + | ||
| 4 | +from utils.query import query | ||
| 5 | +from utils.errResp import errorResponse | ||
| 6 | + | ||
| 7 | + | ||
| 8 | +ub = Blueprint('user',__name__,url_prefix='/user',template_folder='templates') | ||
| 9 | + | ||
| 10 | +@ub.route('/login',methods=['GET','POST']) | ||
| 11 | +def login(): | ||
| 12 | + if request.method == 'GET': | ||
| 13 | + return render_template('login.html') | ||
| 14 | + else: | ||
| 15 | + def filter_fn(user): | ||
| 16 | + return request.form['username'] in user and request.form['password'] in user | ||
| 17 | + users = query('select * from user', [], 'select') | ||
| 18 | + login_success = list(filter(filter_fn,users)) | ||
| 19 | + if not len(login_success):return errorResponse('账号或密码错误') | ||
| 20 | + | ||
| 21 | + session['username'] = request.form['username'] | ||
| 22 | + return redirect('/page/home') | ||
| 23 | + | ||
| 24 | +@ub.route('/register',methods=['GET','POST']) | ||
| 25 | +def register(): | ||
| 26 | + if request.method == 'GET': | ||
| 27 | + return render_template('register.html') | ||
| 28 | + else: | ||
| 29 | + if request.form['password'] != request.form['checkPassword']:return errorResponse('两次密码不符合') | ||
| 30 | + def filter_fn(user): | ||
| 31 | + return request.form['username'] in user | ||
| 32 | + | ||
| 33 | + users = query('select * from user',[],'select') | ||
| 34 | + filter_list = list(filter(filter_fn,users)) | ||
| 35 | + if len(filter_list): | ||
| 36 | + return errorResponse('该用户名已被注册') | ||
| 37 | + else: | ||
| 38 | + time_tuple = time.localtime(time.time()) | ||
| 39 | + query(''' | ||
| 40 | + insert into user(username,password,createTime) values(%s,%s,%s) | ||
| 41 | + ''',[request.form['username'],request.form['password'],str(time_tuple[0]) + '-' + str(time_tuple[1]) + '-' + str(time_tuple[2])]) | ||
| 42 | + | ||
| 43 | + return redirect('/user/login') | ||
| 44 | + | ||
| 45 | +@ub.route('/logOut') | ||
| 46 | +def logOut(): | ||
| 47 | + session.clear() | ||
| 48 | + return redirect('/user/login') |
-
Please register or login to post a comment