diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..394a0a0
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,2 @@
+#!/usr/bin/env python
+# coding: utf-8
diff --git a/weixin.py b/weixin.py
index ea1172c..7741522 100755
--- a/weixin.py
+++ b/weixin.py
@@ -9,7 +9,10 @@ import json
import time, re, sys, os, random
import multiprocessing
import platform
+import logging
from collections import defaultdict
+from urlparse import urlparse
+from lxml import html
def catchKeyboardInterrupt(fn):
def wrapper(*args):
@@ -17,6 +20,7 @@ def catchKeyboardInterrupt(fn):
return fn(*args)
except KeyboardInterrupt:
print '\n[*] 强制退出程序'
+ logging.debug('[*] 强制退出程序')
return wrapper
def _decode_list(data):
@@ -79,16 +83,32 @@ class WebWeixin(object):
self.GroupList = []
self.autoReplyMode = False
self.syncHost = ''
+ self.user_agent = 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_11_3) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/48.0.2564.109 Safari/537.36'
+ self.interactive = False
+ self.autoOpen = False
+ self.saveFolder = os.path.join(os.getcwd(), 'saved')
+ self.saveSubFolders = {'webwxgeticon': 'icons', 'webwxgetheadimg': 'headimgs', 'webwxgetmsgimg': 'msgimgs', 'webwxgetvideo': 'videos', 'webwxgetvoice': 'voices', '_showQRCodeImg': 'qrcodes'}
+ self.appid = 'wx782c26e4c19acffb'
+ self.lang = 'zh_CN'
+
opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookielib.CookieJar()))
+ opener.addheaders = [('User-agent', self.user_agent)]
urllib2.install_opener(opener)
+ def loadConfig(self, config):
+ if config['DEBUG']: self.DEBUG = config['DEBUG']
+ if config['autoReplyMode']: self.autoReplyMode = config['autoReplyMode']
+ if config['user_agent']: self.user_agent = config['user_agent']
+ if config['interactive']: self.interactive = config['interactive']
+ if config['autoOpen']: self.autoOpen = config['autoOpen']
+
def getUUID(self):
url = 'https://login.weixin.qq.com/jslogin'
params = {
- 'appid': 'wx782c26e4c19acffb',
+ 'appid': self.appid,
'fun': 'new',
- 'lang': 'zh_CN',
+ 'lang': self.lang,
'_': int(time.time()),
}
data = self._post(url, params, False)
@@ -101,23 +121,20 @@ class WebWeixin(object):
return False
def genQRCode(self):
- if sys.platform.find('win') >= 0:
+ if sys.platform.startswith('win'):
self._showQRCodeImg()
else:
self._str2qr('https://login.weixin.qq.com/l/' + self.uuid)
def _showQRCodeImg(self):
- QRCODE_PATH = os.path.join(os.getcwd(),'qrcode.jpg')
url = 'https://login.weixin.qq.com/qrcode/' + self.uuid
params = {
't' : 'webwx',
'_' : int(time.time())
}
- data = self._post(url,params,False)
- with open(QRCODE_PATH, 'wb') as f:
- f.write(data)
-
+ data = self._post(url, params, False)
+ QRCODE_PATH = self._saveFile('qrcode.jpg', data, '_showQRCodeImg')
os.startfile(QRCODE_PATH)
def waitForLogin(self, tip = 1):
@@ -268,6 +285,7 @@ class WebWeixin(object):
dic = self._post(url, params)
if self.DEBUG:
print json.dumps(dic, indent=4)
+ logging.debug(json.dumps(dic, indent=4))
if dic['BaseResponse']['Ret'] == 0:
self.SyncKey = dic['SyncKey']
@@ -294,41 +312,47 @@ class WebWeixin(object):
dic = r.json()
return dic['BaseResponse']['Ret'] == 0
+ def _saveFile(self, filename, data, api=None):
+ fn = filename
+ if self.saveSubFolders[api]:
+ dirName = os.path.join(self.saveFolder, self.saveSubFolders[api])
+ if not os.path.exists(dirName):
+ os.makedirs(dirName)
+ fn = os.path.join(dirName, filename)
+ logging.debug('Saved file: %s' % fn)
+ with open(fn, 'wb') as f: f.write(data); f.close()
+ return fn
+
def webwxgeticon(self, id):
url = self.base_uri + '/webwxgeticon?username=%s&skey=%s' % (id, self.skey)
data = self._get(url)
fn = 'img_'+id+'.jpg'
- with open(fn, 'wb') as f: f.write(data)
- return fn
+ return self._saveFile(fn, data, 'webwxgeticon')
def webwxgetheadimg(self, id):
url = self.base_uri + '/webwxgetheadimg?username=%s&skey=%s' % (id, self.skey)
data = self._get(url)
fn = 'img_'+id+'.jpg'
- with open(fn, 'wb') as f: f.write(data)
- return fn
+ return self._saveFile(fn, data, 'webwxgetheadimg')
def webwxgetmsgimg(self, msgid):
url = self.base_uri + '/webwxgetmsgimg?MsgID=%s&skey=%s' % (msgid, self.skey)
data = self._get(url)
fn = 'img_'+msgid+'.jpg'
- with open(fn, 'wb') as f: f.write(data)
- return fn
+ return self._saveFile(fn, data, 'webwxgetmsgimg')
# Not work now for weixin haven't support this API
def webwxgetvideo(self, msgid):
url = self.base_uri + '/webwxgetvideo?msgid=%s&skey=%s' % (msgid, self.skey)
- data = self._get(url)
+ data = self._get(url, api='webwxgetvideo')
fn = 'video_'+msgid+'.mp4'
- with open(fn, 'wb') as f: f.write(data)
- return fn
+ return self._saveFile(fn, data, 'webwxgetvideo')
def webwxgetvoice(self, msgid):
url = self.base_uri + '/webwxgetvoice?msgid=%s&skey=%s' % (msgid, self.skey)
data = self._get(url)
fn = 'voice_'+msgid+'.mp3'
- with open(fn, 'wb') as f: f.write(data)
- return fn
+ return self._saveFile(fn, data, 'webwxgetvoice')
def getUserRemarkName(self, id):
name = '未知群' if id[:2] == '@@' else '陌生人'
@@ -344,14 +368,69 @@ class WebWeixin(object):
return member['UserName']
return None
+ def _showMsg(self, message):
+
+ srcName = None
+ dstName = None
+ groupName = None
+ content = None
+
+ msg = message
+ logging.debug(msg)
+
+ if msg['raw_msg']:
+ srcName = self.getUserRemarkName(msg['raw_msg']['FromUserName'])
+ dstName = self.getUserRemarkName(msg['raw_msg']['ToUserName'])
+ content = msg['raw_msg']['Content'].replace('<','<').replace('>','>')
+ message_id = msg['raw_msg']['MsgId']
+
+ if content.find('http://weixin.qq.com/cgi-bin/redirectforward?args=') != -1:
+ # 地理位置消息
+ data = self._get(content).decode('gbk').encode('utf-8')
+ pos = self._searchContent('title', data, 'xml')
+ tree = html.fromstring(self._get(content))
+ url = tree.xpath('//html/body/div/img')[0].attrib['src']
+
+ for item in urlparse(url).query.split('&'):
+ if item.split('=')[0] == 'center': loc = item.split('=')[-1:]
+
+ content = '%s 发送了一个 位置消息 - 我在 [%s](%s) @ %s]' % (srcName, pos, url, loc)
+
+ if msg['raw_msg']['ToUserName'] == 'filehelper':
+ # 文件传输助手
+ dstName = '文件传输助手'
+
+ if msg['raw_msg']['FromUserName'][:2] == '@@':
+ # 接收到来自群的消息
+ [people, content] = content.split(':
')
+ groupName = srcName
+ srcName = self.getUserRemarkName(people)
+ dstName = 'GROUP'
+ elif msg['raw_msg']['ToUserName'][:2] == '@@':
+ # 自己发给群的消息
+ groupName = dstName
+ dstName = 'GROUP'
+
+ # 指定了消息内容
+ if 'message' in msg.keys(): content = msg['message']
+
+ if groupName != None:
+ print '%s |%s| %s -> %s: %s' % (message_id, groupName.strip(), srcName.strip(), dstName.strip(), content.replace('
','\n'))
+ logging.info('%s |%s| %s -> %s: %s' % (message_id, groupName.strip(), srcName.strip(), dstName.strip(), content.replace('
','\n')))
+ else:
+ print '%s %s -> %s: %s' % (message_id, srcName.strip(), dstName.strip(), content.replace('
','\n'))
+ logging.info('%s %s -> %s: %s' % (message_id, srcName.strip(), dstName.strip(), content.replace('
','\n')))
+
def handleMsg(self, r):
for msg in r['AddMsgList']:
print '[*] 你有新的消息,请注意查收'
+ logging.debug('[*] 你有新的消息,请注意查收')
if self.DEBUG:
fn = 'msg' + str(int(random.random() * 1000)) + '.json'
with open(fn, 'w') as f: f.write(json.dumps(msg))
print '[*] 该消息已储存到文件: ' + fn
+ logging.debug('[*] 该消息已储存到文件: %s' % (fn))
msgType = msg['MsgType']
name = self.getUserRemarkName(msg['FromUserName'])
@@ -359,78 +438,83 @@ class WebWeixin(object):
msgid = msg['MsgId']
if msgType == 51:
print '[*] 成功截获微信初始化消息'
+ logging.debug('[*] 成功截获微信初始化消息')
elif msgType == 1:
- if content.find('http://weixin.qq.com/cgi-bin/redirectforward?args=') != -1:
- # 地理位置消息
- data = self._get(content).decode('gbk').encode('utf-8')
- pos = self._searchContent('title', data, 'xml')
- print '%s 给你发送了一个位置消息 [我在%s]' % (name, pos)
- elif msg['ToUserName'] == 'filehelper':
- print '%s -> 文件传输助手: %s' % (name, content.replace('
','\n'))
- elif msg['FromUserName'] == self.User['UserName']:
- pass
- elif msg['FromUserName'][:2] == '@@':
- [people, content] = content.split(':
')
- group = self.getUserRemarkName(msg['FromUserName'])
- name = self.getUserRemarkName(people)
- print '|%s| %s: %s' % (group, name, content.replace('
','\n'))
- else:
- print name+': '+content
- if self.autoReplyMode:
- ans = self._xiaodoubi(content)+'\n[微信机器人自动回复]'
- if self.webwxsendmsg(ans, msg['FromUserName']):
- print '自动回复: '+ans
- else:
- print '自动回复失败'
+ raw_msg = { 'raw_msg': msg }
+ self._showMsg(raw_msg)
+ if self.autoReplyMode:
+ ans = self._xiaodoubi(content)+'\n[微信机器人自动回复]'
+ if self.webwxsendmsg(ans, msg['FromUserName']):
+ print '自动回复: '+ans
+ logging.info('自动回复: '+ans)
+ else:
+ print '自动回复失败'
+ logging.info('自动回复失败')
elif msgType == 3:
image = self.webwxgetmsgimg(msgid)
- print '%s 给你发送了一张图片: %s' % (name, image)
+ raw_msg = { 'raw_msg': msg, 'message': '%s 发送了一张图片: %s' % (name, image) }
+ self._showMsg(raw_msg)
self._safe_open(image)
elif msgType == 34:
voice = self.webwxgetvoice(msgid)
- print '%s 给你发了一段语音: %s' % (name, voice)
+ raw_msg = { 'raw_msg': msg, 'message': '%s 发了一段语音: %s' % (name, voice) }
+ self._showMsg(raw_msg)
self._safe_open(voice)
elif msgType == 42:
info = msg['RecommendInfo']
- print '%s 给你发送了一张名片:' % name
+ print '%s 发送了一张名片:' % name
print '========================='
print '= 昵称: %s' % info['NickName']
print '= 微信号: %s' % info['Alias']
print '= 地区: %s %s' % (info['Province'], info['City'])
print '= 性别: %s' % ['未知', '男', '女'][info['Sex']]
print '========================='
+ logging.info('%s 发送了一张名片: %s' % (name.strip(), json.dumps(info)))
elif msgType == 47:
url = self._searchContent('cdnurl', content)
- print '%s 给你发了一个动画表情,点击下面链接查看:\n%s' % (name, url)
+ raw_msg = { 'raw_msg': msg, 'message': '%s 发了一个动画表情,点击下面链接查看: %s' % (name, url) }
+ self._showMsg(raw_msg)
self._safe_open(url)
elif msgType == 49:
appMsgType = defaultdict(lambda : "")
appMsgType.update({5:'链接', 3:'音乐', 7:'微博'})
- print '%s 给你分享了一个%s:' % (name, appMsgType[msg['AppMsgType']])
+ print '%s 分享了一个%s:' % (name, appMsgType[msg['AppMsgType']])
print '========================='
print '= 标题: %s' % msg['FileName']
print '= 描述: %s' % self._searchContent('des', content, 'xml')
print '= 链接: %s' % msg['Url']
print '= 来自: %s' % self._searchContent('appname', content, 'xml')
print '========================='
+ string = "%s 分享了一个%s:" % (name, appMsgType[msg['AppMsgType']])
elif msgType == 62:
- print name+' 给你发了一个小视频,请在手机上查看'
+ video = self.webwxgetvideo(msgid)
+ raw_msg = { 'raw_msg': msg, 'message': '%s 发了一段语音: %s' % (name, video) }
+ self._showMsg(raw_msg)
+ self._safe_open(video)
elif msgType == 10002:
print name+' 撤回消息'
+ logging.info(name+' 撤回消息')
else:
print '[*] 该消息类型为: %d,可能是表情,图片或链接' % msg['MsgType']
- print msg
+ logging.debug('[*] 该消息类型为: %d,可能是表情,图片或链接: %s' % (msg['MsgType'], json.dumps(msg)))
def listenMsgMode(self):
print '[*] 进入消息监听模式 ... 成功'
+ logging.debug('[*] 进入消息监听模式 ... 成功')
self._run('[*] 进行同步线路测试 ... ', self.testsynccheck)
playWeChat = 0
while True:
[retcode, selector] = self.synccheck()
if self.DEBUG: print 'retcode: %s, selector: %s' % (retcode, selector)
+ logging.debug('retcode: %s, selector: %s' % (retcode, selector))
if retcode == '1100':
print '[*] 你在手机上登出了微信,债见'
+ logging.debug('[*] 你在手机上登出了微信,债见')
break
+ if retcode == '1101':
+ print '[*] 你在其他地方登录了 WEB 版微信,债见'
+ logging.debug('[*] 你在其他地方登录了 WEB 版微信,债见')
+ break
elif retcode == '0':
if selector == '2':
r = self.webwxsync()
@@ -438,6 +522,7 @@ class WebWeixin(object):
elif selector == '7':
playWeChat += 1
print '[*] 你在手机上玩微信被我发现了 %d 次' % playWeChat
+ logging.debug('[*] 你在手机上玩微信被我发现了 %d 次' % playWeChat)
r = self.webwxsync()
elif selector == '0':
time.sleep(1)
@@ -458,10 +543,13 @@ class WebWeixin(object):
else:
if self.webwxsendmsg(word, id):
print '[*] 消息发送成功'
+ logging.debug('[*] 消息发送成功')
else:
print '[*] 消息发送失败'
+ logging.debug('[*] 消息发送失败')
else:
print '[*] 此用户不存在'
+ logging.debug('[*] 此用户不存在')
def sendMsgToAll(self, word):
for contact in self.ContactList:
@@ -476,23 +564,26 @@ class WebWeixin(object):
@catchKeyboardInterrupt
def start(self):
- print '[*] 微信网页版 ... 开动'
+ self._echo('[*] 微信网页版 ... 开动'); print; logging.debug('[*] 微信网页版 ... 开动')
self._run('[*] 正在获取 uuid ... ', self.getUUID)
- print '[*] 正在获取二维码 ... 成功'; self.genQRCode()
+ self._echo('[*] 正在获取二维码 ... 成功'); print; logging.debug('[*] 微信网页版 ... 开动'); self.genQRCode()
self._run('[*] 请使用微信扫描二维码以登录 ... ', self.waitForLogin)
self._run('[*] 请在手机上点击确认以登录 ... ', self.waitForLogin, 0)
self._run('[*] 正在登录 ... ', self.login)
self._run('[*] 微信初始化 ... ', self.webwxinit)
self._run('[*] 开启状态通知 ... ', self.webwxstatusnotify)
self._run('[*] 获取联系人 ... ', self.webwxgetcontact)
- print '[*] 共有 %d 位联系人' % len(self.ContactList)
+ self._echo('[*] 共有 %d 位联系人' % len(self.ContactList)); print; logging.debug('[*] 微信网页版 ... 开动')
if self.DEBUG: print self
+ logging.debug(self)
- if raw_input('[*] 是否开启自动回复模式(y/n): ') == 'y':
+ if self.interactive and raw_input('[*] 是否开启自动回复模式(y/n): ') == 'y':
self.autoReplyMode = True
print '[*] 自动回复模式 ... 开启'
+ logging.debug('[*] 自动回复模式 ... 开启')
else:
print '[*] 自动回复模式 ... 关闭'
+ logging.debug('[*] 自动回复模式 ... 关闭')
listenProcess = multiprocessing.Process(target=self.listenMsgMode)
listenProcess.start()
@@ -502,6 +593,7 @@ class WebWeixin(object):
if text == 'quit':
listenProcess.terminate()
print('[*] 退出微信')
+ logging.debug('[*] 退出微信')
exit()
elif text[:2] == '->':
[name, word] = text[2:].split(':')
@@ -512,19 +604,22 @@ class WebWeixin(object):
self.sendMsg(name, file, True)
elif text[:3] == 'f->':
print '发送文件'
+ logging.debug('发送文件')
elif text[:3] == 'i->':
print '发送图片'
+ logging.debug('发送图片')
def _safe_open(self, path):
- if platform.system() == "Linux":
- os.system("xdg-open %s &" % path)
- else:
- os.system('open %s &' % path)
+ if self.autoOpen:
+ if platform.system() == "Linux":
+ os.system("xdg-open %s &" % path)
+ else:
+ os.system('open %s &' % path)
def _run(self, str, func, *args):
self._echo(str)
- if func(*args): print '成功'
- else: print('失败\n[*] 退出程序');exit()
+ if func(*args): print '成功'; logging.debug('%s... 开启' % (str))
+ else: print('失败\n[*] 退出程序'); logging.debug('%s... 失败' % (str)); logging.debug('[*] 退出程序'); exit()
def _echo(self, str):
sys.stdout.write(str)
@@ -552,10 +647,14 @@ class WebWeixin(object):
result = data.decode('utf-8')
return result
- def _get(self, url):
+ def _get(self, url, api=None):
request = urllib2.Request(url = url)
+ request.add_header('Referer', 'https://wx.qq.com/')
+ if api == 'webwxgetvoice': request.add_header('Range', 'bytes=0-')
+ if api == 'webwxgetvideo': request.add_header('Range', 'bytes=0-')
response = urllib2.urlopen(request)
data = response.read()
+ logging.debug(url)
return data
def _post(self, url, params, jsonfmt = True):
@@ -590,7 +689,7 @@ class WebWeixin(object):
pm = re.search(key+'\s?=\s?"([^"<]+)"', content)
if pm: return pm.group(1)
elif fmat == 'xml':
- pm=re.search('<{0}>([^<]+){0}>'.format(key),content)
+ pm = re.search('<{0}>([^<]+){0}>'.format(key),content)
if pm: return pm.group(1)
return '未知'
@@ -615,5 +714,10 @@ if sys.stdout.encoding == 'cp936':
if __name__ == '__main__':
+
+ logger = logging.getLogger(__name__)
+ import coloredlogs
+ coloredlogs.install(level='DEBUG')
+
webwx = WebWeixin()
webwx.start()