def ModifyInfo():
global conn
if request.method == "GET":
userName=request.args.get('name')
tempPsw=request.args.get('password').encode(encoding="utf-8")
userPsw=base64.encodestring(tempPsw).decode()
userEmail=request.args.get('email')
userCountry=request.args.get('country')
user=users.Users(conn,userName,userPsw,userEmail,userCountry)
userid=session.get("userid")
temp=session.get("username")
if(temp != userName):
flag=1
else:
flag=0
result=user.modifyUserInfo(userid,flag);
if result == True:
session["username"]=userName
return """<script>alert('submit successful');location.replace("/Information");</script>"""
else:
user.clean()
return """<script>alert('name exist');location.replace("/Information");</script>"""
else:
return """<script>alert('create new account error')</script>"""
def cancelfollow():
global conn
username=session.get("username")
userid=session.get("userid")
data=json.loads(request.form.get('data'))
friendid=int(data["userid"])
user=users.Users(conn)
user.cancelFollow(userid,friendid)
return ""
def Information():
global conn
username=session.get("username")
if username == None:
return login();
else:
userid=session.get("userid")
user=users.Users(conn)
informations=user.getAllInformation(userid)
userpassword=base64.decodestring(informations[0][1].encode(encoding="utf-8")).decode()
useremail=informations[0][2]
usercountry=informations[0][3]
return render_template('information.html',username=username,\
password=userpassword,email=useremail,country=usercountry)
def searchUser():
global conn
user=users.Users(conn)
data=json.loads(request.form.get('data'))
username=data["username"]
userid=session.get("userid")
allUsers=user.getUsersByName(userid,username)
if allUsers != []:
for item in allUsers:
datalist={
'userid':item[0],
'username':item[1],
'country':item[2],
'follow':item[3]
}
AllUsers=json.dumps(datalist)
return AllUsers;
else:
return "nobody"
def check():
if request.method == "POST":
UserName=request.form.get('username')
tempPsw=request.form.get('password').encode(encoding="utf-8")
PassWord=base64.encodestring(tempPsw).decode()
#PassWord=request.form.get('password')
user=users.Users(conn,UserName,PassWord)
result=user.userLogin()
if result == True:
userid=user.getUserID()
session["username"]=user.name
session["userid"]=userid
return """<script>location.replace("/mainWindow");</script>"""
else:
user.clean()
return """<script>alert('username or password error');location.replace("/login");</script>"""
else:
print("get data error")
def friends():
global conn
username=session.get("username")
if username == None:
return login();
userid=session.get("userid")
user=users.Users(conn)
allUsers=user.getUsers(userid)
user_data=[];
for item in allUsers:
datalist={
'userid':item[0],
'username':item[1],
'country':item[2],
'follow':item[3]
}
user_data.append(datalist)
AllUsers=json.dumps(user_data)
return render_template('friends.html',Musername=username,users=AllUsers)
def createUser():
if request.method == "GET":
userName=request.args.get('name')
tempPsw=request.args.get('password').encode(encoding="utf-8")
userPsw=base64.encodestring(tempPsw).decode()
userEmail=request.args.get('email')
userCountry=request.args.get('country')
global conn
user=users.Users(conn,userName,userPsw,userEmail,userCountry)
result=user.userApply()
if result == True:
session["username"]=userName
userid=user.getUserID()
session["userid"]=userid
return """<script>alert('create new account successful');location.replace("/mainWindow");</script>"""
else:
user.clean()
return """<script>alert('name exist');location.replace("/apply");</script>"""
else:
return """<script>alert('create new account error')</script>"""
:license: GOMEOPS, see LICENSE_FILE for more details.
"""
import sys
if sys.version_info[0] < 3:
reload(sys)
sys.setdefaultencoding('utf8')
from flask import (Blueprint, render_template, jsonify, flash, request,
redirect, url_for, session)
b_users = Blueprint('b_users', __name__)
from modules import users
user = users.Users()
@b_users.route('/login', methods=['POST', 'GET'])
def login():
'''登陆路由'''
if request.method == 'POST':
phone = request.form.get('phone', None)
password = request.form.get('password', None)
ok, message = user.login(phone, password)
if not ok:
flash(message, 'danger')
return render_template('login.html')
username, _ = message
session['username'] = username
