mattuy/wxbot
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Python-3

Browse Source 
transfer to Python-3
This commit is contained in:
Haskely
2017-02-20 23:52:03 +08:00
parent 17a8a07d9b
commit 52699bbd01
1 changed files with 86 additions and 82 deletions
Download Patch File Download Diff File Expand all files Collapse all files
weixin.py
+86 -82
View File
@@ -1,9 +1,9 @@
#!/usr/bin/env python
# coding: utf-8
import qrcode
import urllib
import urllib2
import cookielib
import urllib.request, urllib.parse, urllib.error
import urllib.request, urllib.error, urllib.parse
import http.cookiejar
import requests
import xml.dom.minidom
import json
@@ -16,9 +16,9 @@ import random
import multiprocessing
import platform
import logging
import httplib
import http.client
from collections import defaultdict
from urlparse import urlparse
from urllib.parse import urlparse
from lxml import html
#import pdb
@@ -32,7 +32,7 @@ def catchKeyboardInterrupt(fn):
try:
return fn(*args)
except KeyboardInterrupt:
print '\n[*] 强制退出程序'
print('\n[*] 强制退出程序')
logging.debug('[*] 强制退出程序')
return wrapper
@@ -40,7 +40,7 @@ def catchKeyboardInterrupt(fn):
def _decode_list(data):
rv = []
for item in data:
if isinstance(item, unicode):
if isinstance(item, str):
item = item.encode('utf-8')
elif isinstance(item, list):
item = _decode_list(item)
@@ -52,10 +52,10 @@ def _decode_list(data):
def _decode_dict(data):
rv = {}
for key, value in data.iteritems():
if isinstance(key, unicode):
for key, value in data.items():
if isinstance(key, str):
key = key.encode('utf-8')
if isinstance(value, unicode):
if isinstance(value, str):
value = value.encode('utf-8')
elif isinstance(value, list):
value = _decode_list(value)
@@ -118,10 +118,10 @@ class WebWeixin(object):
self.TimeOut = 20 # 同步最短时间间隔（单位：秒）
self.media_count = -1
self.cookie = cookielib.CookieJar()
opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(self.cookie))
self.cookie = http.cookiejar.CookieJar()
opener = urllib.request.build_opener(urllib.request.HTTPCookieProcessor(self.cookie))
opener.addheaders = [('User-agent', self.user_agent)]
urllib2.install_opener(opener)
urllib.request.install_opener(opener)
def loadConfig(self, config):
if config['DEBUG']:
@@ -281,7 +281,7 @@ class WebWeixin(object):
PublicUsersList = self.PublicUsersList[:]
SpecialUsersList = self.SpecialUsersList[:]
for i in xrange(len(ContactList) - 1, -1, -1):
for i in range(len(ContactList) - 1, -1, -1):
Contact = ContactList[i]
if Contact['VerifyFlag'] & 8 != 0: # 公众号/服务号
ContactList.remove(Contact)
@@ -316,7 +316,7 @@ class WebWeixin(object):
ContactCount = dic['Count']
self.GroupList = ContactList
for i in xrange(len(ContactList) - 1, -1, -1):
for i in range(len(ContactList) - 1, -1, -1):
Contact = ContactList[i]
MemberList = Contact['MemberList']
for member in MemberList:
@@ -368,7 +368,7 @@ class WebWeixin(object):
'_': int(time.time()),
}
url = 'https://' + self.syncHost + \
'/cgi-bin/mmwebwx-bin/synccheck?' + urllib.urlencode(params)
'/cgi-bin/mmwebwx-bin/synccheck?' + urllib.parse.urlencode(params)
data = self._get(url)
if data == '':
return [-1,-1]
@@ -391,7 +391,7 @@ class WebWeixin(object):
if dic == '':
return None
if self.DEBUG:
print json.dumps(dic, indent=4)
print(json.dumps(dic, indent=4))
(json.dumps(dic, indent=4))
if dic['BaseResponse']['Ret'] == 0:
@@ -539,7 +539,7 @@ class WebWeixin(object):
r = requests.post(url, data=data, headers=headers)
dic = r.json()
if self.DEBUG:
print json.dumps(dic, indent=4)
print(json.dumps(dic, indent=4))
logging.debug(json.dumps(dic, indent=4))
return dic['BaseResponse']['Ret'] == 0
@@ -722,28 +722,28 @@ class WebWeixin(object):
msg['message'] = content
# 指定了消息内容
if 'message' in msg.keys():
if 'message' in list(msg.keys()):
content = msg['message']
if groupName != None:
print '%s |%s| %s -> %s: %s' % (message_id, groupName.strip(), srcName.strip(), dstName.strip(), content.replace('<br/>', '\n'))
print('%s |%s| %s -> %s: %s' % (message_id, groupName.strip(), srcName.strip(), dstName.strip(), content.replace('<br/>', '\n')))
logging.info('%s |%s| %s -> %s: %s' % (message_id, groupName.strip(),
srcName.strip(), dstName.strip(), content.replace('<br/>', '\n')))
else:
print '%s %s -> %s: %s' % (message_id, srcName.strip(), dstName.strip(), content.replace('<br/>', '\n'))
print('%s %s -> %s: %s' % (message_id, srcName.strip(), dstName.strip(), content.replace('<br/>', '\n')))
logging.info('%s %s -> %s: %s' % (message_id, srcName.strip(),
dstName.strip(), content.replace('<br/>', '\n')))
def handleMsg(self, r):
for msg in r['AddMsgList']:
print '[*] 你有新的消息，请注意查收'
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
print('[*] 该消息已储存到文件: ' + fn)
logging.debug('[*] 该消息已储存到文件: %s' % (fn))
msgType = msg['MsgType']
@@ -754,13 +754,17 @@ class WebWeixin(object):
if msgType == 1:
raw_msg = {'raw_msg': msg}
self._showMsg(raw_msg)
#自己加的代码-------------------------------------------#
#if self.autoReplyRevokeMode:
# store
#自己加的代码-------------------------------------------#
if self.autoReplyMode:
ans = self._xiaodoubi(content) + '\n[微信机器人自动回复]'
if self.webwxsendmsg(ans, msg['FromUserName']):
print '自动回复: ' + ans
print('自动回复: ' + ans)
logging.info('自动回复: ' + ans)
else:
print '自动回复失败'
print('自动回复失败')
logging.info('自动回复失败')
elif msgType == 3:
image = self.webwxgetmsgimg(msgid)
@@ -776,13 +780,13 @@ class WebWeixin(object):
self._safe_open(voice)
elif msgType == 42:
info = msg['RecommendInfo']
print '%s 发送了一张名片:' % name
print '========================='
print '= 昵称: %s' % info['NickName']
print '= 微信号: %s' % info['Alias']
print '= 地区: %s %s' % (info['Province'], info['City'])
print '= 性别: %s' % ['未知', '男', '女'][info['Sex']]
print '========================='
print('%s 发送了一张名片:' % name)
print('=========================')
print('= 昵称: %s' % info['NickName'])
print('= 微信号: %s' % info['Alias'])
print('= 地区: %s %s' % (info['Province'], info['City']))
print('= 性别: %s' % ['未知', '男', '女'][info['Sex']])
print('=========================')
raw_msg = {'raw_msg': msg, 'message': '%s 发送了一张名片: %s' % (
name.strip(), json.dumps(info))}
self._showMsg(raw_msg)
@@ -795,13 +799,13 @@ class WebWeixin(object):
elif msgType == 49:
appMsgType = defaultdict(lambda: "")
appMsgType.update({5: '链接', 3: '音乐', 7: '微博'})
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 '========================='
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('=========================')
card = {
'title': msg['FileName'],
'description': self._searchContent('des', content, 'xml'),
@@ -831,7 +835,7 @@ class WebWeixin(object):
self._showMsg(raw_msg)
def listenMsgMode(self):
print '[*] 进入消息监听模式 ... 成功'
print('[*] 进入消息监听模式 ... 成功')
logging.debug('[*] 进入消息监听模式 ... 成功')
self._run('[*] 进行同步线路测试 ... ', self.testsynccheck)
playWeChat = 0
@@ -840,14 +844,14 @@ class WebWeixin(object):
self.lastCheckTs = time.time()
[retcode, selector] = self.synccheck()
if self.DEBUG:
print 'retcode: %s, selector: %s' % (retcode, selector)
print('retcode: %s, selector: %s' % (retcode, selector))
logging.debug('retcode: %s, selector: %s' % (retcode, selector))
if retcode == '1100':
print '[*] 你在手机上登出了微信，债见'
print('[*] 你在手机上登出了微信，债见')
logging.debug('[*] 你在手机上登出了微信，债见')
break
if retcode == '1101':
print '[*] 你在其他地方登录了 WEB 版微信，债见'
print('[*] 你在其他地方登录了 WEB 版微信，债见')
logging.debug('[*] 你在其他地方登录了 WEB 版微信，债见')
break
elif retcode == '0':
@@ -858,11 +862,11 @@ class WebWeixin(object):
elif selector == '6':
# TODO
redEnvelope += 1
print '[*] 收到疑似红包消息 %d 次' % redEnvelope
print('[*] 收到疑似红包消息 %d 次' % redEnvelope)
logging.debug('[*] 收到疑似红包消息 %d 次' % redEnvelope)
elif selector == '7':
playWeChat += 1
print '[*] 你在手机上玩微信被我发现了 %d 次' % playWeChat
print('[*] 你在手机上玩微信被我发现了 %d 次' % playWeChat)
logging.debug('[*] 你在手机上玩微信被我发现了 %d 次' % playWeChat)
r = self.webwxsync()
elif selector == '0':
@@ -879,19 +883,19 @@ class WebWeixin(object):
line = line.replace('\n', '')
self._echo('-> ' + name + ': ' + line)
if self.webwxsendmsg(line, id):
print ' [成功]'
print(' [成功]')
else:
print ' [失败]'
print(' [失败]')
time.sleep(1)
else:
if self.webwxsendmsg(word, id):
print '[*] 消息发送成功'
print('[*] 消息发送成功')
logging.debug('[*] 消息发送成功')
else:
print '[*] 消息发送失败'
print('[*] 消息发送失败')
logging.debug('[*] 消息发送失败')
else:
print '[*] 此用户不存在'
print('[*] 此用户不存在')
logging.debug('[*] 此用户不存在')
def sendMsgToAll(self, word):
@@ -901,9 +905,9 @@ class WebWeixin(object):
id = contact['UserName']
self._echo('-> ' + name + ': ' + word)
if self.webwxsendmsg(word, id):
print ' [成功]'
print(' [成功]')
else:
print ' [失败]'
print(' [失败]')
time.sleep(1)
def sendImg(self, name, file_name):
@@ -925,18 +929,18 @@ class WebWeixin(object):
@catchKeyboardInterrupt
def start(self):
self._echo('[*] 微信网页版 ... 开动')
print
print()
logging.debug('[*] 微信网页版 ... 开动')
while True:
self._run('[*] 正在获取 uuid ... ', self.getUUID)
self._echo('[*] 正在获取二维码 ... 成功')
print
print()
logging.debug('[*] 微信网页版 ... 开动')
self.genQRCode()
print '[*] 请使用微信扫描二维码以登录 ... '
print('[*] 请使用微信扫描二维码以登录 ... ')
if not self.waitForLogin():
continue
print '[*] 请在手机上点击确认以登录 ... '
print('[*] 请在手机上点击确认以登录 ... ')
if not self.waitForLogin(0):
continue
break
@@ -947,33 +951,33 @@ class WebWeixin(object):
self._run('[*] 获取联系人 ... ', self.webwxgetcontact)
self._echo('[*] 应有 %s 个联系人，读取到联系人 %d 个' %
(self.MemberCount, len(self.MemberList)))
print
print()
self._echo('[*] 共有 %d 个群 | %d 个直接联系人 | %d 个特殊账号 ｜ %d 公众号或服务号' % (len(self.GroupList),
len(self.ContactList), len(self.SpecialUsersList), len(self.PublicUsersList)))
print
print()
self._run('[*] 获取群 ... ', self.webwxbatchgetcontact)
logging.debug('[*] 微信网页版 ... 开动')
if self.DEBUG:
print self
print(self)
logging.debug(self)
if self.interactive and raw_input('[*] 是否开启自动回复模式(y/n): ') == 'y':
if self.interactive and input('[*] 是否开启自动回复模式(y/n): ') == 'y':
self.autoReplyMode = True
print '[*] 自动回复模式 ... 开启'
print('[*] 自动回复模式 ... 开启')
logging.debug('[*] 自动回复模式 ... 开启')
else:
print '[*] 自动回复模式 ... 关闭'
print('[*] 自动回复模式 ... 关闭')
logging.debug('[*] 自动回复模式 ... 关闭')
if sys.platform.startswith('win'):
import thread
thread.start_new_thread(self.listenMsgMode())
import _thread
_thread.start_new_thread(self.listenMsgMode())
else:
listenProcess = multiprocessing.Process(target=self.listenMsgMode)
listenProcess.start()
while True:
text = raw_input('')
text = input('')
if text == 'quit':
listenProcess.terminate()
print('[*] 退出微信')
@@ -989,15 +993,15 @@ class WebWeixin(object):
[name, file] = text[3:].split(':')
self.sendMsg(name, file, True)
elif text[:3] == 'f->':
print '发送文件'
print('发送文件')
logging.debug('发送文件')
elif text[:3] == 'i->':
print '发送图片'
print('发送图片')
[name, file_name] = text[3:].split(':')
self.sendImg(name, file_name)
logging.debug('发送图片')
elif text[:3] == 'e->':
print '发送表情'
print('发送表情')
[name, file_name] = text[3:].split(':')
self.sendEmotion(name, file_name)
logging.debug('发送表情')
@@ -1012,7 +1016,7 @@ class WebWeixin(object):
def _run(self, str, func, *args):
self._echo(str)
if func(*args):
print '成功'
print('成功')
logging.debug('%s... 成功' % (str))
else:
print('失败\n[*] 退出程序')
@@ -1028,7 +1032,7 @@ class WebWeixin(object):
for i in mat:
BLACK = '\033[40m \033[0m'
WHITE = '\033[47m \033[0m'
print ''.join([BLACK if j else WHITE for j in i])
print(''.join([BLACK if j else WHITE for j in i]))
def _str2qr(self, str):
print(str)
@@ -1046,29 +1050,29 @@ class WebWeixin(object):
if not data:
return data
result = None
if type(data) == unicode:
if type(data) == str:
result = data
elif type(data) == str:
result = data.decode('utf-8')
return result
def _get(self, url, api=None):
request = urllib2.Request(url=url)
request = urllib.request.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-')
try:
response = urllib2.urlopen(request)
response = urllib.request.urlopen(request)
data = response.read()
logging.debug(url)
return data
except urllib2.HTTPError, e:
except urllib.error.HTTPError as e:
logging.error('HTTPError = ' + str(e.code))
except urllib2.URLError, e:
except urllib.error.URLError as e:
logging.error('URLError = ' + str(e.reason))
except httplib.HTTPException, e:
except http.client.HTTPException as e:
logging.error('HTTPException')
except Exception:
import traceback
@@ -1077,24 +1081,24 @@ class WebWeixin(object):
def _post(self, url, params, jsonfmt=True):
if jsonfmt:
request = urllib2.Request(url=url, data=json.dumps(params))
request = urllib.request.Request(url=url, data=json.dumps(params))
request.add_header(
'ContentType', 'application/json; charset=UTF-8')
else:
request = urllib2.Request(url=url, data=urllib.urlencode(params))
request = urllib.request.Request(url=url, data=urllib.parse.urlencode(params))
try:
response = urllib2.urlopen(request)
response = urllib.request.urlopen(request)
data = response.read()
if jsonfmt:
return json.loads(data, object_hook=_decode_dict)
return data
except urllib2.HTTPError, e:
except urllib.error.HTTPError as e:
logging.error('HTTPError = ' + str(e.code))
except urllib2.URLError, e:
except urllib.error.URLError as e:
logging.error('URLError = ' + str(e.reason))
except httplib.HTTPException, e:
except http.client.HTTPException as e:
logging.error('HTTPException')
except Exception:
import traceback