コード例 #1
0
def receive(wx_appid):
    db = g._db
    rds = g.im_rds
    req = request.args
    signature = req.get('signature', '')
    timestamp = req.get('timestamp', '')
    nonce = req.get('nonce', '')
    if not check_signature(signature, timestamp, nonce):
        logging.warning("check signature fail")
        return ''

    msg_signature = req.get('msg_signature')
    if not msg_signature:
        logging.warning("msg signature is None")
        return ''

    wx_crypt = WXBizMsgCrypt(TOKEN, ENCODING_AES_KEY, APPID)
    r, xml = wx_crypt.DecryptMsg(request.data, msg_signature, timestamp, nonce)

    logging.debug("receive wx message:%s %s %s", wx_appid, r, xml)

    data = Parse.parse(xml)
    msg_type = data.get('MsgType')

    # 接收到事件
    if msg_type == 'event':
        logging.debug("event:%s", data.get('Event'))
        event = data.get('Event')
        session = data.get('SessionFrom')
        openid = data.get('FromUserName')
        gh_id = data.get('ToUserName')
        if event == 'user_enter_tempsession' and session:
            #设置来自小程序的微信用户名
            u = get_user(rds, db, gh_id, openid)
            if u:
                logging.debug("set user name:%s %s %s", gh_id, openid, session)
                WXUser.set_user_name(rds, u.appid, u.uid, session)
        return ''
    # 接收到其他消息
    else:
        openid = data.get('FromUserName')
        gh_id = data.get('ToUserName')

        # 消息格式检查
        if msg_type not in ('text', 'image', 'voice', 'location'):
            msg = Reply.text(openid, gh_id, '该消息格式不支持, 目前只支持文本,图片,语音,位置')
            _, m = wx_crypt.EncryptMsg(msg, nonce, timestamp)
            return m

        logging.debug("msg:%s %s %s", openid, gh_id, msg_type)

        u = get_user(rds, db, gh_id, openid)
        logging.debug("appid:%s uid:%s store id:%s seller id:%s", u.appid,
                      u.uid, u.store_id, u.seller_id)
        if msg_type == 'text':
            content = data.get('Content')
            logging.debug("text content:%s", content)
            obj = {"text": content}
        elif msg_type == 'location':
            x = data.get("Location_X")
            y = data.get("Location_Y")
            obj = {"location": {"latitude": x, "longitude": y}}
        elif msg_type == 'image':
            access_token = get_access_token(rds, db, u.wx_appid)
            if not access_token:
                logging.error("can't get access token")
                return ''

            mp = WXMPAPI(access_token)
            media = mp.get_media(data.get("MediaId"))
            try:
                width, height = get_image_size(media)
            except UnknownImageFormat, e:
                width = 0
                height = 0

            ext = ".jpg"
            name = md5.new(media).hexdigest()
            path = "/images/" + name + ext
            r = FS.upload(path, media)
            if not r:
                logging.error("fs upload image error")
                return ''

            url = config.IM_URL + "/images/" + name + ext
            image2 = {"url": url, "width": width, "height": height}
            obj = {"image": url, "image2": image2}
            logging.debug("image url:%s width:%s height:%s", url, width,
                          height)
        elif msg_type == 'voice':
            access_token = get_access_token(rds, db, u.wx_appid)
            if not access_token:
                logging.error("can't get access token")
                return ''

            mp = WXMPAPI(access_token)
            media = mp.get_media(data.get("MediaId"))

            md5_value = md5.new(media).hexdigest()
            path = "/audios/" + md5_value
            r = FS.upload(path, media)
            if not r:
                logging.error("fs upload audio error")
                return ''

            duration = amr_duration(media)
            url = config.IM_URL + "/audios/" + md5_value
            obj = {"audio": {"url": url, "duration": duration}}
コード例 #2
0
            duration = amr_duration(media)
            url = config.IM_URL + "/audios/" + md5_value
            obj = {"audio": {"url": url, "duration": duration}}
        else:
            logging.debug("unsupport msg type:%s", msg_type)
            obj = None

        now = int(time.time())
        if u.seller_id == 0:
            seller = get_one_supporter(db, rds, u.store_id)
            if not seller:
                logging.warning("no supporter:%d", u.store_id)
            else:
                logging.debug("got seller id:%s", seller['seller_id'])
                WXUser.set_seller_id(rds, gh_id, openid, seller['seller_id'])
                WXUser.set_seller_timestamp(rds, gh_id, openid, now)
                u.seller_id = seller['seller_id']
        elif now - u.seller_timestamp > 3600:
            sellers = Seller.get_sellers(db, u.store_id)
            if not sellers:
                raise ResponseMeta(400, 'store no supporter')

            deleted = True
            for s in sellers:
                if s['id'] == u.seller_id:
                    deleted = False
                    break

            if not deleted:
                WXUser.set_seller_timestamp(rds, gh_id, openid, now)
