def wechat_video_msg(self, msg): mobj = EFBMsg(self) mobj.path, mobj.mime = self.save_file(msg, MsgType.Video) mobj.type = MsgType.Video mobj.text = None mobj.file = open(mobj.path, "rb") return mobj
def wechat_friend_msg(self, msg): mobj = EFBMsg(self) txt = ("Friend request: {NickName}\n" "From: {Province}, {City}\n" "QQ: {QQNum}\n" "ID: {Alias}\n" "Signature: {Signature}\n" "Gender: {Sex}") tdict = msg['Text'].copy() tdict.update(msg['Text']['userInfo']) txt = txt.format(**tdict) mobj.text = txt mobj.type = MsgType.Command mobj.attributes = { "commands": [{ "name": "Send friend request", "callable": "add_friend", "args": [], "kwargs": { "userName": msg['Text']['userInfo']['UserName'], "status": 3, "ticket": msg['Ticket'] } }] } return mobj
def fileMsg(self, msg, isGroupChat=False): mobj = EFBMsg(self) mobj.type = MsgType.File mobj.path, mobj.mime = self.save_file(msg, mobj.type) mobj.text = msg['FileName'] mobj.file = open(mobj.path, "rb") return mobj
def wechat_location_msg(self, msg): mobj = EFBMsg(self) mobj.text = msg['Content'].split('\n')[0][:-1] loc = re.search("=-?([0-9.]+),-?([0-9.]+)", msg['Url']).groups() mobj.attributes = {"longitude": float(loc[1]), "latitude": float(loc[0])} mobj.type = MsgType.Location return mobj
def exit_callback(self): if self.stop_polling: return msg = EFBMsg(self) msg.source = MsgSource.System msg.origin = { 'name': '%s System' % self.channel_name, 'alias': '%s System' % self.channel_name, 'uid': -1 } msg.text = "WeChat server logged out the user." msg.type = MsgType.Text on_log_out = self._flag("on_log_out", "command") on_log_out = on_log_out if on_log_out in ("command", "idle", "reauth") else "command" if on_log_out == "command": msg.type = MsgType.Command msg.attributes = { "commands": [{ "name": "Log in", "callable": "reauth", "args": [], "kwargs": { "command": True } }] } elif on_log_out == "reauth": if self._flag("qr_reload", "master_qr_code") == "console_qr_code": msg.text += "\nPlease visit your console or log for QR code and further instructions." self.reauth() self.queue.put(msg)
def videoMsg(self, msg, isGroupChat=False): mobj = EFBMsg(self) mobj.path, mobj.mime = self.save_file(msg, MsgType.Video) mobj.type = MsgType.Video mobj.text = None mobj.file = open(mobj.path, "rb") return mobj
def master_qr_code(self, uuid, status, qrcode): status = int(status) msg = EFBMsg(self) msg.type = MsgType.Text msg.source = MsgSource.System msg.origin = { 'name': '%s Auth' % self.channel_name, 'alias': '%s Auth' % self.channel_name, 'uid': -1 } if status == 201: msg.type = MsgType.Text msg.text = 'Tap "Confirm" to continue.' elif status == 200: msg.type = MsgType.Text msg.text = "Successfully authenticated." elif uuid != self.qr_uuid: msg.type = MsgType.Image path = os.path.join("storage", self.channel_id) if not os.path.exists(path): os.makedirs(path) path = os.path.join(path, 'QR-%s.jpg' % int(time.time())) self.logger.debug("master_qr_code file path: %s", path) qr_url = "https://login.weixin.qq.com/l/" + uuid QRCode(qr_url).png(path, scale=10) msg.text = 'Scan this QR Code with WeChat to continue.' msg.path = path msg.file = open(path, 'rb') msg.mime = 'image/jpeg' if status in (200, 201) or uuid != self.qr_uuid: self.queue.put(msg) self.qr_uuid = uuid
def linkMsg(self, msg, isGroupChat=False): self.logger.info("---\nNew Link msg, %s", msg) # initiate object mobj = EFBMsg(self) # parse XML itchat.utils.emoji_formatter(msg, 'Content') xmldata = msg['Content'] data = xmltodict.parse(xmldata) # set attributes mobj.attributes = { "title": data['msg']['appmsg']['title'], "description": data['msg']['appmsg']['des'], "image": data['msg']['appmsg']['thumburl'], "url": data['msg']['appmsg']['url'] } if mobj.attributes['url'] is None: txt = mobj.attributes['title'] or '' txt += mobj.attributes['description'] or '' msg['Text'] = txt return self.textMsg(msg, isGroupChat) # format text mobj.text = "" if self._flag("extra_links_on_message", False): extra_link = data.get('msg', {}).get('appmsg', {}).get('mmreader', {}).get('category', {}).get('item', []) if type(extra_link) is list and len(extra_link): for i in extra_link: mobj.text += "🔗 %s\n%s\n%s\n🖼 %s\n\n" % ( i['title'], i['digest'], i['url'], i['cover']) mobj.type = MsgType.Link return mobj
def wechat_card_msg(self, msg): mobj = EFBMsg(self) txt = ("Name card: {NickName}\n" "From: {Province}, {City}\n" "QQ: {QQNum}\n" "ID: {Alias}\n" "Signature: {Signature}\n" "Gender: {Sex}") txt = txt.format(**msg['Text']) mobj.text = txt mobj.type = MsgType.Command mobj.attributes = { "commands": [ { "name": "Send friend request", "callable": "add_friend", "args": [], "kwargs": { "userName": msg['Text']['UserName'], "status": 2, "ticket": "" } } ] } return mobj
def wechat_voice_msg(self, msg): mobj = EFBMsg(self) mobj.type = MsgType.Audio mobj.path, mobj.mime = self.save_file(msg, mobj.type) mobj.text = None mobj.file = open(mobj.path, "rb") return mobj
def wechat_file_msg(self, msg): mobj = EFBMsg(self) mobj.type = MsgType.File mobj.path, mobj.mime = self.save_file(msg, mobj.type) mobj.text = msg['FileName'] mobj.filename = msg['FileName'] or None mobj.file = open(mobj.path, "rb") return mobj
def wechat_picture_msg(self, msg): mobj = EFBMsg(self) mobj.type = MsgType.Image if msg['MsgType'] == 3 else MsgType.Sticker mobj.path, mime = self.save_file(msg, mobj.type) mobj.text = None mobj.file = open(mobj.path, "rb") mobj.mime = mime return mobj
def pictureMsg(self, msg, isGroupChat=False): mobj = EFBMsg(self) mobj.type = MsgType.Image mobj.path, mime = self.save_file(msg, mobj.type) mobj.text = None mobj.file = open(mobj.path, "rb") mobj.mime = mime return mobj
def textMsg(self, msg, isGroupChat=False): self.logger.info("TextMsg!!!\n---") if msg['Text'].startswith( "http://weixin.qq.com/cgi-bin/redirectforward?args="): return self.locationMsg(msg, isGroupChat) mobj = EFBMsg(self) mobj.text = msg['Text'] mobj.type = MsgType.Text return mobj
def wechat_text_msg(self, msg): if msg['FromUserName'] == "newsapp" and msg['Content'].startswith("<mmreader>"): self.wechat_newsapp_msg(msg) return if msg['Text'].startswith("http://weixin.qq.com/cgi-bin/redirectforward?args="): self.wechat_location_msg(msg) return mobj = EFBMsg(self) mobj.text = msg['Text'] mobj.type = MsgType.Text return mobj
def exit_callback(self): msg = EFBMsg(self) msg.type = MsgType.Text msg.source = MsgSource.System msg.origin = { 'name': 'WeChat System Message', 'alias': 'WeChat System Message', 'uid': -1 } msg.text = "WeChat system logged out the user." self.queue.put(msg)
def wechat_raw_link_msg(self, msg, title, description, image, url): mobj = EFBMsg(self) if url: mobj.type = MsgType.Link mobj.attributes = { "title": title, "description": description, "image": image, "url": url } else: mobj.type = MsgType.Text mobj.text = "%s\n%s" % (title, description) if image: mobj.text += "\n\n%s" % image return mobj
def linkMsg(self, msg, isGroupChat=False): self.logger.info("---\nNew Link msg, %s", msg) # initiate object mobj = EFBMsg(self) # parse XML itchat.utils.emoji_formatter(msg, 'Content') xmldata = msg['Content'] data = xmltodict.parse(xmldata) # set attributes mobj.attributes = { "title": data['msg']['appmsg']['title'], "description": data['msg']['appmsg']['des'], "image": None, "url": data['msg']['appmsg']['url'] } # format text mobj.text = "🔗 %s\n%s\n\n%s" % (mobj.attributes['title'], mobj.attributes['description'], mobj.attributes['url']) mobj.type = MsgType.Link return mobj
def friendMsg(self, msg, isGroupChat=False): mobj = EFBMsg(self) txt = ("Friend request: {NickName}\n" "From: {Province}, {City}\n" "QQ: {QQNum}\n" "ID: {Alias}\n" "Signature: {Signature}\n" "Gender: {Sex}") txt = txt.format(**{**msg['Text'], **msg['Text']['userInfo']}) mobj.text = txt mobj.type = MsgType.Command mobj.attributes = { "commands": [{ "name": "Send friend request", "callable": "add_friend", "args": [], "kwargs": { "userName": msg['Text']['userInfo']['UserName'], "status": 3, "ticket": msg['Ticket'] } }] } return mobj
def wechat_system_msg(self, msg): mobj = EFBMsg(self) mobj.text = "System message: %s" % msg['Text'] mobj.type = MsgType.Text return mobj
def systemMsg(self, msg, isGroupChat=False): mobj = EFBMsg(self) mobj.text = "System message: %s" % msg['Text'] mobj.type = MsgType.Text return mobj