def get_group_list_with_group_id(self): """ 获取包含群名和群号的列表, 并存入cache, 其中gc为群号 :type group_id: str :return:list return list sample [ { "gc": 114302207, "gn": "测试群1", "owner": 484216451 }, { "gc": 125299202, "gn": "测试群2", "owner": 242917661 } ] """ try_times = 0 while try_times < 3: if self._get_group_list: response = self._get_group_list else: url = "http://qun.qq.com/cgi-bin/qun_mgr/get_group_list" data = {'bkn': self.bkn} try: response = self.client.post(url, data=data, refer='http://qun.qq.com/member.html') response = unescape_json_response(response) self._get_group_list = response logger.debug("get_group_list response: {}".format(response)) except Exception as e: logger.debug(str(e)) logger.warning("get_group_list_with_group_id API请求失败") try_times += 1 continue try: rsp_json = json.loads(response) except ValueError: try_times += 1 logger.warning("get_group_list_with_group_id fail. {}".format(try_times)) continue if rsp_json.get('ec') == 0: group_id_list = list() group_id_list.extend(rsp_json.get('join') or []) group_id_list.extend(rsp_json.get('manage') or []) group_id_list.extend(rsp_json.get('create') or []) if group_id_list: for group in group_id_list: self.group_id_list[str(group['gc'])] = group return group_id_list else: logger.warning("seems this account didn't join any group: {}".format(response)) return [] else: logger.warning("get_group_list code unknown: {}".format(response)) return None
def get_online_buddies2(self): """ 获取在线好友列表 get_online_buddies2 :return:list """ try: logger.info("RUNTIMELOG Requesting the online buddies.") online_buddies = json.loads(self.client.get( 'http://d1.web2.qq.com/channel/get_online_buddies2?vfwebqq={0}&clientid={1}&psessionid={2}&t={3}' .format( self.vfwebqq, self.client_id, self.psessionid, self.client.get_timestamp()), )) logger.debug("RESPONSE get_online_buddies2 html: " + str(online_buddies)) if online_buddies['retcode'] != 0: raise TypeError('get_online_buddies2 result error') online_buddies = online_buddies['result'] return online_buddies except: logger.warning("RUNTIMELOG get_online_buddies2 fail") return None
def get_friend_info(self, tuin): """ 获取好友详情信息 get_friend_info {"retcode":0,"result":{"face":0,"birthday":{"month":1,"year":1989,"day":30},"occupation":"","phone":"","allow":1,"college":"","uin":3964575484,"constel":1,"blood":3,"homepage":"http://blog.lovewinne.com","stat":20,"vip_info":0,"country":"中国","city":"","personal":"","nick":" 信","shengxiao":5,"email":"*****@*****.**","province":"山东","gender":"male","mobile":"158********"}} :return:dict """ uin = str(tuin) if uin not in self.friend_uin_list: logger.info("RUNTIMELOG Requesting the account info by uin: {}".format(uin)) info = json.loads(self.client.get( 'http://s.web2.qq.com/api/get_friend_info2?tuin={0}&vfwebqq={1}&clientid={2}&psessionid={3}&t={4}'.format( uin, self.vfwebqq, self.client_id, self.psessionid, self.client.get_timestamp() ) )) logger.debug("get_friend_info2 html: {}".format(str(info))) if info['retcode'] != 0: logger.warning('get_friend_info2 retcode unknown: {}'.format(info)) return None info = info['result'] info['account'] = self.uin_to_account(uin) info['longnick'] = self.get_friend_longnick(uin) self.friend_uin_list[uin] = info try: return self.friend_uin_list[uin] except: logger.warning("RUNTIMELOG get_friend_info return fail.") logger.debug("RUNTIMELOG now uin list: " + str(self.friend_uin_list[uin]))
def get_online_friends_list(self): """ 获取在线好友列表 get_online_buddies2 :return:list """ logger.info("RUNTIMELOG Requesting the online buddies.") response = self.client.get( 'http://d1.web2.qq.com/channel/get_online_buddies2?vfwebqq={0}&clientid={1}&psessionid={2}&t={3}'.format( self.vfwebqq, self.client_id, self.psessionid, self.client.get_timestamp(), ) ) # {"result":[],"retcode":0} logger.debug("RESPONSE get_online_buddies2 html:{}".format(response)) try: online_buddies = json.loads(response) except ValueError: logger.warning("get_online_buddies2 response decode as json fail.") return None if online_buddies['retcode'] != 0: logger.warning('get_online_buddies2 retcode is not 0. returning.') return None online_buddies = online_buddies['result'] return online_buddies
def get_group_name_list_mask2(self): """ 获取群列表 get_group_name_list_mask2 {u'gmarklist': [], u'gmasklist': [], u'gnamelist': [{u'code': 2676518731, u'flag': 1090520065, u'gid': 222968641(这是group_uin), u'name': u'测试'}]} :return:dict """ logger.info("RUNTIMELOG Requesting the group list.") response = self.client.post( 'http://s.web2.qq.com/api/get_group_name_list_mask2', { 'r': json.dumps( { "vfwebqq": self.vfwebqq, "hash": self._hash_digest(self._self_info['uin'], self.ptwebqq), } ) }, ) try: response = json.loads(response) except ValueError: logger.warning("RUNTIMELOG The response of group list request can't be load as json") logger.debug("RESPONSE get_group_name_list_mask2 html: " + str(response)) if response['retcode'] != 0: raise TypeError('get_online_buddies2 result error') return response['result']
def get_group_member_info_list(self, group_code): """ 获取指定群的成员信息 :group_code: int, can be "ture" of "fake" group_code {"retcode":0,"result":{"stats":[],"minfo":[{"nick":" 信","province":"山东","gender":"male","uin":3964575484,"country":"中国","city":""},{"nick":"崔震","province":"","gender":"unknown","uin":2081397472,"country":"","city":""},{"nick":"云端的猫","province":"山东","gender":"male","uin":3123065696,"country":"中国","city":"青岛"},{"nick":"要有光","province":"山东","gender":"male","uin":2609717081,"country":"中国","city":"青岛"},{"nick":"小莎机器人","province":"广东","gender":"female","uin":495456232,"country":"中国","city":"深圳"}],"ginfo":{"face":0,"memo":"http://hujj009.ys168.com\r\n0086+区(没0)+电话\r\n0086+手机\r\nhttp://john123951.xinwen365.net/qq/index.htm","class":395,"fingermemo":"","code":3943922314,"createtime":1079268574,"flag":16778241,"level":0,"name":"ぁQQぁ","gid":3931577475,"owner":3964575484,"members":[{"muin":3964575484,"mflag":192},{"muin":2081397472,"mflag":65},{"muin":3123065696,"mflag":128},{"muin":2609717081,"mflag":0},{"muin":495456232,"mflag":0}],"option":2},"cards":[{"muin":3964575484,"card":"●s.Εx2(22222)□"},{"muin":495456232,"card":"小莎机器人"}],"vipinfo":[{"vip_level":0,"u":3964575484,"is_vip":0},{"vip_level":0,"u":2081397472,"is_vip":0},{"vip_level":0,"u":3123065696,"is_vip":0},{"vip_level":0,"u":2609717081,"is_vip":0},{"vip_level":0,"u":495456232,"is_vip":0}]}} :return:dict """ if group_code == 0: return try: url = "http://s.web2.qq.com/api/get_group_info_ext2?gcode=%s&vfwebqq=%s&t=%s" % ( group_code, self.vfwebqq, int(time.time() * 100)) response = self.client.get(url) rsp_json = json.loads(response) logger.debug("get_group_member_info_list info response: {}".format(rsp_json)) retcode = rsp_json["retcode"] if retcode == 0: result = rsp_json["result"] elif retcode == 6: logger.debug("get_group_member_info_list retcode is 6, trying to get true code.") result = self.get_group_member_info_list(self.get_true_group_code(group_code)) else: logger.warning("group_code error.") return self.group_member_info[str(group_code)] = result # 缓存群成员信息, 此处会把真假group_code都加入cache return result except Exception as ex: logger.warning("RUNTIMELOG get_group_member_info_list. Error: " + str(ex)) return
def send_group_msg(self, reply_content, group_code, msg_id, fail_times=0): fix_content = str(reply_content.replace("\\", "\\\\\\\\").replace("\n", "\\\\n").replace("\t", "\\\\t")) rsp = "" try: logger.info("Starting send group message: %s" % reply_content) req_url = "http://d1.web2.qq.com/channel/send_qun_msg2" data = ( ('r', '{{"group_uin":{0}, "face":564,"content":"[\\"{4}\\",[\\"font\\",{{\\"name\\":\\"Arial\\",\\"size\\":\\"10\\",\\"style\\":[0,0,0],\\"color\\":\\"000000\\"}}]]","clientid":{1},"msg_id":{2},"psessionid":"{3}"}}'.format( group_code, self.client_id, msg_id, self.psessionid, fix_content)), ('clientid', self.client_id), ('psessionid', self.psessionid) ) rsp = self.client.post(req_url, data, SMART_QQ_REFER) rsp_json = json.loads(rsp) if 'retcode' in rsp_json and rsp_json['retcode'] not in MESSAGE_SENT: raise ValueError("RUNTIMELOG reply group chat error" + str(rsp_json['retcode'])) logger.info("RUNTIMELOG send_qun_msg: Reply '{}' successfully.".format(reply_content)) logger.debug("RESPONSE send_qun_msg: Reply response: " + str(rsp)) return rsp_json except: logger.warning("RUNTIMELOG send_qun_msg fail") if fail_times < 5: logger.warning("RUNTIMELOG send_qun_msg: Response Error.Wait for 2s and Retrying." + str(fail_times)) logger.debug("RESPONSE send_qun_msg rsp:" + str(rsp)) time.sleep(2) self.send_group_msg(reply_content, group_code, msg_id, fail_times + 1) else: logger.warning("RUNTIMELOG send_qun_msg: Response Error over 5 times.Exit.reply content:" + str(reply_content)) return False
def get_group_name_list_mask2(self): """ 获取群列表 get_group_name_list_mask2 :return:list """ logger.info("RUNTIMELOG Requesting the group list.") response = self.client.post( 'http://s.web2.qq.com/api/get_group_name_list_mask2', { 'r': json.dumps( { "vfwebqq": self.vfwebqq, "hash": self._hash_digest(self._self_info['uin'], self.ptwebqq), } ) }, ) try: response = json.loads(response) except ValueError: logger.warning("RUNTIMELOG The response of group list request can't be load as json") return logger.debug("RESPONSE get_group_name_list_mask2 html: " + str(response)) if response['retcode'] != 0: raise TypeError('get_online_buddies2 result error') return response['result']
def uin_to_account(self, tuin): """ 将uin转换成用户QQ号 :param tuin: :return:str 用户QQ号 """ uin_str = str(tuin) try: logger.info("RUNTIMELOG Requesting the account by uin: " + str(tuin)) info = json.loads( self.client.get( 'http://s.web2.qq.com/api/get_friend_uin2?tuin={0}&type=1&vfwebqq={1}&t={2}'.format( uin_str, self.vfwebqq, self.client.get_timestamp() ), SMART_QQ_REFER ) ) logger.debug("RESPONSE uin_to_account html: " + str(info)) if info['retcode'] != 0: raise TypeError('uin_to_account retcode error') info = info['result']['account'] return info except Exception: logger.exception("RUNTIMELOG uin_to_account fail") return None
def send_friend_msg(self, reply_content, uin, msg_id, fail_times=0): fix_content = str(reply_content.replace("\\", "\\\\\\\\").replace("\n", "\\\\n").replace("\t", "\\\\t")) rsp = "" try: req_url = "http://d1.web2.qq.com/channel/send_buddy_msg2" data = ( ('r', '{{"to":{0}, "face":594, "content":"[\\"{4}\\", [\\"font\\", {{\\"name\\":\\"Arial\\", \\"size\\":\\"10\\", \\"style\\":[0, 0, 0], \\"color\\":\\"000000\\"}}]]", "clientid":{1}, "msg_id":{2}, "psessionid":"{3}"}}'.format( uin, self.client_id, msg_id, self.psessionid, fix_content)), ('clientid', self.client_id), ('psessionid', self.psessionid) ) rsp = self.client.post(req_url, data, SMART_QQ_REFER) rsp_json = json.loads(rsp) if 'errCode' in rsp_json and rsp_json['errCode'] != 0: raise ValueError("reply pmchat error" + str(rsp_json['retcode'])) logger.info("RUNTIMELOG Reply successfully.") logger.debug("RESPONSE Reply response: " + str(rsp)) return rsp_json except: if fail_times < 5: logger.warning("RUNTIMELOG Response Error.Wait for 2s and Retrying." + str(fail_times)) logger.debug("RESPONSE " + str(rsp)) time.sleep(2) self.send_friend_msg(reply_content, uin, msg_id, fail_times + 1) else: logger.warning("RUNTIMELOG Response Error over 5 times.Exit.reply content:" + str(reply_content)) return False
def lady(msg, bot): """ :type bot: smart_qq_bot.bot.QQBot :type msg: smart_qq_bot.messages.GroupMsg """ msg_id = randint(1, 10000) global _ladyWrapper logger.debug('sender:{0} QQ Number:{1}'.format(msg.src_sender_name, msg.src_sender_id)) logger.info(_ladyWrapper.GroupMap) # 发送一条群消息 if isinstance(msg, GroupMsg): now = time.mktime(datetime.datetime.now().timetuple()) if msg.src_sender_id == _ladyWrapper.observe_target: if msg.src_group_id in _ladyWrapper.GroupMap: if now > _ladyWrapper.GroupMap[ msg.src_group_id] + _ladyWrapper.interval: _ladyWrapper.GroupMap[msg.src_group_id] = now bot.send_group_msg("少妇肛肛", msg.from_uin, msg_id) else: logger.info("too frequent... Can't reply until {0}".format( _ladyWrapper.GroupMap[msg.src_group_id] + _ladyWrapper.interval)) else: bot.send_group_msg("少妇肛肛", msg.from_uin, msg_id) _ladyWrapper.GroupMap[msg.src_group_id] = now
def get_discuss_info(self, did): """ 获取指定讨论组的成员信息 :did: str {u'result': {u'info': {u'did': 2966596468, u'discu_name': u'', u'mem_list': [{u'ruin': 466331599, u'mem_uin': 466331599}, {u'ruin': 493658515, u'mem_uin': 556813270}, {u'ruin': 824566900, u'mem_uin': 2606746705}]}, u'mem_status': [], u'mem_info': [{u'nick': u'\\u54a6', u'uin': 466331599}, {u'nick': u'Auro', u'uin': 556813270}, {u'nick': u'-', u'uin': 2606746705}]}, u'retcode': 0} :rtype: dict """ if did == 0: return try: did = str(did) url = "http://d1.web2.qq.com/channel/get_discu_info?did={did}&psessionid={psessionid}&vfwebqq={vfwebqq}&clientid={clientid}&t={t}".format( did=did, psessionid=self.psessionid, vfwebqq=self.vfwebqq, clientid=self.client_id, t=int(time.time() * 100) ) response = self.client.get(url) rsp_json = json.loads(response) logger.debug("get_discuss_info response: {}".format(rsp_json)) retcode = rsp_json["retcode"] if retcode == 0: result = rsp_json["result"] else: logger.warning("get_discuss_info error.") return self.discuss_info[str(did)] = result # 缓存群成员信息, 此处会把真假group_code都加入cache return result except Exception as ex: logger.warning("get_discuss_info error: " + str(ex)) return
def send_friend_msg(self, reply_content, uin, msg_id, fail_times=0): fix_content = str(reply_content.replace("\\", "\\\\\\\\").replace("\n", "\\\\n").replace("\t", "\\\\t")) rsp = "" try: req_url = "http://d1.web2.qq.com/channel/send_buddy_msg2" data = { 'r': '{{"to":{0}, "face":594, "content":"[\\"{4}\\", [\\"font\\", {{\\"name\\":\\"Arial\\", \\"size\\":\\"10\\", \\"style\\":[0, 0, 0], \\"color\\":\\"000000\\"}}]]", "clientid":{1}, "msg_id":{2}, "psessionid":"{3}"}}'.format( uin, self.client_id, msg_id, self.psessionid, fix_content), 'clientid': self.client_id, 'psessionid': self.psessionid } rsp = self.client.post(req_url, data, SMART_QQ_REFER) rsp_json = json.loads(rsp) if 'errCode' in rsp_json and rsp_json['errCode'] != 0: raise ValueError("reply pmchat error" + str(rsp_json['retcode'])) logger.info("RUNTIMELOG Reply successfully.") logger.debug("RESPONSE Reply response: " + str(rsp)) return rsp_json except: if fail_times < 5: logger.warning("RUNTIMELOG Response Error.Wait for 2s and Retrying." + str(fail_times)) logger.debug("RESPONSE " + str(rsp)) time.sleep(2) self.send_friend_msg(reply_content, uin, msg_id, fail_times + 1) else: logger.warning("RUNTIMELOG Response Error over 5 times.Exit.reply content:" + str(reply_content)) return False
def get_group_member_info_list(self, group_code): """ 获取指定群的成员信息 :group_code: int, can be "ture" of "fake" group_code :return:dict """ if group_code == 0: logger.debug("get_group_member_info_list 输入为0,返回 None") return try: def retcode_validator(rsp_json): assert rsp_json["retcode"] == 0 or rsp_json["retcode"] == 6, "group_code error." url = "http://s.web2.qq.com/api/get_group_info_ext2?gcode=%s&vfwebqq=%s&t=%s" % ( group_code, self.vfwebqq, int(time.time() * 100)) rsp_json = self.client.load(url, validator=retcode_validator) retcode = rsp_json["retcode"] if retcode == 0: result = rsp_json["result"] elif retcode == 6: logger.debug("get_group_member_info_list retcode is 6, trying to get true code.") result = self.get_group_member_info_list(self.get_true_group_code(group_code)) self.group_member_info[str(group_code)] = result # 缓存群成员信息, 此处会把真假group_code都加入cache return result except Exception as ex: logger.warning("get_group_member_info_list. Error: " + str(ex)) return
def get_group_member_info_list(self, group_code): """ 获取指定群的成员信息 :group_code: int, can be "ture" of "fake" group_code {"retcode":0,"result":{"stats":[],"minfo":[{"nick":" 信","province":"山东","gender":"male","uin":3964575484,"country":"中国","city":""},{"nick":"崔震","province":"","gender":"unknown","uin":2081397472,"country":"","city":""},{"nick":"云端的猫","province":"山东","gender":"male","uin":3123065696,"country":"中国","city":"青岛"},{"nick":"要有光","province":"山东","gender":"male","uin":2609717081,"country":"中国","city":"青岛"},{"nick":"小莎机器人","province":"广东","gender":"female","uin":495456232,"country":"中国","city":"深圳"}],"ginfo":{"face":0,"memo":"http://hujj009.ys168.com\r\n0086+区(没0)+电话\r\n0086+手机\r\nhttp://john123951.xinwen365.net/qq/index.htm","class":395,"fingermemo":"","code":3943922314,"createtime":1079268574,"flag":16778241,"level":0,"name":"ぁQQぁ","gid":3931577475,"owner":3964575484,"members":[{"muin":3964575484,"mflag":192},{"muin":2081397472,"mflag":65},{"muin":3123065696,"mflag":128},{"muin":2609717081,"mflag":0},{"muin":495456232,"mflag":0}],"option":2},"cards":[{"muin":3964575484,"card":"●s.Εx2(22222)□"},{"muin":495456232,"card":"小莎机器人"}],"vipinfo":[{"vip_level":0,"u":3964575484,"is_vip":0},{"vip_level":0,"u":2081397472,"is_vip":0},{"vip_level":0,"u":3123065696,"is_vip":0},{"vip_level":0,"u":2609717081,"is_vip":0},{"vip_level":0,"u":495456232,"is_vip":0}]}} :return:dict """ if group_code == 0: return try: url = "http://s.web2.qq.com/api/get_group_info_ext2?gcode=%s&vfwebqq=%s&t=%s" % ( group_code, self.vfwebqq, int(time.time() * 100)) response = self.client.get(url) rsp_json = json.loads(response) logger.debug("get_group_member_info_list info response: {}".format( rsp_json)) retcode = rsp_json["retcode"] if retcode == 0: result = rsp_json["result"] elif retcode == 6: logger.debug( "get_group_member_info_list retcode is 6, trying to get true code." ) result = self.get_group_member_info_list( self.get_true_group_code(group_code)) else: logger.warning("group_code error.") return self.group_member_info[str( group_code)] = result # 缓存群成员信息, 此处会把真假group_code都加入cache return result except Exception as ex: logger.warning("RUNTIMELOG get_group_member_info_list. Error: " + str(ex)) return
def send_discuss_msg(self, reply_content, did, msg_id, fail_times=0): fix_content = str(reply_content.replace("\\", "\\\\\\\\").replace("\n", "\\\\n").replace("\t", "\\\\t")) rsp = "" try: logger.info("Starting send discuss group message: %s" % reply_content) req_url = "http://d1.web2.qq.com/channel/send_discu_msg2" data = { 'r': '{{"did":{0}, "face":564,"content":"[\\"{4}\\",[\\"font\\",{{\\"name\\":\\"Arial\\",\\"size\\":\\"10\\",\\"style\\":[0,0,0],\\"color\\":\\"000000\\"}}]]","clientid":{1},"msg_id":{2},"psessionid":"{3}"}}'.format( did, self.client_id, msg_id, self.psessionid, fix_content), 'clientid': self.client_id, 'psessionid': self.psessionid } rsp = self.client.post(req_url, data, SMART_QQ_REFER) rsp_json = json.loads(rsp) if 'retcode' in rsp_json and rsp_json['retcode'] not in MESSAGE_SENT: raise ValueError("RUNTIMELOG reply discuss group error" + str(rsp_json['retcode'])) logger.info("send_discuss_msg: Reply '{}' successfully.".format(reply_content)) logger.debug("send_discuss_msg: Reply response: " + str(rsp)) return rsp_json except: logger.warning("send_discuss_msg fail") if fail_times < 5: logger.warning("send_discuss_msg: Response Error.Wait for 2s and Retrying." + str(fail_times)) logger.debug("send_discuss_msg response:" + str(rsp)) time.sleep(2) self.send_group_msg(reply_content, did, msg_id, fail_times + 1) else: logger.warning("RUNTIMELOG send_qun_msg: Response Error over 5 times.Exit.reply content:" + str(reply_content)) return False
def get_online_friends_list(self): """ 获取在线好友列表 get_online_buddies2 :return:list """ retry_times = 10 while retry_times: logger.info("RUNTIMELOG Requesting the online buddies.") response = self.client.get( 'http://d1.web2.qq.com/channel/get_online_buddies2?vfwebqq={0}&clientid={1}&psessionid={2}&t={3}' .format( self.vfwebqq, self.client_id, self.psessionid, self.client.get_timestamp(), )) # {"result":[],"retcode":0} logger.debug( "RESPONSE get_online_buddies2 html:{}".format(response)) try: online_buddies = json.loads(response) except ValueError: logger.warning( "get_online_buddies2 response decode as json fail.") return None if online_buddies['retcode'] != 0: logger.warning( 'get_online_buddies2 retcode is not 0. returning.') return None online_buddies = online_buddies['result'] return online_buddies
def get_self_info(self): """ 获取自己的信息, 并存入self._self_info get_self_info2 {"retcode":0,"result":{"birthday":{"month":1,"year":1989,"day":30},"face":555,"phone":"","occupation":"","allow":1,"college":"","uin":2609717081,"blood":0,"constel":1,"lnick":"","vfwebqq":"68b5ff5e862ac589de4fc69ee58f3a5a9709180367cba3122a7d5194cfd43781ada3ac814868b474","homepage":"","vip_info":0,"city":"青岛","country":"中国","personal":"","shengxiao":5,"nick":"要有光","email":"","province":"山东","account":2609717081,"gender":"male","mobile":""}} :return:dict """ try_times = 0 while len(self._self_info) is 0: url = "http://s.web2.qq.com/api/get_self_info2?t={}".format( time.time()) response = self.client.get(url) logger.debug("get_self_info2 response:{}".format(response)) rsp_json = json.loads(response) if rsp_json["retcode"] != 0: try_times += 1 logger.warning("get_self_info2 fail. {}".format(try_times)) if try_times >= 5: return {} continue try: self._self_info = rsp_json["result"] except KeyError: logger.warning("get_self_info2 failed. Retrying.") continue return self._self_info
def get_friend_info(self, tuin): """ 获取好友详情信息 get_friend_info {"retcode":0,"result":{"face":0,"birthday":{"month":1,"year":1989,"day":30},"occupation":"","phone":"","allow":1,"college":"","uin":3964575484,"constel":1,"blood":3,"homepage":"http://blog.lovewinne.com","stat":20,"vip_info":0,"country":"中国","city":"","personal":"","nick":" 信","shengxiao":5,"email":"*****@*****.**","province":"山东","gender":"male","mobile":"158********"}} :return:dict """ uin = str(tuin) if uin not in self.friend_uin_list: logger.info( "RUNTIMELOG Requesting the account info by uin: {}".format( uin)) info = json.loads( self.client.get( 'http://s.web2.qq.com/api/get_friend_info2?tuin={0}&vfwebqq={1}&clientid={2}&psessionid={3}&t={4}' .format(uin, self.vfwebqq, self.client_id, self.psessionid, self.client.get_timestamp()))) logger.debug("get_friend_info2 html: {}".format(str(info))) if info['retcode'] != 0: logger.warning( 'get_friend_info2 retcode unknown: {}'.format(info)) return None info = info['result'] info['account'] = self.uin_to_account(uin) info['longnick'] = self.get_friend_longnick(uin) self.friend_uin_list[uin] = info try: return self.friend_uin_list[uin] except: logger.warning("RUNTIMELOG get_friend_info return fail.") logger.debug("RUNTIMELOG now uin list: " + str(self.friend_uin_list[uin]))
def get_friend_info2(self, tuin): """ 获取好友详情信息 get_friend_info2 {"retcode":0,"result":{"face":0,"birthday":{"month":1,"year":1989,"day":30},"occupation":"","phone":"","allow":1,"college":"","uin":3964575484,"constel":1,"blood":3,"homepage":"http://blog.lovewinne.com","stat":20,"vip_info":0,"country":"中国","city":"","personal":"","nick":" 信","shengxiao":5,"email":"*****@*****.**","province":"山东","gender":"male","mobile":"158********"}} :return:dict """ uin_str = str(tuin) try: logger.info("RUNTIMELOG Requesting the account info by uin: " + str(tuin)) info = json.loads( self.client.get( 'http://s.web2.qq.com/api/get_friend_info2?tuin={0}&vfwebqq={1}&clientid={2}&psessionid={3}&t={4}' .format(uin_str, self.vfwebqq, self.client_id, self.psessionid, self.client.get_timestamp()), )) logger.debug("RESPONSE get_friend_info2 html: " + str(info)) if info['retcode'] != 0: raise TypeError('get_friend_info2 result error') info = info['result'] return info except: logger.warning("RUNTIMELOG get_friend_info2 fail") return None
def get_discuss_info(self, did): """ 获取指定讨论组的成员信息 :did: str {u'result': {u'info': {u'did': 2966596468, u'discu_name': u'', u'mem_list': [{u'ruin': 466331599, u'mem_uin': 466331599}, {u'ruin': 493658515, u'mem_uin': 556813270}, {u'ruin': 824566900, u'mem_uin': 2606746705}]}, u'mem_status': [], u'mem_info': [{u'nick': u'\\u54a6', u'uin': 466331599}, {u'nick': u'Auro', u'uin': 556813270}, {u'nick': u'-', u'uin': 2606746705}]}, u'retcode': 0} :rtype: dict """ if did == 0: return try: did = str(did) url = "http://d1.web2.qq.com/channel/get_discu_info?did={did}&psessionid={psessionid}&vfwebqq={vfwebqq}&clientid={clientid}&t={t}".format( did=did, psessionid=self.psessionid, vfwebqq=self.vfwebqq, clientid=self.client_id, t=int(time.time() * 100)) response = self.client.get(url) rsp_json = json.loads(response) logger.debug("get_discuss_info response: {}".format(rsp_json)) retcode = rsp_json["retcode"] if retcode == 0: result = rsp_json["result"] else: logger.warning("get_discuss_info error.") return self.discuss_info[str( did)] = result # 缓存群成员信息, 此处会把真假group_code都加入cache return result except Exception as ex: logger.warning("get_discuss_info error: " + str(ex)) return
def get_friend_info2(self, tuin): """ 获取好友详情信息 get_friend_info2 {"retcode":0,"result":{"face":0,"birthday":{"month":1,"year":1989,"day":30},"occupation":"","phone":"","allow":1,"college":"","uin":3964575484,"constel":1,"blood":3,"homepage":"http://blog.lovewinne.com","stat":20,"vip_info":0,"country":"中国","city":"","personal":"","nick":" 信","shengxiao":5,"email":"*****@*****.**","province":"山东","gender":"male","mobile":"158********"}} :return:dict """ uin_str = str(tuin) try: logger.info("RUNTIMELOG Requesting the account info by uin: " + str(tuin)) info = json.loads(self.client.get( 'http://s.web2.qq.com/api/get_friend_info2?tuin={0}&vfwebqq={1}&clientid={2}&psessionid={3}&t={4}' .format( uin_str, self.vfwebqq, self.client_id, self.psessionid, self.client.get_timestamp()), )) logger.debug("RESPONSE get_friend_info2 html: " + str(info)) if info['retcode'] != 0: raise TypeError('get_friend_info2 result error') info = info['result'] return info except: logger.warning("RUNTIMELOG get_friend_info2 fail") return None
def uin_to_account(self, tuin): """ 将uin转换成用户QQ号 :param tuin: :return:str 用户QQ号 """ uin_str = str(tuin) if sql.fetch_one('select account_id from account_data where account_code={0};'.format(uin_str)): return sql.fetch_one('select account_id from account_data where account_code={0};'.format(uin_str))[0]; else: try: logger.debug("[UIN_TO_ACCOUNT] " + str(tuin)) info = json.loads( self.client.get( 'http://s.web2.qq.com/api/get_friend_uin2?tuin={0}&type=1&vfwebqq={1}&t={2}'.format( uin_str, self.vfwebqq, self.client.get_timestamp() ), SMART_QQ_REFER ) ) logger.debug("RESPONSE uin_to_account html: " + str(info)) if info['retcode'] != 0: raise TypeError('uin_to_account retcode error') info = info['result']['account'] if sql.fetch_one('select * from account_data where account_id={0};'.format(info)): sql.execute('update account_data set account_code={0} where account_id={1};'.format(uin_str,info)) else: sql.execute("insert into account_data(account_id,account_code) values('{0}','{1}');".format(info,uin_str)) return info except Exception: logger.exception("RUNTIMELOG uin_to_account fail") return None
def get_self_info(self): """ 获取自己的信息, 并存入self._self_info get_self_info2 {"retcode":0,"result":{"birthday":{"month":1,"year":1989,"day":30},"face":555,"phone":"","occupation":"","allow":1,"college":"","uin":2609717081,"blood":0,"constel":1,"lnick":"","vfwebqq":"68b5ff5e862ac589de4fc69ee58f3a5a9709180367cba3122a7d5194cfd43781ada3ac814868b474","homepage":"","vip_info":0,"city":"青岛","country":"中国","personal":"","shengxiao":5,"nick":"要有光","email":"","province":"山东","account":2609717081,"gender":"male","mobile":""}} :return:dict """ try_times = 0 while len(self._self_info) is 0: url = "http://s.web2.qq.com/api/get_self_info2?t={}".format(time.time()) response = self.client.get(url) logger.debug("get_self_info2 response:{}".format(response)) rsp_json = json.loads(response) if rsp_json["retcode"] != 0: try_times += 1 logger.warning("get_self_info2 fail. {}".format(try_times)) if try_times >= 5: return {} try: self._self_info = rsp_json["result"] except KeyError: logger.warning("get_self_info2 failed. Retrying.") continue return self._self_info
def update_help_data(): for p in plugin_list: if p!=plugin_name: try: tmp=(__import__('smart_qq_plugins.'+p,fromlist=['HELP'])) logger.debug(tmp.HELP) sql.execute("insert into Help(plugin_name,help) values('{0}','{1}');".format(p,json.dumps(tmp.HELP))) except Exception as e: logger.debug(e)
def get(self, url, refer=None): try: req = requests.get(url,headers={"Referer":refer or SMART_QQ_REFER}) tmp_req = self._opener.get(url,headers={"Referer":refer or SMART_QQ_REFER}) self._cookie.save(COOKIE_FILE, ignore_discard=True,ignore_expires=True) #logger.debug(tmp_req.text) return tmp_req.text except requests.HTTPError as e: logger.debug(e) return e
def query_friends_accounts(self): try: rsp = self.client.post( 'http://qun.qq.com/cgi-bin/qun_mgr/get_friend_list', data={'bkn': self.bkn}, refer='http://qun.qq.com/member.html', ) logger.debug("get_friend_list html:\t{}".format(str(rsp))) qq_list = json.loads(rsp).get('result', {}).get('0', {}).get('mems') rsp = self.client.post( 'http://s.web2.qq.com/api/get_user_friends2', { 'r': json.dumps({ "vfwebqq": self.vfwebqq, "hash": self._hash_digest(self._self_info['uin'], self.ptwebqq), }) }, ) logger.debug("get_user_friends2 html:\t{}".format(str(rsp))) rsp = json.loads(rsp) uin_list = [[str(friend['nick']), str(friend['uin'])] for friend in rsp['result']['info']] for friend in rsp['result'].get('marknames', []): for idx, (nick, uin) in enumerate(uin_list): if str(uin) == str(friend['uin']): uin_list[idx][0] = friend['markname'] result_dict = {} duplicated_name = set() for friend in qq_list: for tgt in uin_list: if friend['name'] == tgt[0]: if str(tgt[1]) not in result_dict: result_dict[str(tgt[1])] = str( friend['uin']) # 这个uin是真实qq号 else: duplicated_name.add(tgt[0]) result_dict[str(tgt[1])] = "" if len(duplicated_name) != 0: logger.warning( "存在多个好友使用以下昵称,无法唯一确定这些好友的真实QQ号,请通过修改备注名以唯一确定:{}".format( " ".join(list(duplicated_name)))) for uin, account in result_dict.items(): self.friend_uin_list[uin] = {'account': account} except Exception as e: logger.warning("获取好友真实qq号失败, {}".format(e))
def get_friend_info(self, tuin): uin_str = str(tuin) if uin_str not in self.friend_list: info = self.get_friend_info2(tuin) or {'nick': '群用户'} info['account'] = self.uin_to_account(tuin) self.friend_list[uin_str] = info try: return '【{0}({1})】'.format(self.friend_list[uin_str]['nick'], self.friend_list[uin_str]['account']) except: logger.warning("RUNTIMELOG get_friend_info return fail.") logger.debug("RUNTIMELOG now uin list: " + str(self.friend_list[uin_str]))
def get_group_name_list_mask2(self): """ 获取群列表, 并存入cache, 其中code为group_code get_group_name_list_mask2 :return:list { u 'gmarklist': [], u 'gmasklist': [], u 'gnamelist': [ { u 'code': 1131597161, # 这是真实group_code u 'flag': 184550417, u 'gid': 1802239929, # 这是msg.group_code, 即假group_code u 'name': u '测试' }, { u 'code': 1131597161, u 'flag': 184550417, u 'gid': 1802239929, u 'name': u '测试' } ] } """ logger.info("RUNTIMELOG Requesting the group list.") response = self.client.post( 'http://s.web2.qq.com/api/get_group_name_list_mask2', { 'r': json.dumps({ "vfwebqq": self.vfwebqq, "hash": self._hash_digest(self._self_info['uin'], self.ptwebqq), }) }, ) try: response = json.loads(response) except ValueError: logger.warning( "RUNTIMELOG The response of group list request can't be load as json" ) return logger.debug("RESPONSE get_group_name_list_mask2 html: " + str(response)) if response['retcode'] != 0: raise TypeError('get_online_buddies2 result error') for group in response['result']['gnamelist']: self.group_code_list[str(group['gid'])] = group return response['result']
def get_true_group_code(self, fake_group_code): """ :type fake_group_code: int """ fake_group_code = str(fake_group_code) logger.debug("正在查询group_code:{}对应的真实group_code".format(fake_group_code)) if fake_group_code not in self.group_code_list: logger.info("尝试更新群列表信息") self.get_group_name_list_mask2() # 先尝试更新群列表 if fake_group_code not in self.group_code_list: logger.warning("没有所查询的group_code, 请检查group_code是否错误") return 0 return self.group_code_list[fake_group_code]['code']
def get_true_group_code(self, fake_group_code): """ :type fake_group_code: int """ fake_group_code = str(fake_group_code) logger.debug( "正在查询group_code:{}对应的真实group_code".format(fake_group_code)) if fake_group_code not in self.group_code_list: logger.info("尝试更新群列表信息") self.get_group_name_list_mask2() # 先尝试更新群列表 if fake_group_code not in self.group_code_list: logger.warning("没有所查询的group_code, 请检查group_code是否错误") return 0 return self.group_code_list[fake_group_code]['code']
def get_group_list_with_group_code(self): """ 获取包含群名和group_code的列表, 并存入cache, 其中code为group_code :type group_code: str :return:list [ { u 'code': 1131597161, # 这是真实group_code u 'flag': 184550417, u 'gid': 1802239929, # 这是msg.group_code, 即假group_code u 'name': u '测试' }, { u 'code': 1131597161, u 'flag': 184550417, u 'gid': 1802239929, u 'name': u '测试' } ] """ logger.info("Requesting the group list.") response = self.client.post( 'http://s.web2.qq.com/api/get_group_name_list_mask2', { 'r': json.dumps({ "vfwebqq": self.vfwebqq, "hash": self._hash_digest(self._self_info['uin'], self.ptwebqq), }) }, ) try: logger.debug("get_group_name_list_mask2 html:\t{}".format( str(response))) response = json.loads(response) except ValueError: logger.warning( "RUNTIMELOG The response of group list request can't be load as json" ) return if response['retcode'] != 0: raise TypeError('get_group_list_with_group_code result error') for group in response['result']['gnamelist']: self.group_code_list[str(group['gid'])] = group self.group_code_list[str(group['code'])] = group return response['result']['gnamelist']
def _get_qr_login_status(self, qr_validation_url, appid, star_time, mibao_css, js_ver, sign, init_url): redirect_url = None login_result = self.client.get( qr_validation_url.format( appid, date_to_millis(datetime.datetime.utcnow()) - star_time, mibao_css, js_ver, sign), init_url) ret_code = int(find_first_result(login_result, r"\d+", None)) redirect_info = re.findall(r"(http.*?)\'", login_result) if redirect_info: logger.debug("redirect_info match is: %s" % redirect_info) redirect_url = redirect_info[0] return ret_code, redirect_url
def check_msg(self): # Pooling the message try: ret = self.client.load( 'http://d1.web2.qq.com/channel/poll2', { 'r': json.dumps( { "ptwebqq": self.ptwebqq, "clientid": self.client_id, "psessionid": self.psessionid, "key": "" } ) }, SMART_QQ_REFER, parser=lambda resp: json.loads(resp.replace(r"\u0026lt;", "<").replace(r"\u0026gt;", ">")) ) except Exception as e: logger.exception(e) return ret_code = ret['retcode'] if ret_code in (0, 116, 1202): self._last_pool_success = True if ret_code == 0: if 'result' not in ret or len(ret['result']) == 0: logger.info("Pooling ends, no new message received.") else: return ret['result'] elif ret_code == 116: self.ptwebqq = ret['p'] logger.debug("ptwebqq updated in this pooling") else: self._last_pool_success = False if ret_code in (103, ): raise NeedRelogin("Pooling received retcode: " + str(ret_code)) elif ret_code in (121,): logger.warning("Pooling error with retcode %s" % ret_code) elif ret_code == 100006: logger.error("Pooling request error, response is: %s" % ret) elif ret_code in (100001, 100012): raise NeedRelogin("Login is expired. Please relogin by qrcode") else: logger.warning("Pooling returns unknown retcode %s" % ret_code) time.sleep(2) return None
def send_sess_msg2_fromGroup(self, reply_content, guin, tuin, msg_id, service_type=0, fail_times=0): group_sig = self._get_group_sig(guin, tuin, service_type) fix_content = str( reply_content.replace("\\", "\\\\\\\\").replace("\n", "\\\\n").replace( "\t", "\\\\t")) rsp = "" try: req_url = "http://d1.web2.qq.com/channel/send_sess_msg2" data = (( 'r', '{{"to":{0}, "face":594, "content":"[\\"{4}\\", [\\"font\\", {{\\"name\\":\\"Arial\\", \\"size\\":\\"10\\", \\"style\\":[0, 0, 0], \\"color\\":\\"000000\\"}}]]", "clientid":{1}, "msg_id":{2}, "psessionid":"{3}", "group_sig":"{5}", "service_type":{6}}}' .format(tuin, self.client_id, msg_id, self.psessionid, fix_content, group_sig, service_type)), ('clientid', self.client_id), ('psessionid', self.psessionid), ('group_sig', group_sig), ('service_type', service_type)) rsp = self.client.post(req_url, data, SMART_QQ_REFER) rsp_json = json.loads(rsp) if 'retcode' in rsp_json and rsp_json['retcode'] != 0: raise ValueError("RUNTIMELOG reply sess chat error" + str(rsp_json['retcode'])) logger.info( "RUNTIMELOG send_sess_msg2_fromGroup: Reply successfully.") logger.debug( "RESPONSE send_sess_msg2_fromGroup: Reply response: " + str(rsp)) return rsp_json except: if fail_times < 5: logger.warning( "RUNTIMELOG send_sess_msg2_fromGroup: Response Error.Wait for 2s and Retrying." + str(fail_times)) logger.debug("RESPONSE " + str(rsp)) time.sleep(2) self.send_sess_msg2_fromGroup(guin, tuin, reply_content, msg_id, service_type, fail_times + 1) else: logger.warning( "RUNTIMELOG send_sess_msg2_fromGroup: Response Error over 5 times.Exit.reply content:" + str(reply_content)) return False
def get_true_group_code(self, fake_group_code): """ 通过假group_code获取真group_code :type fake_group_code: str :return str """ fake_group_code = str(fake_group_code) logger.debug("正在查询group_code:{}对应的真实group_code".format(fake_group_code)) if fake_group_code not in self.group_code_list: logger.info("尝试更新群列表信息") self.get_group_list_with_group_code() # 先尝试更新群列表 if fake_group_code not in self.group_code_list: logger.warning("没有所查询的group_code, 请检查group_code是否错误") return 0 return str(self.group_code_list[fake_group_code]['code'])
def get_group_name_list_mask2(self): """ 获取群列表, 并存入cache, 其中code为group_code get_group_name_list_mask2 :return:list { u 'gmarklist': [], u 'gmasklist': [], u 'gnamelist': [ { u 'code': 1131597161, # 这是真实group_code u 'flag': 184550417, u 'gid': 1802239929, # 这是msg.group_code, 即假group_code u 'name': u '测试' }, { u 'code': 1131597161, u 'flag': 184550417, u 'gid': 1802239929, u 'name': u '测试' } ] } """ logger.info("RUNTIMELOG Requesting the group list.") response = self.client.post( 'http://s.web2.qq.com/api/get_group_name_list_mask2', { 'r': json.dumps( { "vfwebqq": self.vfwebqq, "hash": self._hash_digest(self._self_info['uin'], self.ptwebqq), } ) }, ) try: response = json.loads(response) except ValueError: logger.warning("RUNTIMELOG The response of group list request can't be load as json") return logger.debug("RESPONSE get_group_name_list_mask2 html: " + str(response)) if response['retcode'] != 0: raise TypeError('get_online_buddies2 result error') for group in response['result']['gnamelist']: self.group_code_list[str(group['gid'])] = group return response['result']
def get_group_list_with_group_id(self): """ 获取包含群名和群号的列表, 并存入cache, 其中gc为群号 :type group_id: str :return:list return list sample [ { "gc": 114302207, "gn": "测试群1", "owner": 484216451 }, { "gc": 125299202, "gn": "测试群2", "owner": 242917661 } ] """ if self._get_group_list: response = self._get_group_list else: url = "http://qun.qq.com/cgi-bin/qun_mgr/get_group_list" data = {'bkn': self.bkn} response = self.client.post(url, data=data, refer='http://qun.qq.com/member.html') self._get_group_list = response logger.debug("get_group_list response: {}".format(response)) rsp_json = json.loads(response) if rsp_json['ec'] == 0: group_id_list = list() group_id_list.extend(rsp_json.get('join') or []) group_id_list.extend(rsp_json.get('manage') or []) group_id_list.extend(rsp_json.get('create') or []) if group_id_list: for group in group_id_list: self.group_id_list[str(group['gc'])] = group return group_id_list else: logger.warning( "seems this account didn't join any group: {}".format( response)) return [] else: logger.warning("get_group_list code unknown: {}".format(response)) return None
def search_group_members(self, group_id): """ 获取群成员详细信息的的列表, uin为真实QQ号, 并存入cache :type group_id: str :return:list return list sample [ { "card": "群名片", "flag": 0, "g": 255, "join_time": 1385383309, "last_speak_time": 1471325570, "lv": { "level": 6, "point": 5490 }, "nick": "昵称", "qage": 0, "role": 0, "tags": "-1", "uin": 493658555 } ] """ url = "http://qun.qq.com/cgi-bin/qun_mgr/search_group_members" data = { 'bkn': self.bkn, 'gc': str(group_id), 'st': 0, 'end': 2000, 'sort': 0, } response = self.client.post(url, data=data, refer='http://qun.qq.com/member.html') logger.debug("search_group_members response: {}".format(response)) rsp_json = json.loads(response) if rsp_json['ec'] == 0: return rsp_json.get('mems') else: logger.warning( "search_group_members code unknown: {}".format(response)) return None
def get_group_list_with_group_id(self): """ 获取包含群名和群号的列表, 并存入cache, 其中gc为群号 :type group_id: str :return:list return list sample [ { "gc": 114302207, "gn": "测试群1", "owner": 484216451 }, { "gc": 125299202, "gn": "测试群2", "owner": 242917661 } ] """ if self._get_group_list: response = self._get_group_list else: url = "http://qun.qq.com/cgi-bin/qun_mgr/get_group_list" data = {'bkn': self.bkn} response = self.client.post(url, data=data, refer='http://qun.qq.com/member.html') self._get_group_list = response logger.debug("get_group_list response: {}".format(response)) rsp_json = json.loads(response) if rsp_json['ec'] == 0: group_id_list = list() group_id_list.extend(rsp_json.get('join') or []) group_id_list.extend(rsp_json.get('manage') or []) group_id_list.extend(rsp_json.get('create') or []) if group_id_list: for group in group_id_list: self.group_id_list[str(group['gc'])] = group return group_id_list else: logger.warning("seems this account didn't join any group: {}".format(response)) return [] else: logger.warning("get_group_list code unknown: {}".format(response)) return None
def weather(msg, bot): global query match = re.match( six.text_type('^(weather|天气) (.*)'), msg.content ) if match: logger.info("RUNTIMELOG 查询天气...") command = match.group(1) city = match.group(2) logger.info("RUNTIMELOG 查询天气语句: " + msg.content) if command == 'weather' or command == six.text_type('天气'): try: city_name = parse.quote(city.encode('utf-8')) url_str = "http://api.map.baidu.com/telematics/v3/weather?location={city}&ak={key}&output=json".format( city=city_name, key=KEY ) response = request.urlopen(url_str) data_html = response.read() logger.debug("RESPONSE {}".format(data_html)) json_result = json.loads(data_html.decode('utf-8'))['results'][0] str_data = "" str_data += json_result['currentCity'] + " PM:" + json_result['pm25'] + "\n" try: str_data += json_result["index"][0]['des'] + "\n" except: pass for data in json_result["weather_data"]: str_data += data['date'] + " " str_data += data['weather'] + " " str_data += data['wind'] + " " str_data += data['temperature'] str_data += '\n' except Exception as ex: logger.debug(ex) str_data = "City not found :(" bot.reply_msg(msg, str_data) return True return False
def search_group_members(self, group_id): """ 获取群成员详细信息的的列表, uin为真实QQ号, 并存入cache :type group_id: str :return:list return list sample [ { "card": "群名片", "flag": 0, "g": 255, "join_time": 1385383309, "last_speak_time": 1471325570, "lv": { "level": 6, "point": 5490 }, "nick": "昵称", "qage": 0, "role": 0, "tags": "-1", "uin": 493658555 } ] """ url = "http://qun.qq.com/cgi-bin/qun_mgr/search_group_members" data = { 'bkn': self.bkn, 'gc': str(group_id), 'st': 0, 'end': 2000, 'sort': 0, } response = self.client.post(url, data=data, refer='http://qun.qq.com/member.html') logger.debug("search_group_members response: {}".format(response)) rsp_json = json.loads(response) if rsp_json['ec'] == 0: return rsp_json.get('mems') else: logger.warning("search_group_members code unknown: {}".format(response)) return None
def _handle_one(self, msg): """ :type msg: smart_qq_bot.messages.QMessage """ handlers = self._registry[msg.type] for handler in handlers + self._registry[RAW_TYPE]: if is_active(handler.name): if self.bot.is_self_msg(msg) and not handler.accept_self: logger.debug( "[{}] dropped group message from itself".format( handler.name)) continue self.handler_queue.put( Task(func=handler.func, name=handler.name, kwargs={ "msg": msg, "bot": self.bot }))
def _get_qr_login_status( self, qr_validation_url, appid, star_time, mibao_css, js_ver, sign, init_url ): redirect_url = None login_result = self.client.get( qr_validation_url.format( appid, date_to_millis(datetime.datetime.utcnow()) - star_time, mibao_css, js_ver, sign ), init_url ) ret_code = int(find_first_result(login_result, r"\d+?", None)) redirect_info = re.findall(r"(http.*?)\'", login_result) if redirect_info: logger.debug("redirect_info match is: %s" % redirect_info) redirect_url = redirect_info[0] return ret_code, redirect_url
def is_repeat(recorder, msg): if msg.src_group_name != '圆环之理': return False #[群消息]blabla match = re.match(r'^\[\w+\]\w+', msg.content) if match: return False #不要复读指令 match = re.match(r'^\/|!', msg.content) if match: return False rand = randint(1, 100) logger.debug(rand) if rand > 15: return False time = recorder.last_message_on if datetime.utcnow() < time + timedelta(minutes=1): return False recorder.last_message_on = datetime.utcnow() return True
def get_group_member_info_list(self, group_code): """ 获取指定群的成员信息 :group_code: int, can be "ture" of "fake" group_code {"result":{"cards":[{"card":"测试名片","muin":123456789}],"ginfo":{"class":27,"code":3281366860,"createtime":1457436527,"face":0,"fingermemo":"","flag":1090520065,"gid":3281366860,"level":0,"members":[{"mflag":0,"muin":1145751263},{"mflag":0,"muin":123456789},{"mflag":0,"muin":2123968182},{"mflag":1,"muin":271236123}],"memo":"","name":"测试群名","option":2,"owner":2123968182},"minfo":[{"city":"","country":"","gender":"male","nick":"测试2","province":"","uin":1145751263},{"city":"广州","country":"中国","gender":"male","nick":"测试1","province":"广东","uin":123456789},{"city":"广州","country":"中国","gender":"male","nick":"大","province":"广东","uin":2123968182},{"city":"","country":"中国","gender":"female","nick":"2","province":"","uin":271236123}],"stats":[],"vipinfo":[{"is_vip":0,"u":1145751263,"vip_level":0},{"is_vip":0,"u":123456789,"vip_level":0},{"is_vip":0,"u":2123968182,"vip_level":0},{"is_vip":0,"u":271236123,"vip_level":0}]},"retcode":0} :return:dict """ if group_code == 0: logger.debug("get_group_member_info_list 输入为0,返回 None") return try: url = "http://s.web2.qq.com/api/get_group_info_ext2?gcode=%s&vfwebqq=%s&t=%s" % ( group_code, self.vfwebqq, int(time.time() * 100)) response = self.client.get(url) logger.debug("get_group_member_info_list info response: {}".format( response)) rsp_json = json.loads(response) retcode = rsp_json["retcode"] if retcode == 0: result = rsp_json["result"] elif retcode == 6: logger.debug( "get_group_member_info_list retcode is 6, trying to get true code." ) result = self.get_group_member_info_list( self.get_true_group_code(group_code)) else: logger.warning("group_code error.") return self.group_member_info[str( group_code)] = result # 缓存群成员信息, 此处会把真假group_code都加入cache return result except Exception as ex: logger.warning("get_group_member_info_list. Error: " + str(ex)) return
def weather(msg, bot): global query match = re.match(six.text_type("^(weather|天气) (.*)"), msg.content) if match: logger.info("RUNTIMELOG 查询天气...") command = match.group(1) city = match.group(2) logger.info("RUNTIMELOG 查询天气语句: " + msg.content) if command == "weather" or command == six.text_type("天气"): try: city_name = parse.quote(city.encode("utf-8")) url_str = "http://api.map.baidu.com/telematics/v3/weather?location={city}&ak={key}&output=json".format( city=city_name, key=KEY ) response = request.urlopen(url_str) data_html = response.read() logger.debug("RESPONSE {}".format(data_html)) json_result = json.loads(data_html.decode("utf-8"))["results"][0] str_data = "" str_data += json_result["currentCity"] + " PM:" + json_result["pm25"] + "\n" try: str_data += json_result["index"][0]["des"] + "\n" except: pass for data in json_result["weather_data"]: str_data += data["date"] + " " str_data += data["weather"] + " " str_data += data["wind"] + " " str_data += data["temperature"] str_data += "\n" except Exception as ex: logger.debug(ex) str_data = "City not found :(" bot.reply_msg(msg, str_data) return True return False
def weather(msg, bot): global query match = re.match(ur'^(weather|天气) (\w+|[\u4e00-\u9fa5]+)', msg.content) if match: logger.info("RUNTIMELOG 查询天气...") command = match.group(1) city = match.group(2) logger.info("RUNTIMELOG 查询天气语句: " + msg.content) if command == 'weather' or command == u'天气': try: city_name = urllib2.quote(city.encode('utf-8')) url_str = "http://api.map.baidu.com/telematics/v3/weather?location={city}&ak={key}&output=json".format( city=city_name, key=KEY ) response = urllib2.urlopen(url_str) data_html = response.read() logger.debug("RESPONSE " + data_html) json_result = json.loads(data_html)['results'][0] str_data = "" str_data += json_result['currentCity'] + " PM:" + json_result['pm25'] + "\n" try: str_data += json_result["index"][0]['des'] + "\n" except: pass for data in json_result["weather_data"]: str_data += data['date'] + " " str_data += data['weather'] + " " str_data += data['wind'] + " " str_data += data['temperature'] str_data += '\n' except: str_data = "Not found city" bot.reply_msg(msg, str_data) return True return False
def weather(msg, bot): global query match = re.match(ur'^(weather|天气) (\w+|[\u4e00-\u9fa5]+)', msg.content) if match: logger.info("RUNTIMELOG 查询天气...") command = match.group(1) city = match.group(2) logger.info("RUNTIMELOG 查询天气语句: " + msg.content) if command == 'weather' or command == u'天气': try: city_name = urllib2.quote(city.encode('utf-8')) url_str = "http://api.map.baidu.com/telematics/v3/weather?location={city}&ak={key}&output=json".format( city=city_name, key=KEY) response = urllib2.urlopen(url_str) data_html = response.read() logger.debug("RESPONSE " + data_html) json_result = json.loads(data_html)['results'][0] str_data = "" str_data += json_result['currentCity'] + " PM:" + json_result[ 'pm25'] + "\n" try: str_data += json_result["index"][0]['des'] + "\n" except: pass for data in json_result["weather_data"]: str_data += data['date'] + " " str_data += data['weather'] + " " str_data += data['wind'] + " " str_data += data['temperature'] str_data += '\n' except: str_data = "Not found city" bot.reply_msg(msg, str_data) return True return False
def search_group_members(self, group_id): """ 获取群成员详细信息的的列表, u为真实QQ号, 并存入cache :type group_id: str :return:list return list sample [ { "b": 0, "g": 0, "n": "昵称", "u": 466331426 }, { "b": 0, "g": 0, "n": "昵称", "u": 493658565 } ] """ url = "http://qinfo.clt.qq.com/cgi-bin/qun_info/get_group_members_new" data = { 'bkn': self.bkn, 'gc': str(group_id), 'u': self._self_info.get('uin'), } response = self.client.post( url, data=data, refer='http://qinfo.clt.qq.com/member.html') logger.debug("search_group_members response: {}".format(response)) response = unescape_json_response(response) rsp_json = json.loads(response) if rsp_json['ec'] == 0: return rsp_json.get('mems', []) else: logger.warning( "search_group_members code unknown: {}".format(response)) return None
def main_loop(no_gui=False, new_user=False, debug=False, http=False, password=False): if debug: logger.setLevel(logging.DEBUG) else: logger.setLevel(logging.INFO) if http: run_http_daemon() logger.info("Initializing...") plugin_manager.load_plugin() if new_user: clean_cookie() bot.login(no_gui, password) observer = MessageObserver(bot) for name, func in bot_inited_registry.items(): try: t = threading.Thread(target=func, args=(bot, )) t.daemon = True t.start() except Exception: logging.exception("Error occurs while loading plugin [%s]." % name) while True: try: msg_list = bot.check_msg() if msg_list is not None: observer.handle_msg_list( [mk_msg(msg, bot) for msg in msg_list]) except ServerResponseEmpty: continue except (socket.timeout, IOError): logger.debug("Message pooling timeout, retrying...") except NeedRelogin: exit(0) except Exception: logger.exception("Exception occurs when checking msg.")
def Wiki(msg,bot): (account,account_type)=utils.get_account_and_type(msg) if utils.in_plugins(account,account_type,plugins_name): if utils.is_match('^(百科|Wiki|wiki) (.*)$',msg.content): p['srsearch']=utils.is_match('^(百科|Wiki|wiki) (.*)$',msg.content).group(2) tmp=requests.get(url,p,proxies=proxies) try: params['search']=json.loads(tmp.text)['query']['search'][0]['title'] except: params['search']=p['srsearch'] logger.info('[Wiki] Searching: '+params['search']+' for '+account_type+': '+account) tmp=requests.get(url,params=params,proxies=proxies) if tmp.status_code==200: try: result=json.loads(tmp.text)[2][0] if result!='': bot.reply_msg(msg,result) else: bot.reply_msg(msg,'未找到相关信息哦~') except Exception as e: logger.debug(e) else: bot.reply_msg(msg,"网络出错了哦~")
def check_msg(self): # Pooling the message response = self.client.post( 'http://d1.web2.qq.com/channel/poll2', { 'r': json.dumps( { "ptwebqq": self.ptwebqq, "clientid": self.client_id, "psessionid": self.psessionid, "key": "" } ) }, SMART_QQ_REFER ) logger.debug("Pooling returns response: %s" % response) if response == "": return try: ret = json.loads(response) except ValueError: logger.warning("RUNTIMELOG decode poll response error.") logger.debug("RESPONSE {}".format(response)) return ret_code = ret['retcode'] if ret_code in (0, 116): self._last_pool_success = True if ret_code == 0: if 'result' not in ret or len(ret['result']) == 0: logger.info("Pooling ends, no new message received.") else: return ret['result'] elif ret_code == 116: self.ptwebqq = ret['p'] logger.debug("ptwebqq updated in this pooling") else: self._last_pool_success = False if ret_code in (103, ): logger.warning("Pooling received retcode: " + str(ret_code)) elif ret_code in (121,): logger.warning("Pooling error with retcode %s" % ret_code) elif ret_code == 100006: logger.error("Pooling request error, response is: %s" % ret) else: logger.warning("Pooling returns unknown retcode %s" % ret_code) return None
def get_group_list_with_group_code(self): """ 获取包含群名和group_code的列表, 并存入cache, 其中code为group_code :type group_code: str :return:list [ { u 'code': 1131597161, # 这是真实group_code u 'flag': 184550417, u 'gid': 1802239929, # 这是msg.group_code, 即假group_code u 'name': u '测试' }, { u 'code': 1131597161, u 'flag': 184550417, u 'gid': 1802239929, u 'name': u '测试' } ] """ def _hash_digest(uin, ptwebqq): """ 计算hash,貌似TX的这个算法会经常变化,暂时不使用 get_group_code_list 会依赖此数据 提取自http://pub.idqqimg.com/smartqq/js/mq.js :param uin: :param ptwebqq: :return: """ N = [0, 0, 0, 0] # print(N[0]) for t in range(len(ptwebqq)): N[t % 4] ^= ord(ptwebqq[t]) U = ["EC", "OK"] V = [0, 0, 0, 0] V[0] = int(uin) >> 24 & 255 ^ ord(U[0][0]) V[1] = int(uin) >> 16 & 255 ^ ord(U[0][1]) V[2] = int(uin) >> 8 & 255 ^ ord(U[1][0]) V[3] = int(uin) & 255 ^ ord(U[1][1]) U = [0, 0, 0, 0, 0, 0, 0, 0] for T in range(8): if T % 2 == 0: U[T] = N[T >> 1] else: U[T] = V[T >> 1] N = ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "A", "B", "C", "D", "E", "F"] V = "" for T in range(len(U)): V += N[U[T] >> 4 & 15] V += N[U[T] & 15] return V logger.info("RUNTIMELOG Requesting the group list.") response = self.client.post( 'http://s.web2.qq.com/api/get_group_name_list_mask2', { 'r': json.dumps( { "vfwebqq": self.vfwebqq, "hash": _hash_digest(self._self_info['uin'], self.ptwebqq), } ) }, ) try: response = json.loads(response) except ValueError: logger.warning("RUNTIMELOG The response of group list request can't be load as json") return logger.debug("RESPONSE get_group_name_list_mask2 html: " + str(response)) if response['retcode'] != 0: raise TypeError('get_online_buddies2 result error') for group in response['result']['gnamelist']: self.group_code_list[str(group['gid'])] = group self.group_code_list[str(group['code'])] = group return response['result']['gnamelist']
def Manager(msg,bot): #获取群号 (account,account_type)=utils.get_account_and_type(msg) logger.debug('[Manager] account={0}, account_type={1}, send_uin={2}'.format(account,account_type,msg.send_uin)) logger.debug(isManager(msg,account_type)) if isManager(msg,account_type): if not utils.in_plugins(account,account_type,plugin_name) and utils.is_match(r'^!召唤 (.*)$',msg.content): #捕获昵称 nickname=utils.is_match(r'^!召唤 (.*)$',msg.content).group(1) #将昵称和账号绑定 sql.execute("insert into Nickname(account,nickname,content,suffix,account_type) values('{0}','{1}','{2}','{3}','{4}');".format(account,nickname,json.dumps(REPLY_CONTENT),json.dumps(REPLY_SUFFIX),account_type)) #默认激活所有插件 with open('./config/plugin.json','r') as f: plugin_list=json.load(f).get('plugin_on') for p in plugin_list: sql.execute("insert into plugins(account,plugin_name,account_type) values('{0}','{1}','{2}');".format(account,p,account_type)) logger.info('[Manager] '+account_type+': '+account+' Manager success') bot.reply_msg(msg,'召唤'+nickname+'成功~') elif utils.in_plugins(account,account_type,plugin_name): if utils.is_match(r'^!召唤 (.*)$',msg.content): #捕获昵称 nickname=utils.is_match(r'^!召唤 (.*)$',msg.content).group(1) bot.reply_msg(msg,'已经召唤过'+nickname+'啦~') #关闭插件 elif utils.is_match(r'^!关闭 (.*)$',msg.content): #获取插件名称 close_plugin_name=utils.is_match(r'^!关闭 (.*)$',msg.content).group(1) if close_plugin_name != 'Manager': #从数据库中删除群号-插件关系 l=sql.fetch_all('select plugin_name from plugins where account="{0}" and account_type="{1}";'.format(account,account_type)) tmp=[] for i in l: tmp.append(i[0]) if close_plugin_name in tmp: sql.execute("delete from plugins where account={0} and plugin_name='{1}' and account_type='{2}';".format(account,close_plugin_name,account_type)) logger.info('[Manager] '+account_type+': '+account+' inactivate '+close_plugin_name) bot.reply_msg(msg,"关闭{0}插件成功~".format(close_plugin_name)) else: bot.reply_msg(msg,close_plugin_name+'插件还没开启呢!') else: bot.reply_msg(msg,'不能关闭Manager插件哦~') #启用插件 elif utils.is_match(r'^!启用 (.*)$',msg.content): open_plugin_name=utils.is_match(r'^!启用 (.*)$',msg.content).group(1) tmp=sql.fetch_all('select plugin_name from plugins where account="{0}" and account_type="{1}";'.format(account,account_type)) open_list=[] for i in open_list: open_list.append(i[0]) with open('./config/plugin.json','r') as f: plugin_list=json.load(f).get('plugin_on') if open_plugin_name not in open_list and open_plugin_name in plugin_list: sql.execute("insert into plugins(account,plugin_name,account_type) values ('{0}','{1}','{2}');".format(account,open_plugin_name,account_type)) logger.info('[Manager] '+account_type+': '+account+' Manager '+open_plugin_name) bot.reply_msg(msg,"开启{0}插件成功~".format(open_plugin_name)) elif open_plugin_name in open_list: bot.reply_msg(msg,open_plugin_name+'已经在使用了哦~') else: bot.reply_msg(msg,'别捣乱,哼哼~') #列出插件列表 elif utils.is_match(r'^!已启用插件$',msg.content): l=sql.fetch_all('select plugin_name from plugins where account="{0}" and account_type="{1}";'.format(account,account_type)) s="已启用插件:\n" for i in l: s+=i[0]+'\n' bot.reply_msg(msg,s) #列出可用插件 elif utils.is_match(r'^!可用插件$',msg.content): s="" with open('./config/plugin.json','r') as f: plugin_list=json.load(f).get('plugin_on') tmp=sql.fetch_all('select plugin_name from plugins where account="{0}" and account_type="{1}";'.format(account,account_type)) l=[] for i in tmp: l.append(i[0]) for p in plugin_list: if p not in l: s+=p+'\n' if s!="": bot.reply_msg(msg,s) else: bot.reply_msg(msg,'全部插件都启用了啦~') #添加管理人员 elif account_type=='group' and utils.is_match(r'^!管理 add (\d{5,12})$',msg.content): manager_id=utils.is_match(r'^!管理 add (\d{5,12})$',msg.content).group(1) if sql.execute('select * from Manager where account_id={0} and group_id={1};'.format(manager_id,account)): bot.reply_mag(msg,'已经是管理员啦~') else: sql.execute('insert into Manager(account_id,group_id) values("{0}","{1}");'.format(manager_id,account)) bot.reply_msg(msg,"{0} 成功成为管理人员~".format(manager_id)) #列出管理人员 elif account_type=='group' and utils.is_match(r'^!管理 list$',msg.content): l=sql.fetch_all('select account_id from Manager where group_id={0};'.format(account)) p=sql.fetch_all('select account_id from Manager where group_id="00000";') if p: s='全局管理员:\n' for i in p: s+=i[0]+'\n' if l: s+='群管理员:\n' for i in l: s+=i[0]+'\n' bot.reply_msg(msg,s) else: bot.reply_msg(msg,"没有管理人员哦~") #删除管理人员 elif account_type=='group' and utils.is_match(r'^!管理 remove (\d{5,12})$',msg.content): manager_id=utils.is_match(r'^!管理 remove (\d{5,12})$',msg.content).group(1) sql.execute('delete from Manager where account_id={0} and group_id={1};'.format(manager_id,account)) bot.reply_msg(msg,'{0} 已不再是管理员。。。'.format(manager_id)) elif utils.is_match(r'^!名字$',msg.content): nickname=sql.fetch_one('select nickname from Nickname where account="{0}" and account_type="{1}";'.format(account,account_type))[0] bot.reply_msg(msg,'我叫'+nickname+'!') #完全关闭 elif utils.is_match(r'^!回去吧 (.*)$',msg.content): m_nickname=utils.is_match(r'^!回去吧 (.*)$',msg.content).group(1) nickname=sql.fetch_one('select nickname from Nickname where account="{0}" and account_type="{1}";'.format(account,account_type))[0] if m_nickname==nickname: sql.execute('delete from plugins where account="{0}" and account_type="{1}";'.format(account,account_type)) bot.reply_msg(msg,nickname+'走了,拜拜~')
def _login_by_qrcode(self, no_gui): logger.info("RUNTIMELOG Trying to login by qrcode.") logger.info("RUNTIMELOG Requesting the qrcode login pages...") qr_validation_url = 'https://ssl.ptlogin2.qq.com/ptqrlogin?' \ 'webqq_type=10&remember_uin=1&login2qq=1&aid={0}' \ '&u1=http%3A%2F%2Fw.qq.com%2Fproxy.html%3Flogin2qq%3D1%26webqq_type%3D10' \ '&ptredirect=0&ptlang=2052&daid=164&from_ui=1&pttype=1&dumy=' \ '&fp=loginerroralert&action=0-0-{1}&mibao_css={2}' \ '&t=undefined&g=1&js_type=0&js_ver={3}&login_sig={4}' init_url = "https://ui.ptlogin2.qq.com/cgi-bin/login?" \ "daid=164&target=self&style=16&mibao_css=m_webqq" \ "&appid=501004106&enable_qlogin=0&no_verifyimg=1" \ "&s_url=http%3A%2F%2Fw.qq.com%2Fproxy.html" \ "&f_url=loginerroralert&strong_login=1" \ "&login_state=10&t=20131024001" html = self.client.get( init_url, ) appid = find_first_result( html, r'<input type="hidden" name="aid" value="(\d+)" />', 'Get AppId Error', True ) sign = find_first_result( html, r'g_login_sig=encodeURIComponent\("(.*?)"\)', 'Get Login Sign Error', ) js_ver = find_first_result( html, r'g_pt_version=encodeURIComponent\("(\d+)"\)', 'Get g_pt_version Error', True, ) mibao_css = find_first_result( html, r'g_mibao_css=encodeURIComponent\("(.+?)"\)', 'Get g_mibao_css Error', True ) star_time = date_to_millis(datetime.datetime.utcnow()) error_times = 0 ret_code = None login_result = None redirect_url = None while True: error_times += 1 logger.info("Downloading QRCode file...") self.client.download( 'https://ssl.ptlogin2.qq.com/ptqrshow?appid={0}&e=0&l=L&s=8&d=72&v=4'.format(appid), self.qrcode_path ) if not no_gui: thread = Thread(target=show_qr, args=(self.qrcode_path, )) thread.setDaemon(True) thread.start() while True: ret_code, redirect_url = self._get_qr_login_status( qr_validation_url, appid, star_time, mibao_css, js_ver, sign, init_url ) if ret_code in ( QR_CODE_STATUS['succeed'], QR_CODE_STATUS["qr_code_expired"] ): break time.sleep(1) if ret_code == QR_CODE_STATUS['succeed'] or error_times > 10: break if os.path.exists(self.qrcode_path): os.remove(self.qrcode_path) login_failed_tips = "QRCode validation response is:\n%s" % login_result if ret_code is not None and (ret_code != 0): raise QRLoginFailed(login_failed_tips) elif redirect_url is None: raise QRLoginFailed(login_failed_tips) else: html = self.client.get(redirect_url) logger.debug("QR Login redirect_url response: %s" % html) return True