コード例 #3
0
ファイル: message.py プロジェクト: richmonkey/xiaowei_web
def receive(wx_appid):
    db = g._db
    rds = g.im_rds
    req = request.args
    signature = req.get('signature', '')
    timestamp = req.get('timestamp', '')
    nonce = req.get('nonce', '')
    if not check_signature(signature, timestamp, nonce):
        logging.warning("check signature fail")
        return ''

    msg_signature = req.get('msg_signature')
    if not msg_signature:
        logging.warning("msg signature is None")
        return ''

    wx_crypt = WXBizMsgCrypt(TOKEN, ENCODING_AES_KEY, APPID)
    r, xml = wx_crypt.DecryptMsg(request.data, msg_signature, timestamp, nonce)

    logging.debug("receive wx message:%s %s %s", wx_appid, r, xml)

    data = Parse.parse(xml)
    msg_type = data.get('MsgType')

    # 接收到事件
    if msg_type == 'event':
        logging.debug("event:%s", data.get('Event'))
        event = data.get('Event')
        session = data.get('SessionFrom')
        openid = data.get('FromUserName')
        gh_id = data.get('ToUserName')
        if event == 'user_enter_tempsession' and session:
            #设置来自小程序的微信用户名
            u = get_user(rds, db, gh_id, openid)
            if u:
                logging.debug("set user name:%s %s %s", gh_id, openid, session)
                WXUser.set_user_name(rds, u.appid, u.uid, session)
        return ''
    # 接收到其他消息
    else:
        openid = data.get('FromUserName')
        gh_id = data.get('ToUserName')

        # 消息格式检查
        if msg_type not in ('text', 'image', 'voice', 'location'):
            msg = Reply.text(openid, gh_id, '该消息格式不支持, 目前只支持文本,图片,语音,位置')
            _, m = wx_crypt.EncryptMsg(msg, nonce, timestamp)
            return m

        logging.debug("msg:%s %s %s", openid, gh_id, msg_type)

        u = get_user(rds, db, gh_id, openid)
        logging.debug("appid:%s uid:%s store id:%s seller id:%s", u.appid, u.uid, u.store_id, u.seller_id)
        if msg_type == 'text':
            content = data.get('Content')
            logging.debug("text content:%s", content)
            obj = {"text":content}
        elif msg_type == 'location':
            x = data.get("Location_X")
            y = data.get("Location_Y")
            obj = {"location":{"latitude":x, "longitude":y}}
        elif msg_type == 'image':
            access_token = get_access_token(rds, db, u.wx_appid)
            if not access_token:
                logging.error("can't get access token")
                return ''

            mp = WXMPAPI(access_token)
            media = mp.get_media(data.get("MediaId"))
            try:
                width, height = get_image_size(media)
            except UnknownImageFormat, e:
                width = 0
                height = 0
   
            ext = ".jpg"
            name = md5.new(media).hexdigest()
            path = "/images/" + name + ext
            r = FS.upload(path, media)
            if not r:
                logging.error("fs upload image error")
                return ''

            url = config.IM_URL + "/images/" + name + ext
            image2 = {"url":url, "width":width, "height":height}
            obj = {"image":url, "image2":image2}
            logging.debug("image url:%s width:%s height:%s", url, width, height)
        elif msg_type == 'voice':
            access_token = get_access_token(rds, db, u.wx_appid)
            if not access_token:
                logging.error("can't get access token")
                return ''

            mp = WXMPAPI(access_token)
            media = mp.get_media(data.get("MediaId"))

            md5_value = md5.new(media).hexdigest()
            path = "/audios/" + md5_value
            r = FS.upload(path, media)
            if not r:
                logging.error("fs upload audio error")
                return ''

            duration = amr_duration(media)
            url = config.IM_URL + "/audios/" + md5_value
            obj = {"audio":{"url":url, "duration":duration}}
