def registe(request):
if request.method == 'GET':
return render(request, 'mine/registe.html')
elif request.method == 'POST':
user = User()
user.account = request.POST.get('account')
user.password = genarate_password(request.POST.get('password'))
user.name = request.POST.get('name')
user.phone = request.POST.get('phone')
user.addr = request.POST.get('addr')
# 头像
img_name = user.account + '.png'
imagePath = os.path.join(settings.MEDIA_ROOT, img_name)
file = request.FILES.get('icon')
with open(imagePath, 'wb') as fp:
for data in file.chunks():
fp.write(data)
user.img = img_name
user.token = str(uuid.uuid5(uuid.uuid4(), 'register'))
user.save()
# 状态保持
request.session['token'] = user.token
# 重定向
return redirect('myapp:mine')
def load_or_creator_user(token):
'''根据token载入或创建用户
'''
import datetime
user = User.query.filter_by(uid = int(token.uid)).first()
if user is None:
print 'user is not created'
# create user
user_info = client.users.show.get(uid = token.uid)
user = User(token.uid, json.dumps(user_info), json.dumps(token))
user.active = True
user.create_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
user.last_login_time = user.create_time
db.session.add(user)
db.session.commit()
login_user(user)
else:
print 'user has been created'
# load user
user.token = json.dumps(token)
user_info = client.users.show.get(uid = token.uid)
user.info = json.dumps(user_info)
user.active = True
user.last_login_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
db.session.commit()
login_user(user)
def register(request):
if request.method == "GET":
return render(request, 'register.html')
elif request.method == "POST":
account = request.POST.get('account')
password = request.POST.get('password')
username = request.POST.get('nickname')
passwd = request.POST.get('passwd')
verifycode = request.POST.get('verifycode')
verifycode = verifycode.upper()
print(print("verifycode:{}".format(verifycode)))
print(print("rand_str:{}".format(rand_str)))
print(verifycode == rand_str)
if verifycode != rand_str:
return HttpResponse('验证码错误')
if (passwd == password) and password and username and (verifycode
== rand_str):
try:
user = User()
user.name = username
user.password = generate_password(password)
user.token = uuid.uuid3(uuid.uuid4(), 'username')
user.account = account
user.save()
response = redirect('paixienet:mainweb')
response.set_cookie('token', user.token)
return response
except Exception as e:
return HttpResponse('注册失败')
else:
return HttpResponse('注册失败')
