python学习

web/controllers/api/Member.py

@@ -6,6 +6,7 @@ import requests,json
 from common.models.member.Member import Member
 from common.models.member.Oauth_member_bind import OauthMemberBind
 from common.libs.Helper import getCurrentDate
+from common.libs.member.MemberService import MemberService
 
 @route_api.route("/member/login",methods = [ "GET","POST" ])
 def login():
@@ -18,13 +19,11 @@ def login():
         resp['msg'] = "需要code"
         return jsonify(resp)
 
-
-    url = "https://api.weixin.qq.com/sns/jscode2session?appid={0}&secret={1}&js_code={2}&grant_type=authorization_code" \
-        .format(app.config['MINA_APP']['appid'], app.config['MINA_APP']['appkey'], code)
-    r = requests.get(url)
-    res = json.loads(r.text)
-    openid = res['openid']
-
+    openid = MemberService.getWeChatOpenId(code)
+    if openid is None:
+        resp['code'] = -1
+        resp['msg'] = "调用微信出错"
+        return jsonify(resp)
     nickname = req['nickName'] if 'nickName' in req else ''
     sex = req['gender'] if 'gender' in req else 0
     avatar = req['avatarUrl'] if 'avatarUrl' in req else ''
@@ -37,7 +36,7 @@ def login():
         model_member.nickname = nickname
         model_member.sex = sex
         model_member.avatar = avatar
-        model_member.salt = ''
+        model_member.salt = MemberService.geneSalt()
         model_member.updated_time = model_member.created_time = getCurrentDate()
         db.session.add(model_member)
         db.session.commit()
@@ -53,6 +52,39 @@ def login():
 
         bind_info = model_bind
     member_info = Member.query.filter_by(id=bind_info.member_id).first()
-    resp['code'] = 200
-    resp['data'] = {'nickname': member_info.nickname}
+    token = "%s#%s" % (MemberService.geneAuthCode(member_info), member_info.id)
+    resp['data'] = {'token': token}
+    return jsonify(resp)
+
+
+@route_api.route("/member/check-reg",methods = [ "GET","POST" ])
+def checkReg():
+    resp = {'code': 200, 'msg': '操作成功~', 'data': {}}
+    req = request.values
+    code = req['code'] if 'code' in req else ''
+    if not code or len(code) < 1:
+        resp['code'] = -1
+        resp['msg'] = "需要code"
+        return jsonify(resp)
+
+    openid = MemberService.getWeChatOpenId(code)
+    if openid is None:
+        resp['code'] = -1
+        resp['msg'] = "调用微信出错"
+        return jsonify(resp)
+
+    bind_info = OauthMemberBind.query.filter_by(openid=openid, type=1).first()
+    if not bind_info:
+        resp['code'] = -1
+        resp['msg'] = "未绑定"
+        return jsonify(resp)
+
+    member_info = Member.query.filter_by( id = bind_info.member_id).first()
+    if not member_info:
+        resp['code'] = -1
+        resp['msg'] = "未查询到绑定信息"
+        return jsonify(resp)
+
+    token = "%s#%s"%( MemberService.geneAuthCode( member_info ),member_info.id )
+    resp['data'] = { 'token':token }
     return jsonify(resp)