コード例 #4
0
ファイル: message.py プロジェクト: richmonkey/xiaowei_web
            duration = amr_duration(media)
            url = config.IM_URL + "/audios/" + md5_value
            obj = {"audio":{"url":url, "duration":duration}}
        else:
            logging.debug("unsupport msg type:%s", msg_type)
            obj = None

        now = int(time.time())
        if u.seller_id == 0:
            seller = get_one_supporter(db, rds, u.store_id)
            if not seller:
                logging.warning("no supporter:%d", u.store_id)
            else:
                logging.debug("got seller id:%s", seller['seller_id'])
                WXUser.set_seller_id(rds, gh_id, openid, seller['seller_id'])
                WXUser.set_seller_timestamp(rds, gh_id, openid, now)
                u.seller_id = seller['seller_id']
        elif now - u.seller_timestamp > 3600:
            sellers = Seller.get_sellers(db, u.store_id)
            if not sellers:
                raise ResponseMeta(400, 'store no supporter')

            deleted = True
            for s in sellers:
                if s['id'] == u.seller_id:
                    deleted = False
                    break
                            
            if not deleted:
                WXUser.set_seller_timestamp(rds, gh_id, openid, now)
コード例 #5
0
def get_user(rds, db, gh_id, openid):
    now = int(time.time())
    u = WXUser.get_wx_user(rds, gh_id, openid)
    if not u:
        wx = App.get_wx_by_ghid(db, gh_id)
        if not wx:
            logging.error("invalid gh_id:%s", gh_id)
            return None

        store_id = wx['store_id']
        if not store_id:
            logging.error("can't find store id with gh_id:%s", gh_id)
            return None

        access_token = get_access_token(rds, db, wx['wx_app_id'])
        if not access_token:
            logging.error("can't get access token")
            return None

        mp = WXMPAPI(access_token)

        uid = WXUser.gen_id(rds)
        u = WXUser()
        u.gh_id = gh_id
        u.openid = openid
        u.appid = wx['appid']
        u.wx_appid = wx['wx_app_id']
        u.uid = uid
        u.store_id = store_id
        u.timestamp = now
        u.seller_id = 0
        WXUser.save_wx_user(rds, u)
        WXUser.bind_openid(rds, u.appid, u.uid, openid)
        logging.debug("bind openid:%s %s %s", u.appid, u.uid, openid)

        r = mp.get_user_by_openid(openid)
        if r.get('errcode'):
            logging.error("get user error:%s %s", r['errcode'], r['errmsg'])
        else:
            avatar = r.get('headimgurl', '')
            name = r.get('nickname', '')
            logging.debug("gh_id:%s openid:%s name:%s avatar:%s", gh_id,
                          openid, name, avatar)
            WXUser.set_user_name(rds, u.appid, u.uid, name, avatar)
    elif now - u.timestamp > 4 * 3600:
        #更新用户信息
        wx = App.get_wx_by_ghid(db, gh_id)
        if not wx:
            logging.error("invalid gh_id:%s", gh_id)
            return None

        store_id = wx.get('store_id')
        if not store_id:
            logging.error("gh_id:%s is unauth", gh_id)
            return None

        appid = wx['appid']
        #切换到其它账号或者删除之后重新授权
        if u.store_id != store_id or u.appid != appid:
            logging.debug(
                "store/app changed, gh_id:%s openid:%s store id:%s -> %s appid:%s -> %s",
                gh_id, openid, u.store_id, store_id, u.appid, appid)
            #生成新的用户id
            uid = WXUser.gen_id(rds)
            u = WXUser()
            u.gh_id = gh_id
            u.openid = openid
            u.appid = wx['appid']
            u.wx_appid = wx['wx_app_id']
            u.uid = uid
            u.store_id = store_id
            u.timestamp = now
            u.seller_id = 0
            WXUser.save_wx_user(rds, u)
            WXUser.bind_openid(rds, u.appid, u.uid, openid)
            logging.debug("bind openid:%s %s %s", u.appid, u.uid, openid)

        if wx['is_app']:
            #小程序获取不到微信的用户信息
            WXUser.set_timestamp(rds, gh_id, openid, now)
        else:
            access_token = get_access_token(rds, db, wx['wx_app_id'])
            if not access_token:
                logging.error("can't get access token")
                return None

            mp = WXMPAPI(access_token)
            r = mp.get_user_by_openid(openid)
            if r.get('errcode'):
                logging.error("get user error:%s %s", r['errcode'],
                              r['errmsg'])
            else:
                avatar = r.get('headimgurl', '')
                name = r.get('nickname', '')
                logging.debug("gh_id:%s openid:%s name:%s avatar:%s", gh_id,
                              openid, name, avatar)
                WXUser.set_user_name(rds, u.appid, u.uid, name, avatar)
                WXUser.set_timestamp(rds, gh_id, openid, now)

    return u
