python学习

web/controllers/user/User.py

@@ -1,7 +1,8 @@
 # -*- coding: utf-8 -*-
-from flask import Blueprint, render_template, request,jsonify,make_response,redirect
+from flask import Blueprint, request,jsonify,make_response,redirect,g
 from common.models.User import User
 from common.libs.user.UserService import UserService
+from common.libs.Helper import (ops_render)
 from application import app,db
 import json
 from common.libs.UrlManager import UrlManager
@@ -13,7 +14,7 @@ route_user = Blueprint('user_page', __name__)
 @route_user.route("/login")
 def login():
     if request.method == "GET":
-        return render_template("user/login.html")
+        return ops_render("user/login.html")
     resp = {'code': 200, 'msg': '登录成功', 'data': {}}
     req = request.values
     login_name = req['login_name'] if 'login_name' in req else ''
@@ -45,14 +46,81 @@ def login():
     return response
 
 
-@route_user.route("/edit")
+@route_user.route("/edit",methods=["GET", "POST"])
 def edit():
-    return render_template("user/edit.html")
+    # if request.method == "GET":
+    #     return ops_render("user/edit.html")
+
+    if request.method == "GET":
+        return ops_render("user/edit.html", {'current': 'edit'})
+
+    resp = {'code': 200, 'msg': '操作成功~', 'data': {}}
+    req = request.values
+    nickname = req['nickname'] if 'nickname' in req else ''
+    email = req['email'] if 'email' in req else ''
+
+    if nickname is None or len(nickname) < 1:
+        resp['code'] = -1
+        resp['msg'] = "请输入符合规范的姓名~~"
+        return jsonify(resp)
+
+    if email is None or len(email) < 1:
+        resp['code'] = -1
+        resp['msg'] = "请输入符合规范的邮箱~~"
+        return jsonify(resp)
+
+    user_info = g.current_user
+    user_info.nickname = nickname
+    user_info.email = email
+
+    db.session.add(user_info)
+    db.session.commit()
+    return jsonify(resp)
 
 
-@route_user.route("/reset-pwd")
+@route_user.route("/reset-pwd",methods=["GET", "POST"])
 def resetPwd():
-    return render_template("user/reset_pwd.html")
+    if request.method == "GET":
+        return ops_render("user/reset_pwd.html", {'current': 'reset-pwd'})
+
+    resp = {'code': 200, 'msg': '操作成功~', 'data': {}}
+    req = request.values
+
+    old_password = req['old_password'] if 'old_password' in req else ''
+    new_password = req['new_password'] if 'new_password' in req else ''
+
+    if old_password is None or len(old_password) < 6:
+        resp['code'] = -1
+        resp['msg'] = "请输入符合规范的原密码~~"
+        return jsonify(resp)
+
+    if new_password is None or len(new_password) < 6:
+        resp['code'] = -1
+        resp['msg'] = "请输入符合规范的新密码~~"
+        return jsonify(resp)
+
+    if old_password == new_password:
+        resp['code'] = -1
+        resp['msg'] = "请重新输入一个吧，新密码和原密码不能相同哦~~"
+        return jsonify(resp)
+
+    user_info = g.current_user
+
+    # if user_info.uid == 1:
+    #     resp['code'] = -1
+    #     resp['msg'] = "该用户是演示账号，不准修改密码和登录用户名~~"
+    #     return jsonify(resp)
+
+    user_info.login_pwd = UserService.genePwd(new_password, user_info.login_salt)
+
+    db.session.add(user_info)
+    db.session.commit()
+
+    response = make_response(json.dumps(resp))
+    response.set_cookie(app.config['AUTH_COOKIE_NAME'], '%s#%s' % (
+        UserService.geneAuthCode(user_info), user_info.uid), 60 * 60 * 24 * 120)  # 保存120天
+    return response
+
 
 @route_user.route("/logout")
 def logout():