python学习

web/controllers/api/Member.py

@@ -1,36 +1,38 @@
 # -*- coding: utf-8 -*-
 from web.controllers.api import route_api
-from flask import request,jsonify
-from application import app,db
+from  flask import request,jsonify,g
+from application import  app,db
 import requests,json
 from common.models.member.Member import Member
-from common.models.member.Oauth_member_bind import OauthMemberBind
+from common.models.member.OauthMemberBind import OauthMemberBind
+from common.models.food.WxShareHistory import WxShareHistory
 from common.libs.Helper import getCurrentDate
 from common.libs.member.MemberService import MemberService
 
 @route_api.route("/member/login",methods = [ "GET","POST" ])
 def login():
-    resp = {'code': 200, 'msg': '操作成功~', 'data': {}}
+    resp = { 'code':200 ,'msg':'操作成功~','data':{} }
     req = request.values
     code = req['code'] if 'code' in req else ''
-    app.logger.info(code)
-    if not code or len(code) < 1:
+    if not code or len( code ) < 1:
         resp['code'] = -1
         resp['msg'] = "需要code"
         return jsonify(resp)
 
-    openid = MemberService.getWeChatOpenId(code)
+
+    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 ''
     '''
         判断是否已经测试过，注册了直接返回一些信息
     '''
-    bind_info = OauthMemberBind.query.filter_by(openid=openid, type=1).first()
+    bind_info = OauthMemberBind.query.filter_by( openid = openid,type = 1 ).first()
     if not bind_info:
         model_member = Member()
         model_member.nickname = nickname
@@ -51,10 +53,11 @@ def login():
         db.session.commit()
 
         bind_info = model_bind
-    member_info = Member.query.filter_by(id=bind_info.member_id).first()
+
+    member_info = Member.query.filter_by(id = bind_info.member_id).first()
     token = "%s#%s" % (MemberService.geneAuthCode(member_info), member_info.id)
     resp['data'] = {'token': token}
-    return jsonify(resp)
+    return jsonify( resp )
 
 
 @route_api.route("/member/check-reg",methods = [ "GET","POST" ])
@@ -87,4 +90,30 @@ def checkReg():
 
     token = "%s#%s"%( MemberService.geneAuthCode( member_info ),member_info.id )
     resp['data'] = { 'token':token }
+    return jsonify(resp)
+
+@route_api.route("/member/share",methods = [ "POST" ])
+def memberShare():
+    resp = {'code': 200, 'msg': '操作成功~', 'data': {}}
+    req = request.values
+    url = req['url'] if 'url' in req else ''
+    member_info = g.member_info
+    model_share = WxShareHistory()
+    if member_info:
+        model_share.member_id = member_info.id
+    model_share.share_url = url
+    model_share.created_time = getCurrentDate()
+    db.session.add(model_share)
+    db.session.commit()
+    return jsonify(resp)
+
+
+@route_api.route("/member/info")
+def memberInfo():
+    resp = {'code': 200, 'msg': '操作成功~', 'data': {}}
+    member_info = g.member_info
+    resp['data']['info'] = {
+        "nickname":member_info.nickname,
+        "avatar_url":member_info.avatar
+    }
     return jsonify(resp)