コード例 #6
0
ファイル: wx_token.py プロジェクト: richmonkey/xiaowei_web
def get_user(rds, db, gh_id, openid):
    now = int(time.time())
    u = WXUser.get_wx_user(rds, gh_id, openid)
    if not u:
        wx = App.get_wx_by_ghid(db, gh_id)
        if not wx:
            logging.error("invalid gh_id:%s", gh_id)
            return None

        store_id = wx['store_id']
        if not store_id:
            logging.error("can't find store id with gh_id:%s", gh_id)
            return None

        access_token = get_access_token(rds, db, wx['wx_app_id'])
        if not access_token:
            logging.error("can't get access token")
            return None

        mp = WXMPAPI(access_token)
        
        uid = WXUser.gen_id(rds)
        u = WXUser()
        u.gh_id = gh_id
        u.openid = openid
        u.appid = wx['appid']
        u.wx_appid = wx['wx_app_id']
        u.uid = uid
        u.store_id = store_id
        u.timestamp = now
        u.seller_id = 0
        WXUser.save_wx_user(rds, u)
        WXUser.bind_openid(rds, u.appid, u.uid, openid)
        logging.debug("bind openid:%s %s %s", u.appid, u.uid, openid)
        
        r = mp.get_user_by_openid(openid)
        if r.get('errcode'):
            logging.error("get user error:%s %s", 
                          r['errcode'], r['errmsg'])
        else:
            avatar = r.get('headimgurl', '')
            name = r.get('nickname', '')
            logging.debug("gh_id:%s openid:%s name:%s avatar:%s", gh_id, openid, name, avatar)
            WXUser.set_user_name(rds, u.appid, u.uid, name, avatar)
    elif now - u.timestamp > 4*3600:
        #更新用户信息
        wx = App.get_wx_by_ghid(db, gh_id)
        if not wx:
            logging.error("invalid gh_id:%s", gh_id)
            return None

        store_id = wx.get('store_id')
        if not store_id:
            logging.error("gh_id:%s is unauth", gh_id)
            return None
        
        appid = wx['appid']
        #切换到其它账号或者删除之后重新授权
        if u.store_id != store_id or u.appid != appid:
            logging.debug("store/app changed, gh_id:%s openid:%s store id:%s -> %s appid:%s -> %s",
                          gh_id, openid, u.store_id, store_id, u.appid, appid)
            #生成新的用户id
            uid = WXUser.gen_id(rds)
            u = WXUser()
            u.gh_id = gh_id
            u.openid = openid
            u.appid = wx['appid']
            u.wx_appid = wx['wx_app_id']
            u.uid = uid
            u.store_id = store_id
            u.timestamp = now
            u.seller_id = 0
            WXUser.save_wx_user(rds, u)
            WXUser.bind_openid(rds, u.appid, u.uid, openid)
            logging.debug("bind openid:%s %s %s", u.appid, u.uid, openid)

        if wx['is_app']:
            #小程序获取不到微信的用户信息
            WXUser.set_timestamp(rds, gh_id, openid, now)
        else:
            access_token = get_access_token(rds, db, wx['wx_app_id'])
            if not access_token:
                logging.error("can't get access token")
                return None
             
            mp = WXMPAPI(access_token)
            r = mp.get_user_by_openid(openid)
            if r.get('errcode'):
                logging.error("get user error:%s %s", 
                              r['errcode'], r['errmsg'])
            else:
                avatar = r.get('headimgurl', '')
                name = r.get('nickname', '')
                logging.debug("gh_id:%s openid:%s name:%s avatar:%s", gh_id, openid, name, avatar)
                WXUser.set_user_name(rds, u.appid, u.uid, name, avatar)
                WXUser.set_timestamp(rds, gh_id, openid, now)
        
    return u