i just come out and take a bubble, haha \O^O/
This commit is contained in:
urinx
+5
-1
@@ -94,4 +94,8 @@ ENV/
|
|||||||
.idea
|
.idea
|
||||||
|
|
||||||
*.pyc
|
*.pyc
|
||||||
saved/qrcodes/qrcode.jpg
|
wxbot_demo_py3/saved/
|
||||||
|
wxbot_project_py2.7/tmp_data/
|
||||||
|
wxbot_project_py2.7/wechat/wechat_js_backup/
|
||||||
|
wxbot_project_py2.7/config/wechat.conf
|
||||||
|
wxbot_project_py2.7/wiki/
|
||||||
|
|||||||
@@ -1,45 +1,42 @@
|
|||||||
# WeixinBot [](http://github.com/Urinx/WeixinBot) [](http://github.com/Urinx/WeixinBot/fork) 
|
# WeixinBot [](http://github.com/Urinx/WeixinBot) [](http://github.com/Urinx/WeixinBot/fork) 
|
||||||
|
|
||||||
网页版微信API，包含终端版微信及微信机器人
|
网页版微信API，包含终端版微信及微信机器人
|
||||||
|
|
||||||
## Demo
|
## Contents
|
||||||
为了运行 `weixin.py` 示例脚本，你需要有安装 `qrcode` 包，你可以通过 `pip install qrcode` 来安装。
|
* [Demo](#Demo)
|
||||||
|
* [Web Weixin Pipeline](#Web-Weixin-Pipeline)
|
||||||
|
* [Web Weixin API](#Web-Weixin-API)
|
||||||
|
* [Discussion Group](#Discussion-Group)
|
||||||
|
* [Recent Update](#Recent-Update)
|
||||||
|
|
||||||
|
## Demo
|
||||||
|
为了确保能正常运行示例脚本，请安装所需的第三方包。
|
||||||
|
|
||||||
|
```
|
||||||
|
pip install -r requirements.txt
|
||||||
|
```
|
||||||
|
|
||||||
|
注：下面演示的图片与功能可能不是最新的，具体请看源码。
|
||||||
|
|
||||||
|
<div align=center>
|
||||||
|
<img src="imgs/1.png" width="500" height="550"/>
|
||||||
|
</div>
|
||||||
|
|
||||||
按照操作指示在手机微信上扫描二维码然后登录，你可以选择是否开启自动回复模式。
|
按照操作指示在手机微信上扫描二维码然后登录，你可以选择是否开启自动回复模式。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
开启自动回复模式后，如果接收到的是文字消息就会自动回复，包括群消息。
|
开启自动回复模式后，如果接收到的是文字消息就会自动回复，包括群消息。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
现在，名片，链接，动画表情和地址位置消息都可以正常接收。
|
名片，链接，动画表情和地址位置消息。
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
**目前支持的命令**：
|
网页版上有的功能目前基本上都能支持。
|
||||||
|
|
||||||
`->[昵称或ID]:[内容]` 给好友发送消息
|
|
||||||
|
|
||||||
`m->[昵称或ID]:[文件路径]` 给好友发送文件中的内容
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
`f->[昵称或ID]:[文件路径]` 给好友发送文件
|
|
||||||
|
|
||||||
`i->[昵称或ID]:[图片路径]` 给好友发送图片
|
|
||||||
|
|
||||||
`e->[昵称或ID]:[文件路径]` 给好友发送表情(jpg/gif)
|
|
||||||
|
|
||||||
`quit` 退出程序
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
注意，以上命令均不包含方括号。
|
|
||||||
|
|
||||||
## Web Weixin Pipeline
|
## Web Weixin Pipeline
|
||||||
|
|
||||||
@@ -69,14 +66,7 @@
|
|||||||
|QRCode Scaned|
|
|QRCode Scaned|
|
||||||
+-------------+
|
+-------------+
|
||||||
```
|
```
|
||||||
## Discussion Group
|
|
||||||
如果你希望和 WeixinBot 的其他开发者交流，或者有什么问题和建议，欢迎大家加入微信群【Youth fed the dog】一起讨论。扫描下面的二维码添加机器人为好友，并回复【Aidog】获取入群链接。
|
|
||||||
|
|
||||||
<div align=center>
|
|
||||||
<img src="saved/qrcodes/groupQrcode.jpg" width="220" height="220" alt="join us"/>
|
|
||||||
</div>
|
|
||||||
|
|
||||||
注：这个不是群的二维码，是机器人拉你入群，记得回复机器人【Aidog】哦~ （secret code: Aidog）
|
|
||||||
|
|
||||||
## Web Weixin API
|
## Web Weixin API
|
||||||
|
|
||||||
@@ -95,12 +85,29 @@ window.QRLogin.code = 200; window.QRLogin.uuid = "xxx"
|
|||||||
```
|
```
|
||||||
> 注：这里的appid就是在微信开放平台注册的应用的AppID。网页版微信有两个AppID，早期的是`wx782c26e4c19acffb`，在微信客户端上显示为应用名称为`Web微信`；现在用的是`wxeb7ec651dd0aefa9`，显示名称为`微信网页版`。
|
> 注：这里的appid就是在微信开放平台注册的应用的AppID。网页版微信有两个AppID，早期的是`wx782c26e4c19acffb`，在微信客户端上显示为应用名称为`Web微信`；现在用的是`wxeb7ec651dd0aefa9`，显示名称为`微信网页版`。
|
||||||
|
|
||||||
|
<div align=center>
|
||||||
|
<img src="imgs/8.jpg" width="320" height="211"/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<br>
|
||||||
|
| API | 绑定登陆（webwxpushloginurl） |
|
||||||
|
| --- | --------- |
|
||||||
|
| url | https://wx.qq.com/cgi-bin/mmwebwx-bin/webwxpushloginurl |
|
||||||
|
| method | GET |
|
||||||
|
| params | **uin**: xxx |
|
||||||
|
|
||||||
|
返回数据(String):
|
||||||
|
```
|
||||||
|
{'msg': 'all ok', 'uuid': 'xxx', 'ret': '0'}
|
||||||
|
|
||||||
|
通过这种方式可以省掉扫二维码这步操作，更加方便
|
||||||
|
```
|
||||||
<br>
|
<br>
|
||||||
|
|
||||||
| API | 生成二维码 |
|
| API | 生成二维码 |
|
||||||
| --- | --------- |
|
| --- | --------- |
|
||||||
| url | https://login.weixin.qq.com/l/ `uuid` |
|
| url | https://login.weixin.qq.com/l/ `uuid` |
|
||||||
|
| method | GET |
|
||||||
<br>
|
<br>
|
||||||
|
|
||||||
| API | 二维码扫描登录 |
|
| API | 二维码扫描登录 |
|
||||||
@@ -299,7 +306,7 @@ window.redirect_uri="https://wx.qq.com/cgi-bin/mmwebwx-bin/webwxnewloginpage?tic
|
|||||||
| API | synccheck |
|
| API | synccheck |
|
||||||
| --- | --------- |
|
| --- | --------- |
|
||||||
| protocol | https |
|
| protocol | https |
|
||||||
| host | webpush.weixin.qq.com <br> webpush2.weixin.qq.com <br> webpush.wechat.com <br> webpush1.wechat.com <br> webpush2.wechat.com <br> webpush.wechatapp.com <br> webpush1.wechatapp.com |
|
| host | webpush.weixin.qq.com <br> webpush.wx2.qq.com <br> webpush.wx8.qq.com <br> webpush.wx.qq.com <br> webpush.web2.wechat.com <br> webpush.web.wechat.com |
|
||||||
| path | /cgi-bin/mmwebwx-bin/synccheck |
|
| path | /cgi-bin/mmwebwx-bin/synccheck |
|
||||||
| method | GET |
|
| method | GET |
|
||||||
| data | URL Encode |
|
| data | URL Encode |
|
||||||
@@ -396,6 +403,24 @@ selector:
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
| API | webwxrevokemsg |
|
||||||
|
| --- | ------------ |
|
||||||
|
| url | https://wx.qq.com/cgi-bin/mmwebwx-bin/webwxrevokemsg |
|
||||||
|
| method | POST |
|
||||||
|
| data | JSON |
|
||||||
|
| header | ContentType: application/json; charset=UTF-8 |
|
||||||
|
| params | { <br> BaseRequest: { Uin: xxx, Sid: xxx, Skey: xxx, DeviceID: xxx }, <br> SvrMsgId: msg_id, <br> ToUserName: user_id, <br> ClientMsgId: local_msg_id <br> } |
|
||||||
|
|
||||||
|
返回数据(JSON):
|
||||||
|
```
|
||||||
|
{
|
||||||
|
"BaseResponse": {
|
||||||
|
"Ret": 0,
|
||||||
|
"ErrMsg": ""
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
#### 发送表情
|
#### 发送表情
|
||||||
|
|
||||||
| API | webwxsendmsgemotion |
|
| API | webwxsendmsgemotion |
|
||||||
@@ -497,10 +522,10 @@ selector:
|
|||||||
| 1 | 文本消息 |
|
| 1 | 文本消息 |
|
||||||
| 3 | 图片消息 |
|
| 3 | 图片消息 |
|
||||||
| 34 | 语音消息 |
|
| 34 | 语音消息 |
|
||||||
| 37 | VERIFYMSG |
|
| 37 | 好友确认消息 |
|
||||||
| 40 | POSSIBLEFRIEND_MSG |
|
| 40 | POSSIBLEFRIEND_MSG |
|
||||||
| 42 | 共享名片 |
|
| 42 | 共享名片 |
|
||||||
| 43 | 视频通话消息 |
|
| 43 | 视频消息 |
|
||||||
| 47 | 动画表情 |
|
| 47 | 动画表情 |
|
||||||
| 48 | 位置消息 |
|
| 48 | 位置消息 |
|
||||||
| 49 | 分享链接 |
|
| 49 | 分享链接 |
|
||||||
@@ -757,13 +782,17 @@ Content:
|
|||||||
"收到红包，请在手机上查看"
|
"收到红包，请在手机上查看"
|
||||||
```
|
```
|
||||||
|
|
||||||
持续更新中 ...
|
|
||||||
|
|
||||||
## Todo
|
## Discussion Group
|
||||||
- [x] 发送图片或者文件功能
|
如果你希望和 WeixinBot 的其他开发者交流，或者有什么问题和建议，欢迎大家加入微信群【Youth fed the dog】一起讨论。扫描下面的二维码添加机器人为好友，并回复【Aidog】获取入群链接。
|
||||||
- [ ] 主动给群聊发送消息
|
|
||||||
- [ ] 建立群聊
|
|
||||||
- [x] 群发消息
|
|
||||||
- [ ] 补充更多的接口及完善文档
|
|
||||||
|
|
||||||
P.S. 还有啥要补充的也可以在[issue #8](https://github.com/Urinx/WeixinBot/issues/8)下留言
|
<div align=center>
|
||||||
|
<img src="imgs/groupQrcode.jpg" width="220" height="220" alt="join us"/>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
注：这个不是群的二维码，是机器人拉你入群，记得回复机器人【Aidog】哦~ （secret code: Aidog）
|
||||||
|
|
||||||
|
## Recent Update
|
||||||
|
|
||||||
|
- association_login
|
||||||
|
目前网页版微信已经可以脱离扫码，但是依然需要在客户端进行确认登录。
|
||||||
|
|||||||
@@ -1,2 +0,0 @@
|
|||||||
#!/bin/sh
|
|
||||||
pip freeze |grep -v wheel | gawk -F"==" ' { print $1 } ' > requirements.txt
|
|
||||||
|
Before Width: | Height: | Size: 304 KiB After Width: | Height: | Size: 304 KiB |
|
Before Width: | Height: | Size: 80 KiB After Width: | Height: | Size: 80 KiB |
|
Before Width: | Height: | Size: 48 KiB After Width: | Height: | Size: 48 KiB |
|
Before Width: | Height: | Size: 46 KiB After Width: | Height: | Size: 46 KiB |
|
Before Width: | Height: | Size: 72 KiB After Width: | Height: | Size: 72 KiB |
BIN
Binary file not shown.
|
After Width: | Height: | Size: 24 KiB |
|
Before Width: | Height: | Size: 42 KiB After Width: | Height: | Size: 42 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 40 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 215 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 12 KiB |
Binary file not shown.
|
Before Width: | Height: | Size: 24 KiB |
@@ -0,0 +1,37 @@
|
|||||||
|
# wxbot_project_py2.7
|
||||||
|
|
||||||
|
目录结构:
|
||||||
|
```bash
|
||||||
|
.
|
||||||
|
├── README.md
|
||||||
|
├── config
|
||||||
|
│ ├── __init__.py
|
||||||
|
│ ├── config_manager.py
|
||||||
|
│ ├── constant.py
|
||||||
|
│ ├── log.py
|
||||||
|
│ ├── requirements.txt
|
||||||
|
│ └── wechat.conf.bak
|
||||||
|
├── db
|
||||||
|
│ ├── __init__.py
|
||||||
|
│ ├── mysql_db.py
|
||||||
|
│ └── sqlite_db.py
|
||||||
|
├── docker
|
||||||
|
│ ├── Dockerfile
|
||||||
|
│ └── README.md
|
||||||
|
├── server
|
||||||
|
│ ├── index.html
|
||||||
|
│ └── upload.html
|
||||||
|
├── tmp_data
|
||||||
|
├── wechat
|
||||||
|
│ ├── __init__.py
|
||||||
|
│ ├── utils.py
|
||||||
|
│ ├── wechat.py
|
||||||
|
│ ├── wechat_apis.py
|
||||||
|
│ └── wechat_js_backup
|
||||||
|
│ └── index_40649b7.js
|
||||||
|
├── weixin_bot.py
|
||||||
|
└── wx_handler
|
||||||
|
├── __init__.py
|
||||||
|
├── bot.py
|
||||||
|
└── wechat_msg_processor.py
|
||||||
|
```
|
||||||
@@ -0,0 +1,6 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
from config_manager import ConfigManager
|
||||||
|
from constant import Constant
|
||||||
|
from log import Log
|
||||||
@@ -0,0 +1,105 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from constant import Constant
|
||||||
|
#---------------------------------------------------
|
||||||
|
import ConfigParser
|
||||||
|
import os
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
|
||||||
|
class ConfigManager(object):
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.config = Constant.WECHAT_CONFIG_FILE
|
||||||
|
self.cp = ConfigParser.ConfigParser()
|
||||||
|
self.cp.read(self.config)
|
||||||
|
|
||||||
|
data_dir = self.get('setting', 'prefix')
|
||||||
|
upload_dir = self.getpath('uploaddir')
|
||||||
|
if not os.path.exists(data_dir):
|
||||||
|
os.makedirs(data_dir)
|
||||||
|
if not os.path.exists(upload_dir):
|
||||||
|
os.makedirs(upload_dir)
|
||||||
|
|
||||||
|
def get(self, section, option):
|
||||||
|
return self.cp.get(section, option)
|
||||||
|
|
||||||
|
def set(self, section, option, value):
|
||||||
|
self.cp.set(section, option, value)
|
||||||
|
self.cp.write(open(self.config, 'w'))
|
||||||
|
|
||||||
|
def getpath(self, dir):
|
||||||
|
prefix = self.get('setting', 'prefix')
|
||||||
|
return prefix + self.get('setting', dir)
|
||||||
|
|
||||||
|
def setup_database(self):
|
||||||
|
path = self.get('setting', 'prefix')
|
||||||
|
conf = [
|
||||||
|
path + self.get('setting', 'uploaddir'),
|
||||||
|
path + self.get('setting', 'datadir'),
|
||||||
|
path + self.get('setting', 'logdir'),
|
||||||
|
]
|
||||||
|
return conf
|
||||||
|
|
||||||
|
def set_wechat_config(self, conf):
|
||||||
|
for [key, value] in conf.items():
|
||||||
|
self.cp.set('wechat', key, value)
|
||||||
|
self.cp.write(open(self.config, 'w'))
|
||||||
|
|
||||||
|
def get_wechat_config(self):
|
||||||
|
uin = self.cp.get('wechat', 'uin')
|
||||||
|
last_login = self.cp.get('wechat', 'last_login')
|
||||||
|
conf = [
|
||||||
|
self.cp.get('wechat', 'uuid'),
|
||||||
|
self.cp.get('wechat', 'redirect_uri'),
|
||||||
|
int(uin if uin else 0),
|
||||||
|
self.cp.get('wechat', 'sid'),
|
||||||
|
self.cp.get('wechat', 'skey'),
|
||||||
|
self.cp.get('wechat', 'pass_ticket'),
|
||||||
|
self.cp.get('wechat', 'synckey'),
|
||||||
|
self.cp.get('wechat', 'device_id'),
|
||||||
|
float(last_login if last_login else 0),
|
||||||
|
]
|
||||||
|
return conf
|
||||||
|
|
||||||
|
def get_wechat_media_dir(self):
|
||||||
|
prefix = self.get('setting', 'prefix')
|
||||||
|
path = prefix + self.cp.get('setting', 'mediapath')
|
||||||
|
return {
|
||||||
|
'webwxgeticon': path + '/icons',
|
||||||
|
'webwxgetheadimg': path + '/headimgs',
|
||||||
|
'webwxgetmsgimg': path + '/msgimgs',
|
||||||
|
'webwxgetvideo': path + '/videos',
|
||||||
|
'webwxgetvoice': path + '/voices',
|
||||||
|
'_showQRCodeImg': path + '/qrcodes',
|
||||||
|
}
|
||||||
|
|
||||||
|
def get_pickle_files(self):
|
||||||
|
prefix = self.get('setting', 'prefix')
|
||||||
|
return {
|
||||||
|
'User': prefix + self.get('setting', 'contact_user'),
|
||||||
|
'MemberList': prefix + self.get('setting', 'contact_member_list'),
|
||||||
|
'GroupList': prefix + self.get('setting', 'contact_group_list'),
|
||||||
|
'GroupMemeberList': prefix + self.get('setting', 'contact_group_memeber_list'),
|
||||||
|
'SpecialUsersList': prefix + self.get('setting', 'contact_special_users_list'),
|
||||||
|
}
|
||||||
|
|
||||||
|
def get_cookie(self):
|
||||||
|
prefix = self.get('setting', 'prefix')
|
||||||
|
path = prefix + self.get('setting', 'cookie')
|
||||||
|
basedir = os.path.dirname(path)
|
||||||
|
if not os.path.exists(basedir):
|
||||||
|
os.makedirs(basedir)
|
||||||
|
return path
|
||||||
|
|
||||||
|
def mysql(self):
|
||||||
|
mysql = {
|
||||||
|
'host': self.get('mysql', 'host'),
|
||||||
|
'port': self.cp.getint('mysql', 'port'),
|
||||||
|
'user': self.get('mysql', 'user'),
|
||||||
|
'passwd': self.get('mysql', 'passwd'),
|
||||||
|
'database': self.get('mysql', 'database'),
|
||||||
|
}
|
||||||
|
return mysql
|
||||||
@@ -0,0 +1,215 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
import time
|
||||||
|
|
||||||
|
class Constant(object):
|
||||||
|
"""
|
||||||
|
@brief All used constants are listed here
|
||||||
|
"""
|
||||||
|
|
||||||
|
WECHAT_CONFIG_FILE = 'config/wechat.conf'
|
||||||
|
LOGGING_LOGGER_NAME = 'WeChat'
|
||||||
|
|
||||||
|
QRCODE_BLACK = '\033[40m \033[0m'
|
||||||
|
QRCODE_WHITE = '\033[47m \033[0m'
|
||||||
|
|
||||||
|
HTTP_HEADER_USERAGENT = [('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')]
|
||||||
|
HTTP_HEADER_CONTENTTYPE = ['ContentType', 'application/json; charset=UTF-8']
|
||||||
|
HTTP_HEADER_CONNECTION = ['Connection', 'keep-alive']
|
||||||
|
HTTP_HEADER_REFERER = ['Referer', 'https://wx.qq.com/']
|
||||||
|
HTTP_HEADER_RANGE = ['Range', 'bytes=0-']
|
||||||
|
|
||||||
|
REGEX_EMOJI = r'<span class="emoji emoji(\w+)"></span>'
|
||||||
|
|
||||||
|
SERVER_LOG_FORMAT = '%(asctime)s - %(pathname)s:%(lineno)d - %(name)s - %(levelname)s - %(message)s'
|
||||||
|
SERVER_UPLOAD_ALLOWED_EXTENSIONS = set(['txt', 'pdf', 'png', 'jpg', 'jpeg', 'gif'])
|
||||||
|
SERVER_PAGE_UPLOAD = 'Server/upload.html'
|
||||||
|
SERVER_PAGE_INDEX = 'Server/index.html'
|
||||||
|
|
||||||
|
RUN_RESULT_SUCCESS = '成功 %ds\n'
|
||||||
|
RUN_RESULT_FAIL = '失败\n[*] 退出程序\n'
|
||||||
|
MAIN_RESTART = '[*] wait for restart'
|
||||||
|
LOG_MSG_FILE = 'WeChat-Msgs-%Y-%m-%d.json'
|
||||||
|
LOG_MSG_GROUP_LIST_FILE = 'group_list.json'
|
||||||
|
LOG_MSG_QUIT = '\n[*] Force quit.\n'
|
||||||
|
LOG_MSG_FAIL = '失败\n'
|
||||||
|
LOG_MSG_SUCCESS = '成功\n'
|
||||||
|
LOG_MSG_START = '[*] 微信网页版 ... 开动\n'
|
||||||
|
LOG_MSG_RECOVER = '[*] 从配置文件中恢复 ... '
|
||||||
|
LOG_MSG_RECOVER_CONTACT = '[*] 从文件中恢复联系人数据 ... '
|
||||||
|
LOG_MSG_TRY_INIT = '[*] 尝试初始化 ... '
|
||||||
|
LOG_MSG_ASSOCIATION_LOGIN = '[*] 通过关联登录 ... '
|
||||||
|
LOG_MSG_GET_UUID = '[*] 正在获取 uuid ... '
|
||||||
|
LOG_MSG_GET_QRCODE = '[*] 正在获取二维码 ... 成功\n'
|
||||||
|
LOG_MSG_SCAN_QRCODE = '[*] 请使用微信扫描二维码以登录 ... \n'
|
||||||
|
LOG_MSG_CONFIRM_LOGIN = '[*] 请在手机上点击确认以登录 ... \n'
|
||||||
|
LOG_MSG_WAIT_LOGIN_ERR1 = '[登陆超时] \n'
|
||||||
|
LOG_MSG_WAIT_LOGIN_ERR2 = '[登陆异常] \n'
|
||||||
|
LOG_MSG_LOGIN = '[*] 正在登录 ... '
|
||||||
|
LOG_MSG_INIT = '[*] 微信初始化 ... '
|
||||||
|
LOG_MSG_STATUS_NOTIFY = '[*] 开启状态通知 ... '
|
||||||
|
LOG_MSG_GET_CONTACT = '[*] 获取联系人 ... '
|
||||||
|
LOG_MSG_CONTACT_COUNT = '[*] 应有 %s 个联系人，读取到联系人 %d 个\n'
|
||||||
|
LOG_MSG_OTHER_CONTACT_COUNT = '[*] 共有 %d 个群 | %d 个直接联系人 | %d 个特殊账号 ｜ %d 公众号或服务号\n'
|
||||||
|
LOG_MSG_GET_GROUP_MEMBER = '[*] 拉取群聊成员 ... '
|
||||||
|
LOG_MSG_SNAPSHOT = '[*] 保存配置 ... '
|
||||||
|
LOG_MSG_LOGOUT = '[*] 你在手机上登出了微信\n'
|
||||||
|
LOG_MSG_LOGIN_OTHERWHERE = '[*] 你在其他地方登录了 WEB 版微信\n'
|
||||||
|
LOG_MSG_QUIT_ON_PHONE = '[*] 你在手机上主动退出了\n'
|
||||||
|
LOG_MSG_RUNTIME = '[*] Total run: %s\n'
|
||||||
|
LOG_MSG_KILL_PROCESS = 'kill %d'
|
||||||
|
LOG_MSG_NEW_MSG = '>>> %d 条新消息\n'
|
||||||
|
LOG_MSG_LOCATION = '[位置] %s'
|
||||||
|
LOG_MSG_PICTURE = '[图片] %s'
|
||||||
|
LOG_MSG_VOICE = '[语音] %s'
|
||||||
|
LOG_MSG_RECALL = '撤回了一条消息'
|
||||||
|
LOG_MSG_ADD_FRIEND = '%s 请求添加你为好友'
|
||||||
|
LOG_MSG_UNKNOWN_MSG = '[*] 该消息类型为: %d，内容: %s'
|
||||||
|
LOG_MSG_VIDEO = '[小视频] %s'
|
||||||
|
LOG_MSG_NOTIFY_PHONE = '[*] 提示手机网页版微信登录状态\n'
|
||||||
|
LOG_MSG_EMOTION = '[表情] %s'
|
||||||
|
LOG_MSG_NAME_CARD = (
|
||||||
|
'[名片]\n'
|
||||||
|
'=========================\n'
|
||||||
|
'= 昵称: %s\n'
|
||||||
|
'= 微信号: %s\n'
|
||||||
|
'= 地区: %s %s\n'
|
||||||
|
'= 性别: %s\n'
|
||||||
|
'========================='
|
||||||
|
)
|
||||||
|
LOG_MSG_SEX_OPTION = ['未知', '男', '女']
|
||||||
|
LOG_MSG_APP_LINK = (
|
||||||
|
'[%s]\n'
|
||||||
|
'=========================\n'
|
||||||
|
'= 标题: %s\n'
|
||||||
|
'= 描述: %s\n'
|
||||||
|
'= 链接: %s\n'
|
||||||
|
'= 来自: %s\n'
|
||||||
|
'========================='
|
||||||
|
)
|
||||||
|
LOG_MSG_APP_LINK_TYPE = {5: '链接', 3: '音乐', 7: '微博'}
|
||||||
|
LOG_MSG_APP_IMG = (
|
||||||
|
'[图片]\n'
|
||||||
|
'=========================\n'
|
||||||
|
'= 文件: %s\n'
|
||||||
|
'= 来自: %s\n'
|
||||||
|
'========================='
|
||||||
|
)
|
||||||
|
LOG_MSG_SYSTEM = '系统消息'
|
||||||
|
LOG_MSG_UNKNOWN_NAME = '未知_'
|
||||||
|
LOG_MSG_UNKNOWN_GROUP_NAME = '未知群_'
|
||||||
|
|
||||||
|
TABLE_GROUP_MSG_LOG = 'WeChatRoomMessage'
|
||||||
|
TABLE_GROUP_MSG_LOG_COL = """
|
||||||
|
MsgID text,
|
||||||
|
RoomOwnerID text,
|
||||||
|
RoomName text,
|
||||||
|
UserCount text,
|
||||||
|
FromUserName text,
|
||||||
|
ToUserName text,
|
||||||
|
AttrStatus text,
|
||||||
|
DisplayName text,
|
||||||
|
Name text,
|
||||||
|
MsgType text,
|
||||||
|
FaceMsg text,
|
||||||
|
TextMsg text,
|
||||||
|
ImageMsg text,
|
||||||
|
VideoMsg text,
|
||||||
|
SoundMsg text,
|
||||||
|
LinkMsg text,
|
||||||
|
NameCardMsg text,
|
||||||
|
LocationMsg text,
|
||||||
|
RecallMsgID text,
|
||||||
|
SysMsg text,
|
||||||
|
MsgTime text,
|
||||||
|
MsgTimestamp text
|
||||||
|
"""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def TABLE_GROUP_LIST():
|
||||||
|
return 'WeChatRoom_' + time.strftime('%Y%m%d', time.localtime())
|
||||||
|
|
||||||
|
TABLE_GROUP_LIST_COL = """
|
||||||
|
RoomName text,
|
||||||
|
RoomID text,
|
||||||
|
RoomOwnerID text,
|
||||||
|
UserCount text,
|
||||||
|
RoomIcon text
|
||||||
|
"""
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def TABLE_GROUP_USER_LIST():
|
||||||
|
return 'WeChatRoomMember_' + time.strftime('%Y%m%d', time.localtime())
|
||||||
|
|
||||||
|
TABLE_GROUP_USER_LIST_COL = """
|
||||||
|
RoomID text,
|
||||||
|
MemberID text,
|
||||||
|
MemberNickName text,
|
||||||
|
MemberDisplayName text,
|
||||||
|
MemberAttrStatus text
|
||||||
|
"""
|
||||||
|
TABLE_RECORD_ENTER_GROUP = 'WeChatEnterGroupRecord'
|
||||||
|
TABLE_RECORD_ENTER_GROUP_COL = """
|
||||||
|
MsgID text,
|
||||||
|
RoomName text,
|
||||||
|
FromUserName text,
|
||||||
|
ToUserName text,
|
||||||
|
Name text,
|
||||||
|
EnterTime text
|
||||||
|
"""
|
||||||
|
TABLE_RECORD_RENAME_GROUP = 'WeChatRenameGroupRecord'
|
||||||
|
TABLE_RECORD_RENAME_GROUP_COL = """
|
||||||
|
MsgID text,
|
||||||
|
FromName text,
|
||||||
|
ToName text,
|
||||||
|
ModifyPeople text,
|
||||||
|
ModifyTime text
|
||||||
|
"""
|
||||||
|
|
||||||
|
API_APPID = 'wx782c26e4c19acffb'
|
||||||
|
API_WXAPPID = 'wx299208e619de7026' # Weibo
|
||||||
|
# 'wxeb7ec651dd0aefa9' # Weixin
|
||||||
|
API_LANG = 'zh_CN'
|
||||||
|
API_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'
|
||||||
|
)
|
||||||
|
API_SPECIAL_USER = [
|
||||||
|
'newsapp', 'filehelper', 'weibo', 'qqmail',
|
||||||
|
'fmessage', 'tmessage', 'qmessage', 'qqsync',
|
||||||
|
'floatbottle', 'lbsapp', 'shakeapp', 'medianote',
|
||||||
|
'qqfriend', 'readerapp', 'blogapp', 'facebookapp',
|
||||||
|
'masssendapp', 'meishiapp', 'feedsapp', 'voip',
|
||||||
|
'blogappweixin', 'brandsessionholder', 'weixin',
|
||||||
|
'weixinreminder', 'officialaccounts', 'wxitil',
|
||||||
|
'notification_messages', 'wxid_novlwrv3lqwv11',
|
||||||
|
'gh_22b87fa7cb3c', 'userexperience_alarm',
|
||||||
|
]
|
||||||
|
|
||||||
|
EMOTICON = [
|
||||||
|
'[Smile]', '[Grimace]', '[Drool]', '[Scowl]', '[CoolGuy]', '[Sob]', '[Shy]',
|
||||||
|
'[Silent]', '[Sleep]', '[Cry]', '[Awkward]', '[Angry]', '[Tongue]', '[Grin]',
|
||||||
|
'[Surprise]', '[Frown]', '[Ruthless]', '[Blush]', '[Scream]', '[Puke]',
|
||||||
|
'[Chuckle]', '[Joyful]', '[Slight]', '[Smug]', '[Hungry]', '[Drowsy]', '[Panic]',
|
||||||
|
'[Sweat]', '[Laugh]', '[Commando]', '[Determined]', '[Scold]', '[Shocked]', '[Shhh]',
|
||||||
|
'[Dizzy]', '[Tormented]', '[Toasted]', '[Skull]', '[Hammer]', '[Wave]',
|
||||||
|
'[Relief]', '[DigNose]', '[Clap]', '[Shame]', '[Trick]',' [Bah！L]','[Bah！R]',
|
||||||
|
'[Yawn]', '[Lookdown]', '[Wronged]', '[Puling]', '[Sly]', '[Kiss]', '[Uh-oh]',
|
||||||
|
'[Whimper]', '[Cleaver]', '[Melon]', '[Beer]', '[Basketball]', '[PingPong]',
|
||||||
|
'[Coffee]', '[Rice]', '[Pig]', '[Rose]', '[Wilt]', '[Lip]', '[Heart]',
|
||||||
|
'[BrokenHeart]', '[Cake]', '[Lightning]', '[Bomb]', '[Dagger]', '[Soccer]', '[Ladybug]',
|
||||||
|
'[Poop]', '[Moon]', '[Sun]', '[Gift]', '[Hug]', '[Strong]',
|
||||||
|
'[Weak]', '[Shake]', '[Victory]', '[Admire]', '[Beckon]', '[Fist]', '[Pinky]',
|
||||||
|
'[Love]', '[No]', '[OK]', '[InLove]', '[Blowkiss]', '[Waddle]', '[Tremble]',
|
||||||
|
'[Aaagh!]', '[Twirl]', '[Kotow]', '[Lookback]', '[Jump]', '[Give-in]',
|
||||||
|
u'\U0001f604', u'\U0001f637', u'\U0001f639', u'\U0001f61d', u'\U0001f632', u'\U0001f633',
|
||||||
|
u'\U0001f631', u'\U0001f64d', u'\U0001f609', u'\U0001f60c', u'\U0001f612', u'\U0001f47f',
|
||||||
|
u'\U0001f47b', u'\U0001f49d', u'\U0001f64f', u'\U0001f4aa', u'\U0001f4b5', u'\U0001f382',
|
||||||
|
u'\U0001f388', u'\U0001f4e6',
|
||||||
|
]
|
||||||
|
BOT_ZHIHU_URL_LATEST = 'http://news-at.zhihu.com/api/4/news/latest'
|
||||||
|
BOT_ZHIHU_URL_DAILY = 'http://daily.zhihu.com/story/'
|
||||||
|
BOT_TULING_API_KEY = '55e7f30895a0a10535984bae5ad294d1'
|
||||||
|
BOT_TULING_API_URL = 'http://www.tuling123.com/openapi/api?key=%s&info=%s&userid=%s'
|
||||||
|
BOT_TULING_BOT_REPLY = u'麻烦说的清楚一点，我听不懂你在说什么'
|
||||||
@@ -0,0 +1,23 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from constant import Constant
|
||||||
|
from config import ConfigManager
|
||||||
|
#---------------------------------------------------
|
||||||
|
import logging
|
||||||
|
import logging.config
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
cm = ConfigManager()
|
||||||
|
|
||||||
|
logging.config.fileConfig(Constant.WECHAT_CONFIG_FILE)
|
||||||
|
# create logger
|
||||||
|
Log = logging.getLogger(Constant.LOGGING_LOGGER_NAME)
|
||||||
|
|
||||||
|
# 'application' code
|
||||||
|
# Log.debug('debug message')
|
||||||
|
# Log.info('info message')
|
||||||
|
# Log.warn('warn message')
|
||||||
|
# Log.error('error message')
|
||||||
|
# Log.critical('critical message')
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
qrcode
|
||||||
|
flask
|
||||||
|
requests
|
||||||
|
requests_toolbelt
|
||||||
|
pymysql
|
||||||
@@ -0,0 +1,72 @@
|
|||||||
|
[wechat]
|
||||||
|
host = wx.qq.com
|
||||||
|
uuid =
|
||||||
|
redirect_uri =
|
||||||
|
uin =
|
||||||
|
sid =
|
||||||
|
skey =
|
||||||
|
pass_ticket =
|
||||||
|
device_id =
|
||||||
|
last_login =
|
||||||
|
|
||||||
|
[setting]
|
||||||
|
prefix = tmp_data/
|
||||||
|
database = WeChat.db
|
||||||
|
datadir = Data/infos/
|
||||||
|
logdir = Logs
|
||||||
|
mediapath = Data
|
||||||
|
uploaddir = Data/upload
|
||||||
|
qrcodedir = Data/qrcode
|
||||||
|
server_port = 8080
|
||||||
|
cookie = Cookie/WeChat.cookie
|
||||||
|
contact_user = Pickle/User.pkl
|
||||||
|
contact_member_list = Pickle/MemberList.pkl
|
||||||
|
contact_group_list = Pickle/GroupList.pkl
|
||||||
|
contact_group_memeber_list = Pickle/GroupMemeberList.pkl
|
||||||
|
contact_special_users_list = Pickle/SpecialUsersList.pkl
|
||||||
|
server_mode = False
|
||||||
|
server_log_file = server.log
|
||||||
|
log_mode = False
|
||||||
|
|
||||||
|
[mysql]
|
||||||
|
host = localhost
|
||||||
|
port = 3306
|
||||||
|
user = root
|
||||||
|
passwd = root
|
||||||
|
database = wechat
|
||||||
|
|
||||||
|
[loggers]
|
||||||
|
keys = root,WeChat
|
||||||
|
|
||||||
|
[handlers]
|
||||||
|
keys = consoleHandler,fileHandler
|
||||||
|
|
||||||
|
[formatters]
|
||||||
|
keys = simpleFormatter
|
||||||
|
|
||||||
|
[logger_root]
|
||||||
|
level = DEBUG
|
||||||
|
handlers = consoleHandler
|
||||||
|
|
||||||
|
[logger_WeChat]
|
||||||
|
level = DEBUG
|
||||||
|
handlers = fileHandler
|
||||||
|
qualname = WeChat
|
||||||
|
propagate = 0
|
||||||
|
|
||||||
|
[handler_consoleHandler]
|
||||||
|
class = StreamHandler
|
||||||
|
level = DEBUG
|
||||||
|
formatter = simpleFormatter
|
||||||
|
args = (sys.stdout,)
|
||||||
|
|
||||||
|
[handler_fileHandler]
|
||||||
|
class = FileHandler
|
||||||
|
level = DEBUG
|
||||||
|
formatter = simpleFormatter
|
||||||
|
args = ('tmp_data/wechat.log',)
|
||||||
|
|
||||||
|
[formatter_simpleFormatter]
|
||||||
|
format = %(asctime)s - %(name)s - %(levelname)s - %(message)s
|
||||||
|
datefmt =
|
||||||
|
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
from sqlite_db import SqliteDB
|
||||||
|
from mysql_db import MysqlDB
|
||||||
@@ -0,0 +1,200 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from config import Log
|
||||||
|
#---------------------------------------------------
|
||||||
|
import pymysql
|
||||||
|
import threading
|
||||||
|
import traceback
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
def array_join(arr, c):
|
||||||
|
t = ''
|
||||||
|
for a in arr:
|
||||||
|
t += "'%s'" % str(a).replace("'","\\\'") + c
|
||||||
|
return t[:-len(c)]
|
||||||
|
|
||||||
|
class MysqlDB(object):
|
||||||
|
"""
|
||||||
|
修改服务器上的配置文件/etc/my.cnf，在对应位置添加以下设置:
|
||||||
|
[client]
|
||||||
|
default-character-set = utf8mb4
|
||||||
|
|
||||||
|
[mysql]
|
||||||
|
default-character-set = utf8mb4
|
||||||
|
|
||||||
|
[mysqld]
|
||||||
|
character-set-client-handshake = FALSE
|
||||||
|
character-set-server = utf8mb4
|
||||||
|
collation-server = utf8mb4_unicode_ci
|
||||||
|
init_connect='SET NAMES utf8mb4'
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, conf):
|
||||||
|
self.conf = conf
|
||||||
|
config = {
|
||||||
|
'host': conf['host'],
|
||||||
|
'port': conf['port'],
|
||||||
|
'user': conf['user'],
|
||||||
|
'passwd': conf['passwd'],
|
||||||
|
'charset':'utf8mb4', # 支持1-4个字节字符
|
||||||
|
'cursorclass': pymysql.cursors.DictCursor
|
||||||
|
}
|
||||||
|
self.conn = pymysql.connect(**config)
|
||||||
|
self.conn.autocommit(1)
|
||||||
|
# for thread-save
|
||||||
|
self.lock = threading.Lock()
|
||||||
|
|
||||||
|
self.create_db(conf['database'])
|
||||||
|
self.conn.select_db(conf['database'])
|
||||||
|
|
||||||
|
# cache table cols
|
||||||
|
self.table_cols = {}
|
||||||
|
for t in self.show_tables():
|
||||||
|
self.table_cols[t] = self.get_table_column_name(t)
|
||||||
|
|
||||||
|
def show_database(self):
|
||||||
|
c = self.conn.cursor()
|
||||||
|
sql = 'SHOW DATABASES'
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
c.execute(sql)
|
||||||
|
return [r['Database'] for r in c.fetchall()]
|
||||||
|
|
||||||
|
def show_tables(self):
|
||||||
|
c = self.conn.cursor()
|
||||||
|
sql = 'SHOW TABLES'
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
c.execute(sql)
|
||||||
|
return [r['Tables_in_'+self.conf['database']] for r in c.fetchall()]
|
||||||
|
|
||||||
|
def create_db(self, db_name):
|
||||||
|
"""
|
||||||
|
@brief Creates a database
|
||||||
|
@param db_name String
|
||||||
|
"""
|
||||||
|
if self.conf['database'] not in self.show_database():
|
||||||
|
sql = 'CREATE DATABASE IF NOT EXISTS %s CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci' % db_name
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql)
|
||||||
|
|
||||||
|
def create_table(self, table, cols):
|
||||||
|
"""
|
||||||
|
@brief Creates a table in database
|
||||||
|
@param table String
|
||||||
|
@param cols String, the cols in table
|
||||||
|
"""
|
||||||
|
if table not in self.table_cols:
|
||||||
|
sql = 'CREATE TABLE IF NOT EXISTS %s(id int primary key auto_increment, %s) CHARACTER SET utf8mb4 COLLATE utf8mb4_unicode_ci' % (table, cols)
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql)
|
||||||
|
self.table_cols[table] = ['id'] + [c.strip().split(' ')[0] for c in cols.split(',')]
|
||||||
|
|
||||||
|
def delete_table(self, table):
|
||||||
|
"""
|
||||||
|
@brief Delete a table in database
|
||||||
|
@param table String
|
||||||
|
"""
|
||||||
|
if table in self.table_cols:
|
||||||
|
sql = "DROP TABLE IF EXISTS %s" % table
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql)
|
||||||
|
self.table_cols.pop(table)
|
||||||
|
|
||||||
|
def insert(self, table, value):
|
||||||
|
"""
|
||||||
|
@brief Insert a row in table
|
||||||
|
@param table String
|
||||||
|
@param value Tuple
|
||||||
|
"""
|
||||||
|
col_name = self.table_cols[table][1:]
|
||||||
|
sql = "INSERT INTO %s(%s) VALUES (%s)" % (table, str(','.join(col_name)), array_join(value, ','))
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql)
|
||||||
|
|
||||||
|
def insertmany(self, table, values):
|
||||||
|
"""
|
||||||
|
@brief Insert many rows in table
|
||||||
|
@param table String
|
||||||
|
@param values Array of tuple
|
||||||
|
"""
|
||||||
|
col_name = self.table_cols[table][1:]
|
||||||
|
sql = 'INSERT INTO %s(%s) VALUES (%s)' % (table, ','.join(col_name), ','.join(['%s'] * len(values[0])))
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql, values)
|
||||||
|
|
||||||
|
def select(self, table, field='', condition=''):
|
||||||
|
"""
|
||||||
|
@brief select all result from table
|
||||||
|
@param table String
|
||||||
|
@param field String
|
||||||
|
@param condition String
|
||||||
|
@return result Tuple
|
||||||
|
"""
|
||||||
|
sql = "SELECT * FROM %s" % table
|
||||||
|
if field and condition:
|
||||||
|
sql += " WHERE %s='%s'" % (field, condition)
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
return self.execute(sql)
|
||||||
|
|
||||||
|
def get_table_column_name(self, table):
|
||||||
|
"""
|
||||||
|
@brief select all result from table
|
||||||
|
@param table String
|
||||||
|
@return result Array
|
||||||
|
"""
|
||||||
|
c = self.conn.cursor()
|
||||||
|
c.execute("SELECT * FROM %s" % table)
|
||||||
|
names = list(map(lambda x: x[0], c.description))
|
||||||
|
return names
|
||||||
|
|
||||||
|
def execute(self, sql, values=None):
|
||||||
|
"""
|
||||||
|
@brief execute sql commands, return result if it has
|
||||||
|
@param sql String
|
||||||
|
@param value Tuple
|
||||||
|
@return result Array
|
||||||
|
"""
|
||||||
|
c = self.conn.cursor()
|
||||||
|
self.lock.acquire()
|
||||||
|
hasReturn = sql.lstrip().upper().startswith("SELECT")
|
||||||
|
|
||||||
|
result = []
|
||||||
|
try:
|
||||||
|
if values:
|
||||||
|
c.executemany(sql, values)
|
||||||
|
else:
|
||||||
|
c.execute(sql)
|
||||||
|
|
||||||
|
if hasReturn:
|
||||||
|
result = c.fetchall()
|
||||||
|
|
||||||
|
except Exception, e:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
self.conn.rollback()
|
||||||
|
finally:
|
||||||
|
self.lock.release()
|
||||||
|
|
||||||
|
if hasReturn:
|
||||||
|
return result
|
||||||
|
|
||||||
|
def delete(self, table, field='', condition=''):
|
||||||
|
"""
|
||||||
|
@brief execute sql commands, return result if it has
|
||||||
|
@param table String
|
||||||
|
@param field String
|
||||||
|
@param condition String
|
||||||
|
"""
|
||||||
|
sql = "DELETE FROM %s WHERE %s=%s" % (table, field, condition)
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql)
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
"""
|
||||||
|
@brief close connection to database
|
||||||
|
"""
|
||||||
|
Log.debug('DB -> close')
|
||||||
|
# 关闭数据库连接
|
||||||
|
self.conn.close()
|
||||||
|
|
||||||
|
|
||||||
@@ -0,0 +1,173 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from config import Log
|
||||||
|
#---------------------------------------------------
|
||||||
|
import sqlite3
|
||||||
|
import threading
|
||||||
|
import traceback
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
|
||||||
|
def _dict_factory(cursor, row):
|
||||||
|
aDict = {}
|
||||||
|
for iField, field in enumerate (cursor.description):
|
||||||
|
aDict [field [0]] = row [iField]
|
||||||
|
return aDict
|
||||||
|
|
||||||
|
|
||||||
|
class SqliteDB(object):
|
||||||
|
|
||||||
|
def __init__(self, db_file):
|
||||||
|
self.db_file = db_file
|
||||||
|
self.conn = sqlite3.connect(db_file, check_same_thread=False)
|
||||||
|
# use 8-bit strings instead of unicode string
|
||||||
|
self.conn.text_factory = str
|
||||||
|
# not return a tuple but a dict with column name as key
|
||||||
|
self.conn.row_factory = _dict_factory
|
||||||
|
# for thread-save
|
||||||
|
self.lock = threading.Lock()
|
||||||
|
|
||||||
|
def create_table(self, table, cols):
|
||||||
|
"""
|
||||||
|
@brief Creates a table in database
|
||||||
|
@param table String
|
||||||
|
@param cols String, the cols in table
|
||||||
|
"""
|
||||||
|
sql = "CREATE TABLE if not exists %s (%s);" % (table, cols)
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql)
|
||||||
|
|
||||||
|
def delete_table(self, table):
|
||||||
|
"""
|
||||||
|
@brief Delete a table in database
|
||||||
|
@param table String
|
||||||
|
"""
|
||||||
|
sql = "DROP TABLE if exists %s;" % table
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql)
|
||||||
|
|
||||||
|
def insert(self, table, value):
|
||||||
|
"""
|
||||||
|
@brief Insert a row in table
|
||||||
|
@param table String
|
||||||
|
@param value Tuple
|
||||||
|
"""
|
||||||
|
sql = ("INSERT INTO %s VALUES (" + ",".join(['?'] * len(value)) + ");") % table
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql, value)
|
||||||
|
|
||||||
|
def insertmany(self, table, values):
|
||||||
|
"""
|
||||||
|
@brief Insert many rows in table
|
||||||
|
@param table String
|
||||||
|
@param values Array of tuple
|
||||||
|
"""
|
||||||
|
c = self.conn.cursor()
|
||||||
|
self.lock.acquire()
|
||||||
|
n = len(values[0])
|
||||||
|
sql = ("INSERT INTO %s VALUES (" + ",".join(['?'] * n) + ");") % table
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
|
||||||
|
try:
|
||||||
|
c.executemany(sql, values)
|
||||||
|
except Exception, e:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
finally:
|
||||||
|
self.lock.release()
|
||||||
|
|
||||||
|
self.conn.commit()
|
||||||
|
|
||||||
|
def select(self, table, field='', condition=''):
|
||||||
|
"""
|
||||||
|
@brief select all result from table
|
||||||
|
@param table String
|
||||||
|
@param field String
|
||||||
|
@param condition String
|
||||||
|
@return result Tuple
|
||||||
|
"""
|
||||||
|
result = []
|
||||||
|
if field and condition:
|
||||||
|
cond = (condition,)
|
||||||
|
sql = "SELECT * FROM %s WHERE %s=?" % (table, field)
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
result = self.execute(sql, cond)
|
||||||
|
else:
|
||||||
|
sql = "SELECT * FROM %s" % table
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
result = self.execute(sql)
|
||||||
|
return result
|
||||||
|
|
||||||
|
def update(self, table, dic, condition=''):
|
||||||
|
k_arr = []
|
||||||
|
v_arr = []
|
||||||
|
for (k, v) in dic.items():
|
||||||
|
k_arr.append('%s=?' % k)
|
||||||
|
v_arr.append(v)
|
||||||
|
|
||||||
|
sql = "UPDATE %s SET %s" % (table, ','.join(k_arr))
|
||||||
|
if condition:
|
||||||
|
sql += " WHERE %s" % condition
|
||||||
|
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
self.execute(sql, tuple(v_arr))
|
||||||
|
|
||||||
|
def get_table_column_name(self, table):
|
||||||
|
"""
|
||||||
|
@brief select all result from table
|
||||||
|
@param table String
|
||||||
|
@return result Array
|
||||||
|
"""
|
||||||
|
c = self.conn.cursor()
|
||||||
|
c.execute("SELECT * FROM %s" % table)
|
||||||
|
names = list(map(lambda x: x[0], c.description))
|
||||||
|
return names
|
||||||
|
|
||||||
|
def execute(self, sql, value=None):
|
||||||
|
"""
|
||||||
|
@brief execute sql commands, return result if it has
|
||||||
|
@param sql String
|
||||||
|
@param value Tuple
|
||||||
|
@return result Array
|
||||||
|
"""
|
||||||
|
c = self.conn.cursor()
|
||||||
|
self.lock.acquire()
|
||||||
|
hasReturn = sql.lstrip().upper().startswith("SELECT")
|
||||||
|
|
||||||
|
try:
|
||||||
|
if value:
|
||||||
|
c.execute(sql, value)
|
||||||
|
else:
|
||||||
|
c.execute(sql)
|
||||||
|
|
||||||
|
if hasReturn:
|
||||||
|
result = c.fetchall()
|
||||||
|
except Exception, e:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
finally:
|
||||||
|
self.lock.release()
|
||||||
|
|
||||||
|
self.conn.commit()
|
||||||
|
|
||||||
|
if hasReturn:
|
||||||
|
return result
|
||||||
|
|
||||||
|
def delete(self, table, field='', condition=''):
|
||||||
|
"""
|
||||||
|
@brief execute sql commands, return result if it has
|
||||||
|
@param table String
|
||||||
|
@param field String
|
||||||
|
@param condition String
|
||||||
|
"""
|
||||||
|
sql = "DELETE FROM %s WHERE %s=?" % (table, field)
|
||||||
|
Log.debug('DB -> %s' % sql)
|
||||||
|
cond = (condition,)
|
||||||
|
self.execute(sql, cond)
|
||||||
|
|
||||||
|
def close(self):
|
||||||
|
"""
|
||||||
|
@brief close connection to database
|
||||||
|
"""
|
||||||
|
Log.debug('DB -> close')
|
||||||
|
self.conn.close()
|
||||||
@@ -0,0 +1,18 @@
|
|||||||
|
FROM ubuntu:16.04
|
||||||
|
MAINTAINER Urinx <uri.lqy@gmail.com>
|
||||||
|
|
||||||
|
RUN apt-get update && \
|
||||||
|
apt-get install -y python \
|
||||||
|
python-dev \
|
||||||
|
python-pip && \
|
||||||
|
apt-get clean && \
|
||||||
|
apt-get autoclean && \
|
||||||
|
rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
|
||||||
|
|
||||||
|
ADD weixin_bot.tar.gz /
|
||||||
|
WORKDIR /weixin_bot
|
||||||
|
|
||||||
|
RUN pip install -r config/requirements.txt
|
||||||
|
EXPOSE 80
|
||||||
|
ENTRYPOINT ["./weixin_bot.py"]
|
||||||
|
CMD [""]
|
||||||
@@ -0,0 +1,49 @@
|
|||||||
|
# 本地构建 wechat_bot docker 镜像
|
||||||
|
|
||||||
|
拉下镜像:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker pull ubuntu:16.04
|
||||||
|
```
|
||||||
|
|
||||||
|
打包本项目，将压缩包放到`docker`目录下:
|
||||||
|
|
||||||
|
```
|
||||||
|
tar -czf weixin_bot.tar.gz wxbot_project_py2.7/
|
||||||
|
```
|
||||||
|
|
||||||
|
切换到`docker`目录，执行`build`命令:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker build -t wechat-bot .
|
||||||
|
```
|
||||||
|
|
||||||
|
导出镜像:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker save wechat-bot > wechat.tar
|
||||||
|
```
|
||||||
|
|
||||||
|
导入镜像:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker load < wechat.tar
|
||||||
|
```
|
||||||
|
|
||||||
|
运行:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker run -d -P --name xxx -v /src/data/dir:/Wechat_bot/test wechat-bot
|
||||||
|
```
|
||||||
|
|
||||||
|
删除镜像:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker rmi -f wechat-bot
|
||||||
|
```
|
||||||
|
|
||||||
|
查看log:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker log wechat-bot
|
||||||
|
```
|
||||||
@@ -0,0 +1,210 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html>
|
||||||
|
<head>
|
||||||
|
<meta charset="utf-8">
|
||||||
|
<title>WeChat Bot Server</title>
|
||||||
|
<style type="text/css">
|
||||||
|
.container {
|
||||||
|
display: flex;
|
||||||
|
}
|
||||||
|
div.doc{
|
||||||
|
margin: auto;
|
||||||
|
text-align: left;
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body class="container">
|
||||||
|
|
||||||
|
<div class="doc">
|
||||||
|
<pre>
|
||||||
|
#####################
|
||||||
|
# WeChat Bot Server #
|
||||||
|
#####################
|
||||||
|
|
||||||
|
|
||||||
|
## APIs
|
||||||
|
|
||||||
|
- /qrcode
|
||||||
|
------------------------
|
||||||
|
@brief login qrcode
|
||||||
|
@method get
|
||||||
|
@return raw image
|
||||||
|
------------------------
|
||||||
|
|
||||||
|
- /runtime
|
||||||
|
-----------------------------
|
||||||
|
@brief basic info
|
||||||
|
@method get
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
'runtime': '',
|
||||||
|
'total_size': '',
|
||||||
|
'db_size': '',
|
||||||
|
'msg_count': '',
|
||||||
|
'image_count': '',
|
||||||
|
'voice_count': '',
|
||||||
|
'video_count': '',
|
||||||
|
}
|
||||||
|
-----------------------------
|
||||||
|
|
||||||
|
- /group_list
|
||||||
|
------------------------
|
||||||
|
@brief list groups
|
||||||
|
@method get
|
||||||
|
------------------------
|
||||||
|
|
||||||
|
- /group_member_list/<g_id>
|
||||||
|
-----------------------------
|
||||||
|
@brief list group member
|
||||||
|
@method get
|
||||||
|
@param g_id String
|
||||||
|
-----------------------------
|
||||||
|
|
||||||
|
- /group_chat_log/<g_name>
|
||||||
|
-------------------------------
|
||||||
|
@brief list group chat log
|
||||||
|
@method get
|
||||||
|
@param g_name String
|
||||||
|
-------------------------------
|
||||||
|
|
||||||
|
- /upload
|
||||||
|
-------------------------------
|
||||||
|
@brief upload a file
|
||||||
|
@method get/post
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
'msg': '',
|
||||||
|
}
|
||||||
|
-------------------------------
|
||||||
|
|
||||||
|
- /send_msg/<to>/<msg>
|
||||||
|
-------------------------------------------
|
||||||
|
@brief send message to user or gourp
|
||||||
|
@method get
|
||||||
|
@param to: String, user id or group id
|
||||||
|
@param msg: String, words
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
}
|
||||||
|
-------------------------------------------
|
||||||
|
|
||||||
|
- /send_img/<to>/<img>
|
||||||
|
-------------------------------------------
|
||||||
|
@brief send image to user or gourp
|
||||||
|
@method get
|
||||||
|
@param to: String, user id or group id
|
||||||
|
@param img: String, image file name
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
}
|
||||||
|
-------------------------------------------
|
||||||
|
|
||||||
|
- /send_emot/<to>/<emot>
|
||||||
|
-------------------------------------------
|
||||||
|
@brief send emotion to user or gourp
|
||||||
|
@method get
|
||||||
|
@param to: String, user id or group id
|
||||||
|
@param emot: String, emotion file name
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
}
|
||||||
|
-------------------------------------------
|
||||||
|
|
||||||
|
- /send_file/<to>/<file>
|
||||||
|
-------------------------------------------
|
||||||
|
@brief send emotion to user or gourp
|
||||||
|
@method get
|
||||||
|
@param to: String, user id or group id
|
||||||
|
@param file: String, file name
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
}
|
||||||
|
-------------------------------------------
|
||||||
|
|
||||||
|
- /mass_send_msg
|
||||||
|
-------------------------------------------
|
||||||
|
@brief send text to mass users or gourps
|
||||||
|
@method post
|
||||||
|
@param application/json
|
||||||
|
{
|
||||||
|
'to_list': [
|
||||||
|
'group_id',
|
||||||
|
...
|
||||||
|
],
|
||||||
|
'msg': '',
|
||||||
|
}
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
'unsend_list': [],
|
||||||
|
}
|
||||||
|
-------------------------------------------
|
||||||
|
|
||||||
|
- /mass_send_img
|
||||||
|
-------------------------------------------
|
||||||
|
@brief send image to mass users or gourps
|
||||||
|
@method post
|
||||||
|
@param application/json
|
||||||
|
{
|
||||||
|
'to_list': [
|
||||||
|
'group_id',
|
||||||
|
...
|
||||||
|
],
|
||||||
|
'msg': '',
|
||||||
|
}
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
'unsend_list': [],
|
||||||
|
}
|
||||||
|
-------------------------------------------
|
||||||
|
|
||||||
|
- /mass_send_emot
|
||||||
|
-------------------------------------------
|
||||||
|
@brief send emoticon to mass users or gourps
|
||||||
|
@method post
|
||||||
|
@param application/json
|
||||||
|
{
|
||||||
|
'to_list': [
|
||||||
|
'group_id',
|
||||||
|
...
|
||||||
|
],
|
||||||
|
'msg': '',
|
||||||
|
}
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
'unsend_list': [],
|
||||||
|
}
|
||||||
|
-------------------------------------------
|
||||||
|
|
||||||
|
- /mass_send_file
|
||||||
|
-------------------------------------------
|
||||||
|
@brief send file to mass users or gourps
|
||||||
|
@method post
|
||||||
|
@param application/json
|
||||||
|
{
|
||||||
|
'to_list': [
|
||||||
|
'group_id',
|
||||||
|
...
|
||||||
|
],
|
||||||
|
'msg': '',
|
||||||
|
}
|
||||||
|
@return application/json
|
||||||
|
{
|
||||||
|
'ret': 0,
|
||||||
|
'unsend_list': [],
|
||||||
|
}
|
||||||
|
-------------------------------------------
|
||||||
|
</pre>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
</body>
|
||||||
|
</html>
|
||||||
|
|
||||||
@@ -0,0 +1,7 @@
|
|||||||
|
<!doctype html>
|
||||||
|
<title>Upload new File</title>
|
||||||
|
<h1>Upload new File</h1>
|
||||||
|
<form action="upload" method=post enctype=multipart/form-data>
|
||||||
|
<p><input type=file name=file>
|
||||||
|
<input type=submit value=Upload>
|
||||||
|
</form>
|
||||||
@@ -1,2 +1,4 @@
|
|||||||
#!/usr/bin/env python
|
#!/usr/bin/env python
|
||||||
# coding: utf-8
|
# coding: utf-8
|
||||||
|
|
||||||
|
from wechat import WeChat
|
||||||
@@ -0,0 +1,359 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from config import Log
|
||||||
|
from config import Constant
|
||||||
|
#---------------------------------------------------
|
||||||
|
import qrcode
|
||||||
|
import re
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import json
|
||||||
|
import urllib
|
||||||
|
import urllib2
|
||||||
|
import cookielib
|
||||||
|
import cPickle as pickle
|
||||||
|
import traceback
|
||||||
|
import time
|
||||||
|
import hashlib
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
|
||||||
|
def _decode_data(data):
|
||||||
|
"""
|
||||||
|
@brief decode array or dict to utf-8
|
||||||
|
@param data array or dict
|
||||||
|
@return utf-8
|
||||||
|
"""
|
||||||
|
if isinstance(data, dict):
|
||||||
|
rv = {}
|
||||||
|
for key, value in data.iteritems():
|
||||||
|
if isinstance(key, unicode):
|
||||||
|
key = key.encode('utf-8')
|
||||||
|
rv[key] = _decode_data(value)
|
||||||
|
return rv
|
||||||
|
elif isinstance(data, list):
|
||||||
|
rv = []
|
||||||
|
for item in data:
|
||||||
|
item = _decode_data(item)
|
||||||
|
rv.append(item)
|
||||||
|
return rv
|
||||||
|
elif isinstance(data, unicode):
|
||||||
|
return data.encode('utf-8')
|
||||||
|
else:
|
||||||
|
return data
|
||||||
|
|
||||||
|
|
||||||
|
def str2qr_terminal(text):
|
||||||
|
"""
|
||||||
|
@brief convert string to qrcode matrix and outprint
|
||||||
|
@param text The string
|
||||||
|
"""
|
||||||
|
Log.debug(text)
|
||||||
|
qr = qrcode.QRCode()
|
||||||
|
qr.border = 1
|
||||||
|
qr.add_data(text)
|
||||||
|
mat = qr.get_matrix()
|
||||||
|
print_qr(mat)
|
||||||
|
|
||||||
|
|
||||||
|
def str2qr_image(text, image_path):
|
||||||
|
"""
|
||||||
|
@brief convert string to qrcode image & save
|
||||||
|
@param text The string
|
||||||
|
@param image_path Save image to the path
|
||||||
|
"""
|
||||||
|
qr = qrcode.QRCode()
|
||||||
|
qr.border = 1
|
||||||
|
qr.add_data(text)
|
||||||
|
qr.make(fit=True)
|
||||||
|
img = qr.make_image()
|
||||||
|
img.save(image_path)
|
||||||
|
|
||||||
|
|
||||||
|
def print_qr(mat):
|
||||||
|
for i in mat:
|
||||||
|
BLACK = Constant.QRCODE_BLACK
|
||||||
|
WHITE = Constant.QRCODE_WHITE
|
||||||
|
echo(''.join([BLACK if j else WHITE for j in i])+'\n')
|
||||||
|
|
||||||
|
|
||||||
|
def echo(str):
|
||||||
|
Log.info(str[:-1])
|
||||||
|
sys.stdout.write(str)
|
||||||
|
sys.stdout.flush()
|
||||||
|
|
||||||
|
|
||||||
|
def run(str, func, *args):
|
||||||
|
t = time.time()
|
||||||
|
echo(str)
|
||||||
|
r = False
|
||||||
|
try:
|
||||||
|
r = func(*args)
|
||||||
|
except:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
if r:
|
||||||
|
totalTime = int(time.time() - t)
|
||||||
|
echo(Constant.RUN_RESULT_SUCCESS % totalTime)
|
||||||
|
else:
|
||||||
|
echo(Constant.RUN_RESULT_FAIL)
|
||||||
|
exit()
|
||||||
|
|
||||||
|
|
||||||
|
def get(url, api=None):
|
||||||
|
"""
|
||||||
|
@brief http get request
|
||||||
|
@param url String
|
||||||
|
@param api wechat api
|
||||||
|
@return http response
|
||||||
|
"""
|
||||||
|
Log.debug('GET -> ' + url)
|
||||||
|
request = urllib2.Request(url=url)
|
||||||
|
request.add_header(*Constant.HTTP_HEADER_CONNECTION)
|
||||||
|
request.add_header(*Constant.HTTP_HEADER_REFERER)
|
||||||
|
if api in ['webwxgetvoice', 'webwxgetvideo']:
|
||||||
|
request.add_header(*Constant.HTTP_HEADER_RANGE)
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
response = urllib2.urlopen(request, timeout=30)
|
||||||
|
data = response.read()
|
||||||
|
response.close()
|
||||||
|
if api == None:
|
||||||
|
Log.debug(data)
|
||||||
|
return data
|
||||||
|
except (KeyboardInterrupt, SystemExit):
|
||||||
|
raise
|
||||||
|
except:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
|
||||||
|
def post(url, params, jsonfmt=True):
|
||||||
|
"""
|
||||||
|
@brief http post request
|
||||||
|
@param url String
|
||||||
|
@param params Dict, post params
|
||||||
|
@param jsonfmt Bool, whether is json format
|
||||||
|
@return http response
|
||||||
|
"""
|
||||||
|
Log.debug('POST -> '+url)
|
||||||
|
Log.debug(params)
|
||||||
|
if jsonfmt:
|
||||||
|
request = urllib2.Request(url=url, data=json.dumps(params, ensure_ascii=False).encode('utf8'))
|
||||||
|
request.add_header(*Constant.HTTP_HEADER_CONTENTTYPE)
|
||||||
|
else:
|
||||||
|
request = urllib2.Request(url=url, data=urllib.urlencode(params))
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
response = urllib2.urlopen(request, timeout=30)
|
||||||
|
data = response.read()
|
||||||
|
response.close()
|
||||||
|
|
||||||
|
if jsonfmt:
|
||||||
|
Log.debug(data)
|
||||||
|
return json.loads(data, object_hook=_decode_data)
|
||||||
|
return data
|
||||||
|
except (KeyboardInterrupt, SystemExit):
|
||||||
|
raise
|
||||||
|
except:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
|
||||||
|
def set_cookie(cookie_file):
|
||||||
|
"""
|
||||||
|
@brief Load cookie from file
|
||||||
|
@param cookie_file
|
||||||
|
@param user_agent
|
||||||
|
@return cookie, LWPCookieJar
|
||||||
|
"""
|
||||||
|
cookie = cookielib.LWPCookieJar(cookie_file)
|
||||||
|
try:
|
||||||
|
cookie.load(ignore_discard=True)
|
||||||
|
except:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(cookie))
|
||||||
|
opener.addheaders = Constant.HTTP_HEADER_USERAGENT
|
||||||
|
urllib2.install_opener(opener)
|
||||||
|
return cookie
|
||||||
|
|
||||||
|
|
||||||
|
def generate_file_name(filename):
|
||||||
|
"""
|
||||||
|
@brief generate file name
|
||||||
|
@return new file name
|
||||||
|
"""
|
||||||
|
i = filename.rfind('.')
|
||||||
|
ext = filename[i:]
|
||||||
|
tmp = filename + str(int(time.time()))
|
||||||
|
hash_md5 = hashlib.md5(tmp)
|
||||||
|
return hash_md5.hexdigest() + ext
|
||||||
|
|
||||||
|
|
||||||
|
def save_file(filename, data, dirName):
|
||||||
|
"""
|
||||||
|
@brief Saves raw data to file.
|
||||||
|
@param filename String
|
||||||
|
@param data Binary data
|
||||||
|
@param dirName String
|
||||||
|
@return file path
|
||||||
|
"""
|
||||||
|
Log.debug('save file: ' + filename)
|
||||||
|
fn = filename
|
||||||
|
if not os.path.exists(dirName):
|
||||||
|
os.makedirs(dirName)
|
||||||
|
fn = os.path.join(dirName, filename)
|
||||||
|
with open(fn, 'wb') as f:
|
||||||
|
f.write(data)
|
||||||
|
return fn
|
||||||
|
|
||||||
|
|
||||||
|
def save_json(filename, data, dirName, mode='w+'):
|
||||||
|
"""
|
||||||
|
@brief Saves dict to json file.
|
||||||
|
@param filename String
|
||||||
|
@param data Dict
|
||||||
|
@param dirName String
|
||||||
|
@return file path
|
||||||
|
"""
|
||||||
|
Log.debug('save json: ' + filename)
|
||||||
|
fn = filename
|
||||||
|
if not os.path.exists(dirName):
|
||||||
|
os.makedirs(dirName)
|
||||||
|
fn = os.path.join(dirName, filename)
|
||||||
|
with open(fn, mode) as f:
|
||||||
|
f.write(json.dumps(data, indent=4)+'\n')
|
||||||
|
return fn
|
||||||
|
|
||||||
|
|
||||||
|
def pickle_save(data, file):
|
||||||
|
"""
|
||||||
|
@brief Use pickle to save python object into file
|
||||||
|
@param data The pyhton data
|
||||||
|
@param file The file
|
||||||
|
"""
|
||||||
|
basedir = os.path.dirname(file)
|
||||||
|
if not os.path.exists(basedir):
|
||||||
|
os.makedirs(basedir)
|
||||||
|
with open(file, 'wb') as f:
|
||||||
|
pickle.dump(data, f, pickle.HIGHEST_PROTOCOL)
|
||||||
|
|
||||||
|
|
||||||
|
def pickle_load(file):
|
||||||
|
"""
|
||||||
|
@brief Use pickle to load python object from file
|
||||||
|
@param file The file
|
||||||
|
@return python data
|
||||||
|
"""
|
||||||
|
if os.path.isfile(file):
|
||||||
|
with open(file, 'rb') as f:
|
||||||
|
return pickle.load(f)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def search_content(key, content, fmat='attr'):
|
||||||
|
"""
|
||||||
|
@brief Search content from xml or html format
|
||||||
|
@param key String
|
||||||
|
@param content String
|
||||||
|
@param fmat attr
|
||||||
|
xml
|
||||||
|
@return String
|
||||||
|
"""
|
||||||
|
if fmat == 'attr':
|
||||||
|
pm = re.search(key + '\s?=\s?"([^"<]+)"', content)
|
||||||
|
if pm:
|
||||||
|
return pm.group(1)
|
||||||
|
elif fmat == 'xml':
|
||||||
|
pm = re.search('<{0}>([^<]+)</{0}>'.format(key), content)
|
||||||
|
if not pm:
|
||||||
|
pm = re.search(
|
||||||
|
'<{0}><\!\[CDATA\[(.*?)\]\]></{0}>'.format(key), content)
|
||||||
|
if pm:
|
||||||
|
return pm.group(1)
|
||||||
|
return 'unknown'
|
||||||
|
|
||||||
|
|
||||||
|
def is_str(s):
|
||||||
|
"""
|
||||||
|
@brief Determines if string.
|
||||||
|
@param s String
|
||||||
|
@return True if string, False otherwise.
|
||||||
|
"""
|
||||||
|
return isinstance(s, basestring)
|
||||||
|
|
||||||
|
|
||||||
|
def trans_coding(data):
|
||||||
|
"""
|
||||||
|
@brief Transform string to unicode
|
||||||
|
@param data String
|
||||||
|
@return unicode
|
||||||
|
"""
|
||||||
|
if not data:
|
||||||
|
return data
|
||||||
|
result = None
|
||||||
|
if type(data) == unicode:
|
||||||
|
result = data
|
||||||
|
elif type(data) == str:
|
||||||
|
result = data.decode('utf-8')
|
||||||
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
def trans_emoji(text):
|
||||||
|
"""
|
||||||
|
@brief Transform emoji html text to unicode
|
||||||
|
@param text String
|
||||||
|
@return emoji unicode
|
||||||
|
"""
|
||||||
|
def _emoji(matched):
|
||||||
|
hex = matched.group(1)
|
||||||
|
return ('\\U%08x' % int(hex, 16)).decode('unicode-escape').encode('utf-8')
|
||||||
|
|
||||||
|
replace_t = re.sub(Constant.REGEX_EMOJI, _emoji, text)
|
||||||
|
return replace_t
|
||||||
|
|
||||||
|
|
||||||
|
def auto_reload(mod):
|
||||||
|
"""
|
||||||
|
@brief reload modules
|
||||||
|
@param mod: the need reload modules
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
module = sys.modules[mod]
|
||||||
|
except:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
return False
|
||||||
|
|
||||||
|
filename = module.__file__
|
||||||
|
# .pyc 修改时间不会变
|
||||||
|
# 所以就用 .py 的修改时间
|
||||||
|
if filename.endswith(".pyc"):
|
||||||
|
filename = filename.replace(".pyc", ".py")
|
||||||
|
mod_time = os.path.getmtime(filename)
|
||||||
|
if not "loadtime" in module.__dict__:
|
||||||
|
module.loadtime = 0
|
||||||
|
|
||||||
|
try:
|
||||||
|
if mod_time > module.loadtime:
|
||||||
|
reload(module)
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
except:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
return False
|
||||||
|
|
||||||
|
module.loadtime = mod_time
|
||||||
|
|
||||||
|
echo('[*] load \'%s\' successful.\n' % mod)
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
def split_array(arr, n):
|
||||||
|
for i in xrange(0, len(arr), n):
|
||||||
|
yield arr[i:i+n]
|
||||||
|
|
||||||
@@ -0,0 +1,697 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from utils import *
|
||||||
|
from wechat_apis import WXAPI
|
||||||
|
from config import ConfigManager
|
||||||
|
from config import Constant
|
||||||
|
from config import Log
|
||||||
|
#---------------------------------------------------
|
||||||
|
import json
|
||||||
|
import re
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
import random
|
||||||
|
from collections import defaultdict
|
||||||
|
from datetime import timedelta
|
||||||
|
import traceback
|
||||||
|
import Queue
|
||||||
|
import threading
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
|
||||||
|
class WeChat(WXAPI):
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
description = \
|
||||||
|
"=========================\n" + \
|
||||||
|
"[#] Web WeChat\n" + \
|
||||||
|
"[#] UUID: " + self.uuid + "\n" + \
|
||||||
|
"[#] Uin: " + str(self.uin) + "\n" + \
|
||||||
|
"[#] Sid: " + self.sid + "\n" + \
|
||||||
|
"[#] Skey: " + self.skey + "\n" + \
|
||||||
|
"[#] DeviceId: " + self.device_id + "\n" + \
|
||||||
|
"[#] PassTicket: " + self.pass_ticket + "\n" + \
|
||||||
|
"========================="
|
||||||
|
return description
|
||||||
|
|
||||||
|
def __init__(self, host='wx.qq.com'):
|
||||||
|
super(WeChat, self).__init__(host)
|
||||||
|
|
||||||
|
self.db = None
|
||||||
|
self.save_data_folder = '' # 保存图片，语音，小视频的文件夹
|
||||||
|
self.last_login = 0 # 上次退出的时间
|
||||||
|
self.time_out = 5 # 同步时间间隔（单位：秒）
|
||||||
|
self.msg_handler = None
|
||||||
|
self.start_time = time.time()
|
||||||
|
self.bot = None
|
||||||
|
|
||||||
|
cm = ConfigManager()
|
||||||
|
self.save_data_folders = cm.get_wechat_media_dir()
|
||||||
|
self.cookie_file = cm.get_cookie()
|
||||||
|
self.pickle_file = cm.get_pickle_files()
|
||||||
|
self.log_mode = cm.get('setting', 'log_mode') == 'True'
|
||||||
|
self.exit_code = 0
|
||||||
|
|
||||||
|
def start(self):
|
||||||
|
echo(Constant.LOG_MSG_START)
|
||||||
|
run(Constant.LOG_MSG_RECOVER, self.recover)
|
||||||
|
|
||||||
|
timeOut = time.time() - self.last_login
|
||||||
|
echo(Constant.LOG_MSG_TRY_INIT)
|
||||||
|
if self.webwxinit():
|
||||||
|
echo(Constant.LOG_MSG_SUCCESS)
|
||||||
|
run(Constant.LOG_MSG_RECOVER_CONTACT, self.recover_contacts)
|
||||||
|
else:
|
||||||
|
echo(Constant.LOG_MSG_FAIL)
|
||||||
|
|
||||||
|
while True:
|
||||||
|
# first try to login by uin without qrcode
|
||||||
|
echo(Constant.LOG_MSG_ASSOCIATION_LOGIN)
|
||||||
|
if self.association_login():
|
||||||
|
echo(Constant.LOG_MSG_SUCCESS)
|
||||||
|
else:
|
||||||
|
echo(Constant.LOG_MSG_FAIL)
|
||||||
|
# scan qrcode to login
|
||||||
|
run(Constant.LOG_MSG_GET_UUID, self.getuuid)
|
||||||
|
echo(Constant.LOG_MSG_GET_QRCODE)
|
||||||
|
self.genqrcode()
|
||||||
|
echo(Constant.LOG_MSG_SCAN_QRCODE)
|
||||||
|
|
||||||
|
if not self.waitforlogin():
|
||||||
|
continue
|
||||||
|
echo(Constant.LOG_MSG_CONFIRM_LOGIN)
|
||||||
|
if not self.waitforlogin(0):
|
||||||
|
continue
|
||||||
|
break
|
||||||
|
|
||||||
|
run(Constant.LOG_MSG_LOGIN, self.login)
|
||||||
|
run(Constant.LOG_MSG_INIT, self.webwxinit)
|
||||||
|
run(Constant.LOG_MSG_STATUS_NOTIFY, self.webwxstatusnotify)
|
||||||
|
run(Constant.LOG_MSG_GET_CONTACT, self.webwxgetcontact)
|
||||||
|
echo(Constant.LOG_MSG_CONTACT_COUNT % (
|
||||||
|
self.MemberCount, len(self.MemberList)
|
||||||
|
))
|
||||||
|
echo(Constant.LOG_MSG_OTHER_CONTACT_COUNT % (
|
||||||
|
len(self.GroupList), len(self.ContactList),
|
||||||
|
len(self.SpecialUsersList), len(self.PublicUsersList)
|
||||||
|
))
|
||||||
|
run(Constant.LOG_MSG_GET_GROUP_MEMBER, self.fetch_group_contacts)
|
||||||
|
|
||||||
|
run(Constant.LOG_MSG_SNAPSHOT, self.snapshot)
|
||||||
|
|
||||||
|
while True:
|
||||||
|
[retcode, selector] = self.synccheck()
|
||||||
|
Log.debug('retcode: %s, selector: %s' % (retcode, selector))
|
||||||
|
self.exit_code = int(retcode)
|
||||||
|
|
||||||
|
if retcode == '1100':
|
||||||
|
echo(Constant.LOG_MSG_LOGOUT)
|
||||||
|
break
|
||||||
|
if retcode == '1101':
|
||||||
|
echo(Constant.LOG_MSG_LOGIN_OTHERWHERE)
|
||||||
|
break
|
||||||
|
if retcode == '1102':
|
||||||
|
echo(Constant.LOG_MSG_QUIT_ON_PHONE)
|
||||||
|
break
|
||||||
|
elif retcode == '0':
|
||||||
|
if selector == '2':
|
||||||
|
r = self.webwxsync()
|
||||||
|
if r is not None:
|
||||||
|
try:
|
||||||
|
self.handle_msg(r)
|
||||||
|
except:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
elif selector == '7':
|
||||||
|
r = self.webwxsync()
|
||||||
|
elif selector == '0':
|
||||||
|
time.sleep(self.time_out)
|
||||||
|
elif selector == '4':
|
||||||
|
# 保存群聊到通讯录
|
||||||
|
# 修改群名称
|
||||||
|
# 新增或删除联系人
|
||||||
|
# 群聊成员数目变化
|
||||||
|
r = self.webwxsync()
|
||||||
|
if r is not None:
|
||||||
|
try:
|
||||||
|
self.handle_mod(r)
|
||||||
|
except:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
elif selector == '3' or selector == '6':
|
||||||
|
break
|
||||||
|
else:
|
||||||
|
r = self.webwxsync()
|
||||||
|
Log.debug('webwxsync: %s\n' % json.dumps(r))
|
||||||
|
|
||||||
|
# 执行定时任务
|
||||||
|
if self.msg_handler:
|
||||||
|
self.msg_handler.check_schedule_task()
|
||||||
|
|
||||||
|
# if self.bot:
|
||||||
|
# r = self.bot.time_schedule()
|
||||||
|
# if r:
|
||||||
|
# for g in self.GroupList:
|
||||||
|
# echo('[*] 推送 -> %s: %s' % (g['NickName'], r))
|
||||||
|
# g_id = g['UserName']
|
||||||
|
# self.webwxsendmsg(r, g_id)
|
||||||
|
|
||||||
|
def get_run_time(self):
|
||||||
|
"""
|
||||||
|
@brief get how long this run
|
||||||
|
@return String
|
||||||
|
"""
|
||||||
|
totalTime = int(time.time() - self.start_time)
|
||||||
|
t = timedelta(seconds=totalTime)
|
||||||
|
return '%s Day %s' % (t.days, t)
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
"""
|
||||||
|
@brief Save some data and use shell to kill this process
|
||||||
|
"""
|
||||||
|
run(Constant.LOG_MSG_SNAPSHOT, self.snapshot)
|
||||||
|
echo(Constant.LOG_MSG_RUNTIME % self.get_run_time())
|
||||||
|
# close database connect
|
||||||
|
self.db.close()
|
||||||
|
|
||||||
|
def fetch_group_contacts(self):
|
||||||
|
"""
|
||||||
|
@brief Fetches all groups contacts.
|
||||||
|
@return Bool: whether operation succeed.
|
||||||
|
@note This function must be finished in 180s
|
||||||
|
"""
|
||||||
|
Log.debug('fetch_group_contacts')
|
||||||
|
# clean database
|
||||||
|
if self.msg_handler:
|
||||||
|
self.msg_handler.clean_db()
|
||||||
|
|
||||||
|
# sqlite
|
||||||
|
# ----------------------------------------------------
|
||||||
|
# group max_thread_num max_fetch_group_num time(s)
|
||||||
|
# 197 10 10 108
|
||||||
|
# 197 10 15 95
|
||||||
|
# 197 20 10 103
|
||||||
|
# 197 10 20 55
|
||||||
|
# 197 5 30 39
|
||||||
|
# 197 4 50 35
|
||||||
|
# ----------------------------------------------------
|
||||||
|
# mysql
|
||||||
|
# ----------------------------------------------------
|
||||||
|
# group max_thread_num max_fetch_group_num time(s)
|
||||||
|
# 197 4 50 20
|
||||||
|
# ----------------------------------------------------
|
||||||
|
|
||||||
|
max_thread_num = 4
|
||||||
|
max_fetch_group_num = 50
|
||||||
|
group_list_queue = Queue.Queue()
|
||||||
|
|
||||||
|
class GroupListThread(threading.Thread):
|
||||||
|
|
||||||
|
def __init__(self, group_list_queue, wechat):
|
||||||
|
threading.Thread.__init__(self)
|
||||||
|
self.group_list_queue = group_list_queue
|
||||||
|
self.wechat = wechat
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
while not self.group_list_queue.empty():
|
||||||
|
g_list = self.group_list_queue.get()
|
||||||
|
gid_list = []
|
||||||
|
g_dict = {}
|
||||||
|
for g in g_list:
|
||||||
|
gid = g['UserName']
|
||||||
|
gid_list.append(gid)
|
||||||
|
g_dict[gid] = g
|
||||||
|
|
||||||
|
group_member_list = self.wechat.webwxbatchgetcontact(gid_list)
|
||||||
|
|
||||||
|
for member_list in group_member_list:
|
||||||
|
gid = member_list['UserName']
|
||||||
|
g = g_dict[gid]
|
||||||
|
g['MemberCount'] = member_list['MemberCount']
|
||||||
|
g['OwnerUin'] = member_list['OwnerUin']
|
||||||
|
self.wechat.GroupMemeberList[gid] = member_list['MemberList']
|
||||||
|
|
||||||
|
# 如果使用 Mysql 则可以在多线程里操作数据库
|
||||||
|
# 否则请注释下列代码在主线程里更新群列表
|
||||||
|
# -----------------------------------
|
||||||
|
# 处理群成员
|
||||||
|
# if self.wechat.msg_handler:
|
||||||
|
# self.wechat.msg_handler.handle_group_member_list(gid, member_list['MemberList'])
|
||||||
|
# -----------------------------------
|
||||||
|
|
||||||
|
self.group_list_queue.task_done()
|
||||||
|
|
||||||
|
for g_list in split_array(self.GroupList, max_fetch_group_num):
|
||||||
|
group_list_queue.put(g_list)
|
||||||
|
|
||||||
|
for i in range(max_thread_num):
|
||||||
|
t = GroupListThread(group_list_queue, self)
|
||||||
|
t.setDaemon(True)
|
||||||
|
t.start()
|
||||||
|
|
||||||
|
group_list_queue.join()
|
||||||
|
|
||||||
|
if self.msg_handler:
|
||||||
|
# 处理群
|
||||||
|
if self.GroupList:
|
||||||
|
self.msg_handler.handle_group_list(self.GroupList)
|
||||||
|
|
||||||
|
# 这个是用 sqlite 来存储群列表，sqlite 对多线程的支持不太好
|
||||||
|
# ----------------------------------------------------
|
||||||
|
# 处理群成员
|
||||||
|
for (gid, member_list) in self.GroupMemeberList.items():
|
||||||
|
self.msg_handler.handle_group_member_list(gid, member_list)
|
||||||
|
# ----------------------------------------------------
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
def snapshot(self):
|
||||||
|
"""
|
||||||
|
@brief Save basic infos for next login.
|
||||||
|
@return Bool: whether operation succeed.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
conf = {
|
||||||
|
'uuid': self.uuid,
|
||||||
|
'redirect_uri': self.redirect_uri,
|
||||||
|
'uin': self.uin,
|
||||||
|
'sid': self.sid,
|
||||||
|
'skey': self.skey,
|
||||||
|
'pass_ticket': self.pass_ticket,
|
||||||
|
'synckey': self.synckey,
|
||||||
|
'device_id': self.device_id,
|
||||||
|
'last_login': time.time(),
|
||||||
|
}
|
||||||
|
cm = ConfigManager()
|
||||||
|
Log.debug('save wechat config')
|
||||||
|
cm.set_wechat_config(conf)
|
||||||
|
|
||||||
|
# save cookie
|
||||||
|
Log.debug('save cookie')
|
||||||
|
if self.cookie:
|
||||||
|
self.cookie.save(ignore_discard=True)
|
||||||
|
|
||||||
|
# save contacts
|
||||||
|
Log.debug('save contacts')
|
||||||
|
self.save_contacts()
|
||||||
|
except Exception, e:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
return False
|
||||||
|
return True
|
||||||
|
|
||||||
|
def recover(self):
|
||||||
|
"""
|
||||||
|
@brief Recover from snapshot data.
|
||||||
|
@return Bool: whether operation succeed.
|
||||||
|
"""
|
||||||
|
cm = ConfigManager()
|
||||||
|
[self.uuid, self.redirect_uri, self.uin,
|
||||||
|
self.sid, self.skey, self.pass_ticket,
|
||||||
|
self.synckey, device_id, self.last_login] = cm.get_wechat_config()
|
||||||
|
|
||||||
|
if device_id:
|
||||||
|
self.device_id = device_id
|
||||||
|
|
||||||
|
self.base_request = {
|
||||||
|
'Uin': int(self.uin),
|
||||||
|
'Sid': self.sid,
|
||||||
|
'Skey': self.skey,
|
||||||
|
'DeviceID': self.device_id,
|
||||||
|
}
|
||||||
|
|
||||||
|
# set cookie
|
||||||
|
Log.debug('set cookie')
|
||||||
|
self.cookie = set_cookie(self.cookie_file)
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
def save_contacts(self):
|
||||||
|
"""
|
||||||
|
@brief Save contacts.
|
||||||
|
"""
|
||||||
|
pickle_save(self.User, self.pickle_file['User'])
|
||||||
|
pickle_save(self.MemberList, self.pickle_file['MemberList'])
|
||||||
|
pickle_save(self.GroupList, self.pickle_file['GroupList'])
|
||||||
|
pickle_save(self.GroupMemeberList, self.pickle_file['GroupMemeberList'])
|
||||||
|
pickle_save(self.SpecialUsersList, self.pickle_file['SpecialUsersList'])
|
||||||
|
|
||||||
|
def recover_contacts(self):
|
||||||
|
"""
|
||||||
|
@brief recover contacts.
|
||||||
|
@return Bool: whether operation succeed.
|
||||||
|
"""
|
||||||
|
try:
|
||||||
|
self.User = pickle_load(self.pickle_file['User'])
|
||||||
|
self.MemberList = pickle_load(self.pickle_file['MemberList'])
|
||||||
|
self.GroupList = pickle_load(self.pickle_file['GroupList'])
|
||||||
|
self.GroupMemeberList = pickle_load(self.pickle_file['GroupMemeberList'])
|
||||||
|
self.SpecialUsersList = pickle_load(self.pickle_file['SpecialUsersList'])
|
||||||
|
return True
|
||||||
|
except Exception, e:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
return False
|
||||||
|
|
||||||
|
def handle_mod(self, r):
|
||||||
|
# ModContactCount: 变更联系人或群聊成员数目
|
||||||
|
# ModContactList: 变更联系人或群聊列表，或群名称改变
|
||||||
|
Log.debug('handle modify')
|
||||||
|
self.handle_msg(r)
|
||||||
|
for m in r['ModContactList']:
|
||||||
|
if m['UserName'][:2] == '@@':
|
||||||
|
# group
|
||||||
|
in_list = False
|
||||||
|
g_id = m['UserName']
|
||||||
|
for g in self.GroupList:
|
||||||
|
# group member change
|
||||||
|
if g_id == g['UserName']:
|
||||||
|
g['MemberCount'] = m['MemberCount']
|
||||||
|
g['NickName'] = m['NickName']
|
||||||
|
self.GroupMemeberList[g_id] = m['MemberList']
|
||||||
|
in_list = True
|
||||||
|
if self.msg_handler:
|
||||||
|
self.msg_handler.handle_group_member_change(g_id, m['MemberList'])
|
||||||
|
break
|
||||||
|
if not in_list:
|
||||||
|
# a new group
|
||||||
|
self.GroupList.append(m)
|
||||||
|
self.GroupMemeberList[g_id] = m['MemberList']
|
||||||
|
if self.msg_handler:
|
||||||
|
self.msg_handler.handle_group_list_change(m)
|
||||||
|
self.msg_handler.handle_group_member_change(g_id, m['MemberList'])
|
||||||
|
|
||||||
|
elif m['UserName'][0] == '@':
|
||||||
|
# user
|
||||||
|
in_list = False
|
||||||
|
for u in self.MemberList:
|
||||||
|
u_id = m['UserName']
|
||||||
|
if u_id == u['UserName']:
|
||||||
|
u = m
|
||||||
|
in_list = True
|
||||||
|
break
|
||||||
|
# if don't have then add it
|
||||||
|
if not in_list:
|
||||||
|
self.MemberList.append(m)
|
||||||
|
|
||||||
|
def handle_msg(self, r):
|
||||||
|
"""
|
||||||
|
@brief Recover from snapshot data.
|
||||||
|
@param r Dict: message json
|
||||||
|
"""
|
||||||
|
Log.debug('handle message')
|
||||||
|
if self.msg_handler:
|
||||||
|
self.msg_handler.handle_wxsync(r)
|
||||||
|
|
||||||
|
n = len(r['AddMsgList'])
|
||||||
|
if n == 0:
|
||||||
|
return
|
||||||
|
|
||||||
|
if self.log_mode:
|
||||||
|
echo(Constant.LOG_MSG_NEW_MSG % n)
|
||||||
|
|
||||||
|
for msg in r['AddMsgList']:
|
||||||
|
|
||||||
|
msgType = msg['MsgType']
|
||||||
|
msgId = msg['MsgId']
|
||||||
|
content = msg['Content'].replace('<', '<').replace('>', '>')
|
||||||
|
raw_msg = None
|
||||||
|
|
||||||
|
if msgType == self.wx_conf['MSGTYPE_TEXT']:
|
||||||
|
# 地理位置消息
|
||||||
|
if content.find('pictype=location') != -1:
|
||||||
|
location = content.split('<br/>')[1][:-1]
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'location': location,
|
||||||
|
'log': Constant.LOG_MSG_LOCATION % location
|
||||||
|
}
|
||||||
|
# 普通文本消息
|
||||||
|
else:
|
||||||
|
text = content.split(':<br/>')[-1]
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'text': text,
|
||||||
|
'log': text.replace('<br/>', '\n')
|
||||||
|
}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_IMAGE']:
|
||||||
|
data = self.webwxgetmsgimg(msgId)
|
||||||
|
fn = 'img_' + msgId + '.jpg'
|
||||||
|
dir = self.save_data_folders['webwxgetmsgimg']
|
||||||
|
path = save_file(fn, data, dir)
|
||||||
|
raw_msg = {'raw_msg': msg,
|
||||||
|
'image': path,
|
||||||
|
'log': Constant.LOG_MSG_PICTURE % path}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_VOICE']:
|
||||||
|
data = self.webwxgetvoice(msgId)
|
||||||
|
fn = 'voice_' + msgId + '.mp3'
|
||||||
|
dir = self.save_data_folders['webwxgetvoice']
|
||||||
|
path = save_file(fn, data, dir)
|
||||||
|
raw_msg = {'raw_msg': msg,
|
||||||
|
'voice': path,
|
||||||
|
'log': Constant.LOG_MSG_VOICE % path}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_SHARECARD']:
|
||||||
|
info = msg['RecommendInfo']
|
||||||
|
card = Constant.LOG_MSG_NAME_CARD % (
|
||||||
|
info['NickName'],
|
||||||
|
info['Alias'],
|
||||||
|
info['Province'], info['City'],
|
||||||
|
Constant.LOG_MSG_SEX_OPTION[info['Sex']]
|
||||||
|
)
|
||||||
|
namecard = '%s %s %s %s %s' % (
|
||||||
|
info['NickName'], info['Alias'], info['Province'],
|
||||||
|
info['City'], Constant.LOG_MSG_SEX_OPTION[info['Sex']]
|
||||||
|
)
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'namecard': namecard,
|
||||||
|
'log': card
|
||||||
|
}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_EMOTICON']:
|
||||||
|
url = search_content('cdnurl', content)
|
||||||
|
raw_msg = {'raw_msg': msg,
|
||||||
|
'emoticon': url,
|
||||||
|
'log': Constant.LOG_MSG_EMOTION % url}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_APP']:
|
||||||
|
card = ''
|
||||||
|
# 链接, 音乐, 微博
|
||||||
|
if msg['AppMsgType'] in [
|
||||||
|
self.wx_conf['APPMSGTYPE_AUDIO'],
|
||||||
|
self.wx_conf['APPMSGTYPE_URL'],
|
||||||
|
self.wx_conf['APPMSGTYPE_OPEN']
|
||||||
|
]:
|
||||||
|
card = Constant.LOG_MSG_APP_LINK % (
|
||||||
|
Constant.LOG_MSG_APP_LINK_TYPE[msg['AppMsgType']],
|
||||||
|
msg['FileName'],
|
||||||
|
search_content('des', content, 'xml'),
|
||||||
|
msg['Url'],
|
||||||
|
search_content('appname', content, 'xml')
|
||||||
|
)
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'link': msg['Url'],
|
||||||
|
'log': card
|
||||||
|
}
|
||||||
|
# 图片
|
||||||
|
elif msg['AppMsgType'] == self.wx_conf['APPMSGTYPE_IMG']:
|
||||||
|
data = self.webwxgetmsgimg(msgId)
|
||||||
|
fn = 'img_' + msgId + '.jpg'
|
||||||
|
dir = self.save_data_folders['webwxgetmsgimg']
|
||||||
|
path = save_file(fn, data, dir)
|
||||||
|
card = Constant.LOG_MSG_APP_IMG % (
|
||||||
|
path,
|
||||||
|
search_content('appname', content, 'xml')
|
||||||
|
)
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'image': path,
|
||||||
|
'log': card
|
||||||
|
}
|
||||||
|
else:
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'log': Constant.LOG_MSG_UNKNOWN_MSG % (msgType, content)
|
||||||
|
}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_STATUSNOTIFY']:
|
||||||
|
Log.info(Constant.LOG_MSG_NOTIFY_PHONE)
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_MICROVIDEO']:
|
||||||
|
data = self.webwxgetvideo(msgId)
|
||||||
|
fn = 'video_' + msgId + '.mp4'
|
||||||
|
dir = self.save_data_folders['webwxgetvideo']
|
||||||
|
path = save_file(fn, data, dir)
|
||||||
|
raw_msg = {'raw_msg': msg,
|
||||||
|
'video': path,
|
||||||
|
'log': Constant.LOG_MSG_VIDEO % path}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_RECALLED']:
|
||||||
|
recall_id = search_content('msgid', content, 'xml')
|
||||||
|
text = Constant.LOG_MSG_RECALL
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'text': text,
|
||||||
|
'recall_msg_id': recall_id,
|
||||||
|
'log': text
|
||||||
|
}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_SYS']:
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'sys_notif': content,
|
||||||
|
'log': content
|
||||||
|
}
|
||||||
|
elif msgType == self.wx_conf['MSGTYPE_VERIFYMSG']:
|
||||||
|
name = search_content('fromnickname', content)
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'log': Constant.LOG_MSG_ADD_FRIEND % name
|
||||||
|
}
|
||||||
|
else:
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg,
|
||||||
|
'log': Constant.LOG_MSG_UNKNOWN_MSG % (msgType, content)
|
||||||
|
}
|
||||||
|
|
||||||
|
isGroupMsg = '@@' in msg['FromUserName']+msg['ToUserName']
|
||||||
|
if self.msg_handler and raw_msg:
|
||||||
|
if isGroupMsg:
|
||||||
|
# handle group messages
|
||||||
|
g_msg = self.make_group_msg(raw_msg)
|
||||||
|
self.msg_handler.handle_group_msg(g_msg)
|
||||||
|
else:
|
||||||
|
# handle personal messages
|
||||||
|
self.msg_handler.handle_user_msg(raw_msg)
|
||||||
|
|
||||||
|
if self.log_mode:
|
||||||
|
self.show_msg(raw_msg)
|
||||||
|
|
||||||
|
def make_group_msg(self, msg):
|
||||||
|
"""
|
||||||
|
@brief Package the group message for storage.
|
||||||
|
@param msg Dict: raw msg
|
||||||
|
@return raw_msg Dict: packged msg
|
||||||
|
"""
|
||||||
|
Log.debug('make group message')
|
||||||
|
raw_msg = {
|
||||||
|
'raw_msg': msg['raw_msg'],
|
||||||
|
'msg_id': msg['raw_msg']['MsgId'],
|
||||||
|
'group_owner_uin': '',
|
||||||
|
'group_name': '',
|
||||||
|
'group_count': '',
|
||||||
|
'from_user_name': msg['raw_msg']['FromUserName'],
|
||||||
|
'to_user_name': msg['raw_msg']['ToUserName'],
|
||||||
|
'user_attrstatus': '',
|
||||||
|
'user_display_name': '',
|
||||||
|
'user_nickname': '',
|
||||||
|
'msg_type': msg['raw_msg']['MsgType'],
|
||||||
|
'text': '',
|
||||||
|
'link': '',
|
||||||
|
'image': '',
|
||||||
|
'video': '',
|
||||||
|
'voice': '',
|
||||||
|
'emoticon': '',
|
||||||
|
'namecard': '',
|
||||||
|
'location': '',
|
||||||
|
'recall_msg_id': '',
|
||||||
|
'sys_notif': '',
|
||||||
|
'time': '',
|
||||||
|
'timestamp': '',
|
||||||
|
'log': '',
|
||||||
|
}
|
||||||
|
content = msg['raw_msg']['Content'].replace(
|
||||||
|
'<', '<').replace('>', '>')
|
||||||
|
|
||||||
|
group = None
|
||||||
|
src = None
|
||||||
|
|
||||||
|
if msg['raw_msg']['FromUserName'][:2] == '@@':
|
||||||
|
# 接收到来自群的消息
|
||||||
|
g_id = msg['raw_msg']['FromUserName']
|
||||||
|
group = self.get_group_by_id(g_id)
|
||||||
|
|
||||||
|
if re.search(":<br/>", content, re.IGNORECASE):
|
||||||
|
u_id = content.split(':<br/>')[0]
|
||||||
|
src = self.get_group_user_by_id(u_id, g_id)
|
||||||
|
|
||||||
|
elif msg['raw_msg']['ToUserName'][:2] == '@@':
|
||||||
|
# 自己发给群的消息
|
||||||
|
g_id = msg['raw_msg']['ToUserName']
|
||||||
|
u_id = msg['raw_msg']['FromUserName']
|
||||||
|
src = self.get_group_user_by_id(u_id, g_id)
|
||||||
|
group = self.get_group_by_id(g_id)
|
||||||
|
|
||||||
|
if src:
|
||||||
|
raw_msg['user_attrstatus'] = src['AttrStatus']
|
||||||
|
raw_msg['user_display_name'] = src['DisplayName']
|
||||||
|
raw_msg['user_nickname'] = src['NickName']
|
||||||
|
if group:
|
||||||
|
raw_msg['group_count'] = group['MemberCount']
|
||||||
|
raw_msg['group_owner_uin'] = group['OwnerUin']
|
||||||
|
raw_msg['group_name'] = group['ShowName']
|
||||||
|
|
||||||
|
raw_msg['timestamp'] = msg['raw_msg']['CreateTime']
|
||||||
|
t = time.localtime(float(raw_msg['timestamp']))
|
||||||
|
raw_msg['time'] = time.strftime("%Y-%m-%d %T", t)
|
||||||
|
|
||||||
|
for key in [
|
||||||
|
'text', 'link', 'image', 'video', 'voice',
|
||||||
|
'emoticon', 'namecard', 'location', 'log',
|
||||||
|
'recall_msg_id', 'sys_notif'
|
||||||
|
]:
|
||||||
|
if key in msg:
|
||||||
|
raw_msg[key] = msg[key]
|
||||||
|
|
||||||
|
return raw_msg
|
||||||
|
|
||||||
|
def show_msg(self, message):
|
||||||
|
"""
|
||||||
|
@brief Log the message to stdout
|
||||||
|
@param message Dict
|
||||||
|
"""
|
||||||
|
msg = message
|
||||||
|
src = None
|
||||||
|
dst = None
|
||||||
|
group = None
|
||||||
|
|
||||||
|
if msg and msg['raw_msg']:
|
||||||
|
|
||||||
|
content = msg['raw_msg']['Content']
|
||||||
|
content = content.replace('<', '<').replace('>', '>')
|
||||||
|
msg_id = msg['raw_msg']['MsgId']
|
||||||
|
|
||||||
|
if msg['raw_msg']['FromUserName'][:2] == '@@':
|
||||||
|
# 接收到来自群的消息
|
||||||
|
g_id = msg['raw_msg']['FromUserName']
|
||||||
|
group = self.get_group_by_id(g_id)
|
||||||
|
|
||||||
|
if re.search(":<br/>", content, re.IGNORECASE):
|
||||||
|
u_id = content.split(':<br/>')[0]
|
||||||
|
src = self.get_group_user_by_id(u_id, g_id)
|
||||||
|
dst = {'ShowName': 'GROUP'}
|
||||||
|
else:
|
||||||
|
u_id = msg['raw_msg']['ToUserName']
|
||||||
|
src = {'ShowName': 'SYSTEM'}
|
||||||
|
dst = self.get_group_user_by_id(u_id, g_id)
|
||||||
|
elif msg['raw_msg']['ToUserName'][:2] == '@@':
|
||||||
|
# 自己发给群的消息
|
||||||
|
g_id = msg['raw_msg']['ToUserName']
|
||||||
|
u_id = msg['raw_msg']['FromUserName']
|
||||||
|
group = self.get_group_by_id(g_id)
|
||||||
|
src = self.get_group_user_by_id(u_id, g_id)
|
||||||
|
dst = {'ShowName': 'GROUP'}
|
||||||
|
else:
|
||||||
|
# 非群聊消息
|
||||||
|
src = self.get_user_by_id(msg['raw_msg']['FromUserName'])
|
||||||
|
dst = self.get_user_by_id(msg['raw_msg']['ToUserName'])
|
||||||
|
|
||||||
|
if group:
|
||||||
|
echo('%s |%s| %s -> %s: %s\n' % (
|
||||||
|
msg_id,
|
||||||
|
trans_emoji(group['ShowName']),
|
||||||
|
trans_emoji(src['ShowName']),
|
||||||
|
dst['ShowName'],
|
||||||
|
trans_emoji(msg['log'])
|
||||||
|
))
|
||||||
|
else:
|
||||||
|
echo('%s %s -> %s: %s\n' % (
|
||||||
|
msg_id,
|
||||||
|
trans_emoji(src['ShowName']),
|
||||||
|
trans_emoji(dst['ShowName']),
|
||||||
|
trans_emoji(msg['log'])
|
||||||
|
))
|
||||||
@@ -0,0 +1,969 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from utils import *
|
||||||
|
from config import Constant
|
||||||
|
from config import Log
|
||||||
|
#---------------------------------------------------
|
||||||
|
import sys
|
||||||
|
import os
|
||||||
|
import cookielib
|
||||||
|
import random
|
||||||
|
import requests
|
||||||
|
import time
|
||||||
|
import xml.dom.minidom
|
||||||
|
# for media upload
|
||||||
|
import mimetypes
|
||||||
|
from requests_toolbelt.multipart.encoder import MultipartEncoder
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
|
||||||
|
class WXAPI(object):
|
||||||
|
|
||||||
|
def __init__(self, host):
|
||||||
|
self.wx_host = host
|
||||||
|
self.wx_filehost = ''
|
||||||
|
self.wx_conf = {}
|
||||||
|
# jsLogin时这个appid只能使用: wx782c26e4c19acffb
|
||||||
|
self.appid = Constant.API_APPID
|
||||||
|
self.uuid = ''
|
||||||
|
self.redirect_uri = ''
|
||||||
|
self.skey = ''
|
||||||
|
self.sid = ''
|
||||||
|
self.uin = ''
|
||||||
|
self.pass_ticket = ''
|
||||||
|
self.base_request = {}
|
||||||
|
self.synckey_dic = {}
|
||||||
|
self.synckey = ''
|
||||||
|
self.device_id = 'e' + repr(random.random())[2:17]
|
||||||
|
# device_id: 登录手机设备
|
||||||
|
# web wechat 的格式为: e123456789012345 (e+15位随机数)
|
||||||
|
# mobile wechat 的格式为: A1234567890abcde (A+15位随机数字或字母)
|
||||||
|
self.user_agent = Constant.API_USER_AGENT
|
||||||
|
self.cookie = None
|
||||||
|
|
||||||
|
self.conf_factory()
|
||||||
|
|
||||||
|
self.User = [] # 登陆账号信息
|
||||||
|
self.MemberList = [] # 好友+群聊+公众号+特殊账号
|
||||||
|
self.MemberCount = 0
|
||||||
|
self.ContactList = [] # 好友
|
||||||
|
self.GroupList = [] # 群
|
||||||
|
self.GroupMemeberList = {} # 群聊成员字典
|
||||||
|
# "group_id": [
|
||||||
|
# {member}, ...
|
||||||
|
# ]
|
||||||
|
self.PublicUsersList = [] # 公众号／服务号
|
||||||
|
self.SpecialUsersList = [] # 特殊账号
|
||||||
|
|
||||||
|
self.media_count = 0
|
||||||
|
|
||||||
|
def conf_factory(self):
|
||||||
|
e = self.wx_host # wx.qq.com
|
||||||
|
t, o, n = "login.weixin.qq.com", "file.wx.qq.com", "webpush.weixin.qq.com"
|
||||||
|
|
||||||
|
if e.find("wx2.qq.com") > -1:
|
||||||
|
t, o, n = "login.wx2.qq.com", "file.wx2.qq.com", "webpush.wx2.qq.com"
|
||||||
|
elif e.find("wx8.qq.com") > -1:
|
||||||
|
t, o, n = "login.wx8.qq.com", "file.wx8.qq.com", "webpush.wx8.qq.com"
|
||||||
|
elif e.find("qq.com") > -1:
|
||||||
|
t, o, n = "login.wx.qq.com", "file.wx.qq.com", "webpush.wx.qq.com"
|
||||||
|
elif e.find("web2.wechat.com") > -1:
|
||||||
|
t, o, n = "login.web2.wechat.com", "file.web2.wechat.com", "webpush.web2.wechat.com"
|
||||||
|
elif e.find("wechat.com") > -1:
|
||||||
|
t, o, n = "login.web.wechat.com", "file.web.wechat.com", "webpush.web.wechat.com"
|
||||||
|
|
||||||
|
self.wx_filehost = o
|
||||||
|
conf = {
|
||||||
|
'LANG': Constant.API_LANG,
|
||||||
|
'SpecialUsers': Constant.API_SPECIAL_USER,
|
||||||
|
'API_jsLogin': "https://" + t + "/jslogin",
|
||||||
|
'API_qrcode': "https://login.weixin.qq.com/l/",
|
||||||
|
'API_qrcode_img': "https://login.weixin.qq.com/qrcode/",
|
||||||
|
'API_login': "https://" + t + "/cgi-bin/mmwebwx-bin/login",
|
||||||
|
'API_synccheck': "https://" + n + "/cgi-bin/mmwebwx-bin/synccheck",
|
||||||
|
'API_webwxdownloadmedia': "https://" + o + "/cgi-bin/mmwebwx-bin/webwxgetmedia",
|
||||||
|
'API_webwxuploadmedia': "https://" + o + "/cgi-bin/mmwebwx-bin/webwxuploadmedia",
|
||||||
|
'API_webwxpreview': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxpreview",
|
||||||
|
'API_webwxinit': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxinit",
|
||||||
|
'API_webwxgetcontact': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxgetcontact",
|
||||||
|
'API_webwxsync': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxsync",
|
||||||
|
'API_webwxbatchgetcontact': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxbatchgetcontact",
|
||||||
|
'API_webwxgeticon': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxgeticon",
|
||||||
|
'API_webwxsendmsg': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxsendmsg",
|
||||||
|
'API_webwxsendmsgimg': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxsendmsgimg",
|
||||||
|
'API_webwxsendmsgvedio': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxsendvideomsg",
|
||||||
|
'API_webwxsendemoticon': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxsendemoticon",
|
||||||
|
'API_webwxsendappmsg': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxsendappmsg",
|
||||||
|
'API_webwxgetheadimg': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxgetheadimg",
|
||||||
|
'API_webwxgetmsgimg': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxgetmsgimg",
|
||||||
|
'API_webwxgetmedia': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxgetmedia",
|
||||||
|
'API_webwxgetvideo': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxgetvideo",
|
||||||
|
'API_webwxlogout': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxlogout",
|
||||||
|
'API_webwxgetvoice': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxgetvoice",
|
||||||
|
'API_webwxupdatechatroom': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxupdatechatroom",
|
||||||
|
'API_webwxcreatechatroom': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxcreatechatroom",
|
||||||
|
'API_webwxstatusnotify': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxstatusnotify",
|
||||||
|
'API_webwxcheckurl': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxcheckurl",
|
||||||
|
'API_webwxverifyuser': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxverifyuser",
|
||||||
|
'API_webwxfeedback': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxsendfeedback",
|
||||||
|
'API_webwxreport': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxstatreport",
|
||||||
|
'API_webwxsearch': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxsearchcontact",
|
||||||
|
'API_webwxoplog': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxoplog",
|
||||||
|
'API_checkupload': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxcheckupload",
|
||||||
|
'API_webwxrevokemsg': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxrevokemsg",
|
||||||
|
'API_webwxpushloginurl': "https://" + e + "/cgi-bin/mmwebwx-bin/webwxpushloginurl",
|
||||||
|
'CONTACTFLAG_CONTACT': 1,
|
||||||
|
'CONTACTFLAG_CHATCONTACT': 2,
|
||||||
|
'CONTACTFLAG_CHATROOMCONTACT': 4,
|
||||||
|
'CONTACTFLAG_BLACKLISTCONTACT': 8,
|
||||||
|
'CONTACTFLAG_DOMAINCONTACT': 16,
|
||||||
|
'CONTACTFLAG_HIDECONTACT': 32,
|
||||||
|
'CONTACTFLAG_FAVOURCONTACT': 64,
|
||||||
|
'CONTACTFLAG_3RDAPPCONTACT': 128,
|
||||||
|
'CONTACTFLAG_SNSBLACKLISTCONTACT': 256,
|
||||||
|
'CONTACTFLAG_NOTIFYCLOSECONTACT': 512,
|
||||||
|
'CONTACTFLAG_TOPCONTACT': 2048,
|
||||||
|
'MSGTYPE_TEXT': 1,
|
||||||
|
'MSGTYPE_IMAGE': 3,
|
||||||
|
'MSGTYPE_VOICE': 34,
|
||||||
|
'MSGTYPE_VIDEO': 43,
|
||||||
|
'MSGTYPE_MICROVIDEO': 62,
|
||||||
|
'MSGTYPE_EMOTICON': 47,
|
||||||
|
'MSGTYPE_APP': 49,
|
||||||
|
'MSGTYPE_VOIPMSG': 50,
|
||||||
|
'MSGTYPE_VOIPNOTIFY': 52,
|
||||||
|
'MSGTYPE_VOIPINVITE': 53,
|
||||||
|
'MSGTYPE_LOCATION': 48,
|
||||||
|
'MSGTYPE_STATUSNOTIFY': 51,
|
||||||
|
'MSGTYPE_SYSNOTICE': 9999,
|
||||||
|
'MSGTYPE_POSSIBLEFRIEND_MSG': 40,
|
||||||
|
'MSGTYPE_VERIFYMSG': 37,
|
||||||
|
'MSGTYPE_SHARECARD': 42,
|
||||||
|
'MSGTYPE_SYS': 10000,
|
||||||
|
'MSGTYPE_RECALLED': 10002,
|
||||||
|
'APPMSGTYPE_TEXT': 1,
|
||||||
|
'APPMSGTYPE_IMG': 2,
|
||||||
|
'APPMSGTYPE_AUDIO': 3,
|
||||||
|
'APPMSGTYPE_VIDEO': 4,
|
||||||
|
'APPMSGTYPE_URL': 5,
|
||||||
|
'APPMSGTYPE_ATTACH': 6,
|
||||||
|
'APPMSGTYPE_OPEN': 7,
|
||||||
|
'APPMSGTYPE_EMOJI': 8,
|
||||||
|
'APPMSGTYPE_VOICE_REMIND': 9,
|
||||||
|
'APPMSGTYPE_SCAN_GOOD': 10,
|
||||||
|
'APPMSGTYPE_GOOD': 13,
|
||||||
|
'APPMSGTYPE_EMOTION': 15,
|
||||||
|
'APPMSGTYPE_CARD_TICKET': 16,
|
||||||
|
'APPMSGTYPE_REALTIME_SHARE_LOCATION': 17,
|
||||||
|
'APPMSGTYPE_TRANSFERS': 2e3,
|
||||||
|
'APPMSGTYPE_RED_ENVELOPES': 2001,
|
||||||
|
'APPMSGTYPE_READER_TYPE': 100001,
|
||||||
|
'UPLOAD_MEDIA_TYPE_IMAGE': 1,
|
||||||
|
'UPLOAD_MEDIA_TYPE_VIDEO': 2,
|
||||||
|
'UPLOAD_MEDIA_TYPE_AUDIO': 3,
|
||||||
|
'UPLOAD_MEDIA_TYPE_ATTACHMENT': 4,
|
||||||
|
}
|
||||||
|
self.wx_conf = conf
|
||||||
|
|
||||||
|
def getuuid(self):
|
||||||
|
"""
|
||||||
|
@brief Gets the uuid just used for login.
|
||||||
|
@return Bool: whether operation succeed.
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_jsLogin']
|
||||||
|
params = {
|
||||||
|
'appid': self.appid,
|
||||||
|
'fun': 'new',
|
||||||
|
'lang': self.wx_conf['LANG'],
|
||||||
|
'_': int(time.time()),
|
||||||
|
}
|
||||||
|
data = post(url, params, False)
|
||||||
|
regx = r'window.QRLogin.code = (\d+); window.QRLogin.uuid = "(\S+?)"'
|
||||||
|
pm = re.search(regx, data)
|
||||||
|
if pm:
|
||||||
|
code = pm.group(1)
|
||||||
|
self.uuid = pm.group(2)
|
||||||
|
return code == '200'
|
||||||
|
return False
|
||||||
|
|
||||||
|
def genqrcode(self):
|
||||||
|
"""
|
||||||
|
@brief outprint the qrcode to stdout on macos/linux
|
||||||
|
or open image on windows
|
||||||
|
"""
|
||||||
|
if sys.platform.startswith('win'):
|
||||||
|
url = self.wx_conf['API_qrcode_img'] + self.uuid
|
||||||
|
params = {
|
||||||
|
't': 'webwx',
|
||||||
|
'_': int(time.time())
|
||||||
|
}
|
||||||
|
data = post(url, params, False)
|
||||||
|
if data == '':
|
||||||
|
return
|
||||||
|
qrcode_path = save_file('qrcode.jpg', data, './')
|
||||||
|
os.startfile(qrcode_path)
|
||||||
|
else:
|
||||||
|
str2qr_terminal(self.wx_conf['API_qrcode'] + self.uuid)
|
||||||
|
|
||||||
|
def waitforlogin(self, tip=1):
|
||||||
|
"""
|
||||||
|
@brief wait for scaning qrcode to login
|
||||||
|
@param tip 1: wait for scan qrcode
|
||||||
|
0: wait for confirm
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
time.sleep(tip)
|
||||||
|
url = self.wx_conf['API_login'] + '?tip=%s&uuid=%s&_=%s' % (
|
||||||
|
tip, self.uuid, int(time.time()))
|
||||||
|
data = get(url)
|
||||||
|
pm = re.search(r'window.code=(\d+);', data)
|
||||||
|
code = pm.group(1)
|
||||||
|
|
||||||
|
if code == '201':
|
||||||
|
return True
|
||||||
|
elif code == '200':
|
||||||
|
pm = re.search(r'window.redirect_uri="(\S+?)";', data)
|
||||||
|
r_uri = pm.group(1) + '&fun=new'
|
||||||
|
self.redirect_uri = r_uri
|
||||||
|
self.wx_host = r_uri.split('://')[1].split('/')[0]
|
||||||
|
self.conf_factory()
|
||||||
|
return True
|
||||||
|
elif code == '408':
|
||||||
|
echo(Constant.LOG_MSG_WAIT_LOGIN_ERR1)
|
||||||
|
else:
|
||||||
|
echo(Constant.LOG_MSG_WAIT_LOGIN_ERR2)
|
||||||
|
return False
|
||||||
|
|
||||||
|
def login(self):
|
||||||
|
"""
|
||||||
|
@brief login
|
||||||
|
redirect_uri 有效时间是从扫码成功后算起，
|
||||||
|
大概是 300 秒，在此期间可以重新登录，但获取的联系人和群ID会改变
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
data = get(self.redirect_uri)
|
||||||
|
doc = xml.dom.minidom.parseString(data)
|
||||||
|
root = doc.documentElement
|
||||||
|
|
||||||
|
for node in root.childNodes:
|
||||||
|
if node.nodeName == 'ret':
|
||||||
|
if node.childNodes[0].data != "0":
|
||||||
|
return False
|
||||||
|
elif node.nodeName == 'skey':
|
||||||
|
self.skey = node.childNodes[0].data
|
||||||
|
elif node.nodeName == 'wxsid':
|
||||||
|
self.sid = node.childNodes[0].data
|
||||||
|
elif node.nodeName == 'wxuin':
|
||||||
|
self.uin = node.childNodes[0].data
|
||||||
|
elif node.nodeName == 'pass_ticket':
|
||||||
|
self.pass_ticket = node.childNodes[0].data
|
||||||
|
|
||||||
|
if '' in (self.skey, self.sid, self.uin, self.pass_ticket):
|
||||||
|
return False
|
||||||
|
|
||||||
|
self.base_request = {
|
||||||
|
'Uin': int(self.uin),
|
||||||
|
'Sid': self.sid,
|
||||||
|
'Skey': self.skey,
|
||||||
|
'DeviceID': self.device_id,
|
||||||
|
}
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
def webwxinit(self):
|
||||||
|
"""
|
||||||
|
@brief wechat initial
|
||||||
|
掉线后 300 秒可以重新使用此 api 登录
|
||||||
|
获取的联系人和群ID保持不变
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxinit'] + \
|
||||||
|
'?pass_ticket=%s&skey=%s&r=%s' % (
|
||||||
|
self.pass_ticket, self.skey, int(time.time())
|
||||||
|
)
|
||||||
|
params = {
|
||||||
|
'BaseRequest': self.base_request
|
||||||
|
}
|
||||||
|
dic = post(url, params)
|
||||||
|
self.User = dic['User']
|
||||||
|
self.make_synckey(dic)
|
||||||
|
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def webwxstatusnotify(self):
|
||||||
|
"""
|
||||||
|
@brief notify the mobile phone, this not necessary
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxstatusnotify'] + \
|
||||||
|
'?lang=%s&pass_ticket=%s' % (
|
||||||
|
self.wx_conf['LANG'], self.pass_ticket
|
||||||
|
)
|
||||||
|
params = {
|
||||||
|
'BaseRequest': self.base_request,
|
||||||
|
"Code": 3,
|
||||||
|
"FromUserName": self.User['UserName'],
|
||||||
|
"ToUserName": self.User['UserName'],
|
||||||
|
"ClientMsgId": int(time.time())
|
||||||
|
}
|
||||||
|
dic = post(url, params)
|
||||||
|
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def webwxgetcontact(self):
|
||||||
|
"""
|
||||||
|
@brief get all contacts: people, group, public user, special user
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
SpecialUsers = self.wx_conf['SpecialUsers']
|
||||||
|
url = self.wx_conf['API_webwxgetcontact'] + \
|
||||||
|
'?pass_ticket=%s&skey=%s&r=%s' % (
|
||||||
|
self.pass_ticket, self.skey, int(time.time())
|
||||||
|
)
|
||||||
|
dic = post(url, {})
|
||||||
|
|
||||||
|
self.MemberCount = dic['MemberCount']
|
||||||
|
self.MemberList = dic['MemberList']
|
||||||
|
ContactList = self.MemberList[:]
|
||||||
|
GroupList = self.GroupList[:]
|
||||||
|
PublicUsersList = self.PublicUsersList[:]
|
||||||
|
SpecialUsersList = self.SpecialUsersList[:]
|
||||||
|
|
||||||
|
for i in xrange(len(ContactList) - 1, -1, -1):
|
||||||
|
Contact = ContactList[i]
|
||||||
|
if Contact['VerifyFlag'] & 8 != 0: # 公众号/服务号
|
||||||
|
ContactList.remove(Contact)
|
||||||
|
self.PublicUsersList.append(Contact)
|
||||||
|
elif Contact['UserName'] in SpecialUsers: # 特殊账号
|
||||||
|
ContactList.remove(Contact)
|
||||||
|
self.SpecialUsersList.append(Contact)
|
||||||
|
elif Contact['UserName'].find('@@') != -1: # 群聊
|
||||||
|
ContactList.remove(Contact)
|
||||||
|
self.GroupList.append(Contact)
|
||||||
|
elif Contact['UserName'] == self.User['UserName']: # 自己
|
||||||
|
ContactList.remove(Contact)
|
||||||
|
self.ContactList = ContactList
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
def webwxbatchgetcontact(self, gid_list):
|
||||||
|
"""
|
||||||
|
@brief get group contacts
|
||||||
|
@param gid_list, The list of group id
|
||||||
|
@return List, list of group contacts
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxbatchgetcontact'] + \
|
||||||
|
'?type=ex&r=%s&pass_ticket=%s' % (
|
||||||
|
int(time.time()), self.pass_ticket
|
||||||
|
)
|
||||||
|
params = {
|
||||||
|
'BaseRequest': self.base_request,
|
||||||
|
"Count": len(gid_list),
|
||||||
|
"List": [{"UserName": gid, "EncryChatRoomId": ""} for gid in gid_list]
|
||||||
|
}
|
||||||
|
dic = post(url, params)
|
||||||
|
return dic['ContactList']
|
||||||
|
|
||||||
|
def synccheck(self):
|
||||||
|
"""
|
||||||
|
@brief check whether there's a message
|
||||||
|
@return [retcode, selector]
|
||||||
|
retcode: 0 successful
|
||||||
|
1100 logout
|
||||||
|
1101 login otherwhere
|
||||||
|
selector: 0 nothing
|
||||||
|
2 message
|
||||||
|
6 unkonwn
|
||||||
|
7 webwxsync
|
||||||
|
"""
|
||||||
|
params = {
|
||||||
|
'r': int(time.time()),
|
||||||
|
'sid': self.sid,
|
||||||
|
'uin': self.uin,
|
||||||
|
'skey': self.skey,
|
||||||
|
'deviceid': self.device_id,
|
||||||
|
'synckey': self.synckey,
|
||||||
|
'_': int(time.time()),
|
||||||
|
}
|
||||||
|
url = self.wx_conf['API_synccheck'] + '?' + urllib.urlencode(params)
|
||||||
|
data = get(url)
|
||||||
|
reg = r'window.synccheck={retcode:"(\d+)",selector:"(\d+)"}'
|
||||||
|
pm = re.search(reg, data)
|
||||||
|
retcode = pm.group(1)
|
||||||
|
selector = pm.group(2)
|
||||||
|
return [retcode, selector]
|
||||||
|
|
||||||
|
def webwxsync(self):
|
||||||
|
"""
|
||||||
|
@brief sync the messages
|
||||||
|
@return Dict{}
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxsync'] + \
|
||||||
|
'?sid=%s&skey=%s&pass_ticket=%s' % (
|
||||||
|
self.sid, self.skey, self.pass_ticket
|
||||||
|
)
|
||||||
|
params = {
|
||||||
|
'BaseRequest': self.base_request,
|
||||||
|
'SyncKey': self.synckey_dic,
|
||||||
|
'rr': ~int(time.time())
|
||||||
|
}
|
||||||
|
dic = post(url, params)
|
||||||
|
|
||||||
|
if dic['BaseResponse']['Ret'] == 0:
|
||||||
|
self.make_synckey(dic)
|
||||||
|
return dic
|
||||||
|
|
||||||
|
def webwxgetmsgimg(self, msgid):
|
||||||
|
"""
|
||||||
|
@brief get image in message
|
||||||
|
@param msgid The id of message
|
||||||
|
@return binary data of image
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxgetmsgimg'] + \
|
||||||
|
'?MsgID=%s&skey=%s' % (msgid, self.skey)
|
||||||
|
data = get(url, api='webwxgetmsgimg')
|
||||||
|
return data
|
||||||
|
|
||||||
|
def webwxgetvoice(self, msgid):
|
||||||
|
"""
|
||||||
|
@brief get voice in message
|
||||||
|
@param msgid The id of message
|
||||||
|
@return binary data of voice
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxgetvoice'] + \
|
||||||
|
'?msgid=%s&skey=%s' % (msgid, self.skey)
|
||||||
|
data = get(url, api='webwxgetvoice')
|
||||||
|
return data
|
||||||
|
|
||||||
|
def webwxgetvideo(self, msgid):
|
||||||
|
"""
|
||||||
|
@brief get video in message
|
||||||
|
@param msgid The id of message
|
||||||
|
@return binary data of video
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxgetvideo'] + \
|
||||||
|
'?msgid=%s&skey=%s' % (msgid, self.skey)
|
||||||
|
data = get(url, api='webwxgetvideo')
|
||||||
|
return data
|
||||||
|
|
||||||
|
def webwxgeticon(self, id):
|
||||||
|
"""
|
||||||
|
@brief get user small icon
|
||||||
|
@param id String
|
||||||
|
@return binary data of icon
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxgeticon'] + \
|
||||||
|
'?username=%s&skey=%s' % (id, self.skey)
|
||||||
|
data = get(url, api='webwxgeticon')
|
||||||
|
return data
|
||||||
|
|
||||||
|
def webwxgetheadimg(self, id):
|
||||||
|
"""
|
||||||
|
@brief get user head image
|
||||||
|
@param id String
|
||||||
|
@return binary data of image
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxgetheadimg'] + \
|
||||||
|
'?username=%s&skey=%s' % (id, self.skey)
|
||||||
|
data = get(url, api='webwxgetheadimg')
|
||||||
|
return data
|
||||||
|
|
||||||
|
def webwxsendmsg(self, word, to='filehelper'):
|
||||||
|
"""
|
||||||
|
@brief send text message
|
||||||
|
@param word String
|
||||||
|
@param to User id
|
||||||
|
@return dic Dict
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxsendmsg'] + \
|
||||||
|
'?pass_ticket=%s' % (self.pass_ticket)
|
||||||
|
clientMsgId = str(int(time.time() * 1000)) + \
|
||||||
|
str(random.random())[:5].replace('.', '')
|
||||||
|
params = {
|
||||||
|
'BaseRequest': self.base_request,
|
||||||
|
'Msg': {
|
||||||
|
"Type": 1,
|
||||||
|
"Content": trans_coding(word),
|
||||||
|
"FromUserName": self.User['UserName'],
|
||||||
|
"ToUserName": to,
|
||||||
|
"LocalID": clientMsgId,
|
||||||
|
"ClientMsgId": clientMsgId
|
||||||
|
}
|
||||||
|
}
|
||||||
|
dic = post(url, params)
|
||||||
|
return dic
|
||||||
|
|
||||||
|
def webwxuploadmedia(self, file_path):
|
||||||
|
"""
|
||||||
|
@brief upload image
|
||||||
|
@param file_path String
|
||||||
|
@return Dict: json
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxuploadmedia'] + '?f=json'
|
||||||
|
# 计数器
|
||||||
|
self.media_count = self.media_count + 1
|
||||||
|
fn = file_path
|
||||||
|
# mime_type:
|
||||||
|
# 'application/pdf'
|
||||||
|
# 'image/jpeg'
|
||||||
|
# 'image/png'
|
||||||
|
# ...
|
||||||
|
mime_type = mimetypes.guess_type(fn, strict=False)[0]
|
||||||
|
if not mime_type:
|
||||||
|
mime_type = 'text/plain'
|
||||||
|
# 文档格式
|
||||||
|
# 微信服务器目前应该支持3种类型:
|
||||||
|
# pic 直接显示，包含图片，表情
|
||||||
|
# video 不清楚
|
||||||
|
# doc 显示为文件，包含PDF等
|
||||||
|
media_type = 'pic' if mime_type.split('/')[0] == 'image' else 'doc'
|
||||||
|
time_format = "%a %b %d %Y %T GMT%z (%Z)"
|
||||||
|
last_modifie_date = time.strftime(time_format, time.localtime())
|
||||||
|
file_size = os.path.getsize(fn)
|
||||||
|
pass_ticket = self.pass_ticket
|
||||||
|
client_media_id = str(int(time.time() * 1000)) + \
|
||||||
|
str(random.random())[:5].replace('.', '')
|
||||||
|
|
||||||
|
webwx_data_ticket = ''
|
||||||
|
for item in self.cookie:
|
||||||
|
if item.name == 'webwx_data_ticket':
|
||||||
|
webwx_data_ticket = item.value
|
||||||
|
break
|
||||||
|
if (webwx_data_ticket == ''):
|
||||||
|
Log.error("No Cookie\n")
|
||||||
|
return None
|
||||||
|
|
||||||
|
uploadmediarequest = json.dumps({
|
||||||
|
"BaseRequest": self.base_request,
|
||||||
|
"ClientMediaId": client_media_id,
|
||||||
|
"TotalLen": file_size,
|
||||||
|
"StartPos": 0,
|
||||||
|
"DataLen": file_size,
|
||||||
|
"MediaType": 4
|
||||||
|
}, ensure_ascii=False).encode('utf8')
|
||||||
|
|
||||||
|
multipart_encoder = MultipartEncoder(
|
||||||
|
fields={
|
||||||
|
'id': 'WU_FILE_' + str(self.media_count),
|
||||||
|
'name': fn,
|
||||||
|
'type': mime_type,
|
||||||
|
'lastModifieDate': last_modifie_date,
|
||||||
|
'size': str(file_size),
|
||||||
|
'mediatype': media_type,
|
||||||
|
'uploadmediarequest': uploadmediarequest,
|
||||||
|
'webwx_data_ticket': webwx_data_ticket,
|
||||||
|
'pass_ticket': pass_ticket,
|
||||||
|
'filename': (
|
||||||
|
fn,
|
||||||
|
open(fn, 'rb'),
|
||||||
|
mime_type.split('/')[1]
|
||||||
|
)
|
||||||
|
},
|
||||||
|
boundary=(
|
||||||
|
'-----------------------------'
|
||||||
|
'1575017231431605357584454111'
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
headers = {
|
||||||
|
'Host': self.wx_filehost,
|
||||||
|
'User-Agent': self.user_agent,
|
||||||
|
'Accept': (
|
||||||
|
'text/html,application/xhtml+xml,'
|
||||||
|
'application/xml;q=0.9,*/*;q=0.8'
|
||||||
|
),
|
||||||
|
'Accept-Language': 'en-US,en;q=0.5',
|
||||||
|
'Accept-Encoding': 'gzip, deflate',
|
||||||
|
'Referer': 'https://' + self.wx_host,
|
||||||
|
'Content-Type': multipart_encoder.content_type,
|
||||||
|
'Origin': 'https://' + self.wx_host,
|
||||||
|
'Connection': 'keep-alive',
|
||||||
|
'Pragma': 'no-cache',
|
||||||
|
'Cache-Control': 'no-cache',
|
||||||
|
}
|
||||||
|
|
||||||
|
r = requests.post(url, data=multipart_encoder, headers=headers)
|
||||||
|
if dic['BaseResponse']['Ret'] == 0:
|
||||||
|
return dic
|
||||||
|
return None
|
||||||
|
|
||||||
|
def webwxsendmsgimg(self, user_id, media_id):
|
||||||
|
"""
|
||||||
|
@brief send image
|
||||||
|
@param user_id String
|
||||||
|
@param media_id String
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxsendmsgimg'] + \
|
||||||
|
'?fun=async&f=json&pass_ticket=%s' % self.pass_ticket
|
||||||
|
clientMsgId = str(int(time.time() * 1000)) + \
|
||||||
|
str(random.random())[:5].replace('.', '')
|
||||||
|
data_json = {
|
||||||
|
"BaseRequest": self.base_request,
|
||||||
|
"Msg": {
|
||||||
|
"Type": 3,
|
||||||
|
"MediaId": media_id,
|
||||||
|
"FromUserName": self.User['UserName'],
|
||||||
|
"ToUserName": user_id,
|
||||||
|
"LocalID": clientMsgId,
|
||||||
|
"ClientMsgId": clientMsgId
|
||||||
|
}
|
||||||
|
}
|
||||||
|
r = post(url, data_json)
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def webwxsendemoticon(self, user_id, media_id):
|
||||||
|
"""
|
||||||
|
@brief send image
|
||||||
|
@param user_id String
|
||||||
|
@param media_id String
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxsendemoticon'] + \
|
||||||
|
'?fun=sys&f=json&pass_ticket=%s' % self.pass_ticket
|
||||||
|
clientMsgId = str(int(time.time() * 1000)) + \
|
||||||
|
str(random.random())[:5].replace('.', '')
|
||||||
|
data_json = {
|
||||||
|
"BaseRequest": self.base_request,
|
||||||
|
"Msg": {
|
||||||
|
"Type": 47,
|
||||||
|
"EmojiFlag": 2,
|
||||||
|
"MediaId": media_id,
|
||||||
|
"FromUserName": self.User['UserName'],
|
||||||
|
"ToUserName": user_id,
|
||||||
|
"LocalID": clientMsgId,
|
||||||
|
"ClientMsgId": clientMsgId
|
||||||
|
}
|
||||||
|
}
|
||||||
|
r = post(url, data_json)
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def webwxsendappmsg(self, user_id, data):
|
||||||
|
"""
|
||||||
|
@brief send app msg
|
||||||
|
@param user_id String
|
||||||
|
@param data Dict
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxsendappmsg'] + \
|
||||||
|
'?fun=sys&f=json&pass_ticket=%s' % self.pass_ticket
|
||||||
|
clientMsgId = str(int(time.time() * 1000)) + \
|
||||||
|
str(random.random())[:5].replace('.', '')
|
||||||
|
content = ''.join([
|
||||||
|
"<appmsg appid='%s' sdkver=''>" % data['appid'], # 可使用其它AppID
|
||||||
|
"<title>%s</title>" % data['title'],
|
||||||
|
"<des></des>",
|
||||||
|
"<action></action>",
|
||||||
|
"<type>%d</type>" % data['type'],
|
||||||
|
"<content></content>",
|
||||||
|
"<url></url>",
|
||||||
|
"<lowurl></lowurl>",
|
||||||
|
"<appattach>",
|
||||||
|
"<totallen>%d</totallen>" % data['totallen'],
|
||||||
|
"<attachid>%s</attachid>" % data['attachid'],
|
||||||
|
"<fileext>%s</fileext>" % data['fileext'],
|
||||||
|
"</appattach>",
|
||||||
|
"<extinfo></extinfo>",
|
||||||
|
"</appmsg>",
|
||||||
|
])
|
||||||
|
data_json = {
|
||||||
|
"BaseRequest": self.base_request,
|
||||||
|
"Msg": {
|
||||||
|
"Type": data['type'],
|
||||||
|
"Content": content,
|
||||||
|
"FromUserName": self.User['UserName'],
|
||||||
|
"ToUserName": user_id,
|
||||||
|
"LocalID": clientMsgId,
|
||||||
|
"ClientMsgId": clientMsgId
|
||||||
|
},
|
||||||
|
"Scene": 0
|
||||||
|
}
|
||||||
|
r = post(url, data_json)
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def webwxcreatechatroom(self, uid_arr):
|
||||||
|
"""
|
||||||
|
@brief create a chat group
|
||||||
|
@param uid_arr [String]
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxcreatechatroom'] + '?r=%s' % int(time.time())
|
||||||
|
params = {
|
||||||
|
'BaseRequest': self.base_request,
|
||||||
|
'Topic': '',
|
||||||
|
'MemberCount': len(uid_arr),
|
||||||
|
'MemberList': [{'UserName': uid} for uid in uid_arr],
|
||||||
|
}
|
||||||
|
dic = post(url, params)
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def webwxupdatechatroom(self, add_arr, del_arr, invite_arr):
|
||||||
|
"""
|
||||||
|
@brief add/delete/invite member in chat group
|
||||||
|
@param add_arr [uid: String]
|
||||||
|
@param del_arr [uid: String]
|
||||||
|
@param invite_arr [uid: String]
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxupdatechatroom'] + '?r=%s' % int(time.time())
|
||||||
|
params = {
|
||||||
|
'BaseRequest': self.base_request,
|
||||||
|
'ChatRoomName': '',
|
||||||
|
'NewTopic': '',
|
||||||
|
'AddMemberList': add_arr,
|
||||||
|
'DelMemberList': del_arr,
|
||||||
|
'InviteMemberList': invite_arr,
|
||||||
|
}
|
||||||
|
dic = post(url, params)
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def webwxrevokemsg(self, msgid, user_id, client_msgid):
|
||||||
|
"""
|
||||||
|
@brief revoke a message
|
||||||
|
@param msgid String
|
||||||
|
@param user_id String
|
||||||
|
@param client_msgid String
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxrevokemsg'] + '?r=%s' % int(time.time())
|
||||||
|
params = {
|
||||||
|
'BaseRequest': self.base_request,
|
||||||
|
'SvrMsgId': msgid,
|
||||||
|
'ToUserName': user_id,
|
||||||
|
'ClientMsgId': client_msgid
|
||||||
|
}
|
||||||
|
dic = post(url, params)
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def webwxpushloginurl(self, uin):
|
||||||
|
"""
|
||||||
|
@brief push a login confirm alert to mobile device
|
||||||
|
@param uin String
|
||||||
|
@return dic Dict
|
||||||
|
"""
|
||||||
|
url = self.wx_conf['API_webwxpushloginurl'] + '?uin=%s' % uin
|
||||||
|
dic = eval(get(url))
|
||||||
|
return dic
|
||||||
|
|
||||||
|
def association_login(self):
|
||||||
|
"""
|
||||||
|
@brief login without scan qrcode
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
if self.uin != '':
|
||||||
|
dic = self.webwxpushloginurl(self.uin)
|
||||||
|
if dic['ret'] == '0':
|
||||||
|
self.uuid = dic['uuid']
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
def send_text(self, user_id, text):
|
||||||
|
"""
|
||||||
|
@brief send text
|
||||||
|
@param user_id String
|
||||||
|
@param text String
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
dic = self.webwxsendmsg(text, user_id)
|
||||||
|
return dic['BaseResponse']['Ret'] == 0
|
||||||
|
|
||||||
|
def send_img(self, user_id, file_path):
|
||||||
|
"""
|
||||||
|
@brief send image
|
||||||
|
@param user_id String
|
||||||
|
@param file_path String
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
response = self.webwxuploadmedia(file_path)
|
||||||
|
media_id = ""
|
||||||
|
if response is not None:
|
||||||
|
media_id = response['MediaId']
|
||||||
|
return self.webwxsendmsgimg(user_id, media_id)
|
||||||
|
|
||||||
|
def send_emot(self, user_id, file_path):
|
||||||
|
"""
|
||||||
|
@brief send emotion
|
||||||
|
@param user_id String
|
||||||
|
@param file_path String
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
response = self.webwxuploadmedia(file_path)
|
||||||
|
media_id = ""
|
||||||
|
if response is not None:
|
||||||
|
media_id = response['MediaId']
|
||||||
|
return self.webwxsendemoticon(user_id, media_id)
|
||||||
|
|
||||||
|
def send_file(self, user_id, file_path):
|
||||||
|
"""
|
||||||
|
@brief send file
|
||||||
|
@param user_id String
|
||||||
|
@param file_path String
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
title = file_path.split('/')[-1]
|
||||||
|
data = {
|
||||||
|
'appid': Constant.API_WXAPPID,
|
||||||
|
'title': title,
|
||||||
|
'totallen': '',
|
||||||
|
'attachid': '',
|
||||||
|
'type': self.wx_conf['APPMSGTYPE_ATTACH'],
|
||||||
|
'fileext': title.split('.')[-1],
|
||||||
|
}
|
||||||
|
|
||||||
|
response = self.webwxuploadmedia(file_path)
|
||||||
|
if response is not None:
|
||||||
|
data['totallen'] = response['StartPos']
|
||||||
|
data['attachid'] = response['MediaId']
|
||||||
|
else:
|
||||||
|
Log.error('File upload error')
|
||||||
|
|
||||||
|
return self.webwxsendappmsg(user_id, data)
|
||||||
|
|
||||||
|
def make_synckey(self, dic):
|
||||||
|
self.synckey_dic = dic['SyncKey']
|
||||||
|
|
||||||
|
def foo(x):
|
||||||
|
return str(x['Key']) + '_' + str(x['Val'])
|
||||||
|
|
||||||
|
# synckey for synccheck
|
||||||
|
self.synckey = '|'.join(
|
||||||
|
[foo(keyVal) for keyVal in self.synckey_dic['List']])
|
||||||
|
|
||||||
|
def revoke_msg(self, msgid, user_id, client_msgid):
|
||||||
|
"""
|
||||||
|
@brief revoke a message
|
||||||
|
@param msgid String
|
||||||
|
@param user_id String
|
||||||
|
@param client_msgid String
|
||||||
|
@return Bool: whether operation succeed
|
||||||
|
"""
|
||||||
|
return self.webwxrevokemsg(msgid, user_id, client_msgid)
|
||||||
|
|
||||||
|
# -----------------------------------------------------
|
||||||
|
# The following is getting user & group infomation apis
|
||||||
|
def get_user_by_id(self, user_id):
|
||||||
|
"""
|
||||||
|
@brief get all type of name by user id
|
||||||
|
@param user_id The id of user
|
||||||
|
@return Dict: {
|
||||||
|
'UserName' # 微信动态ID
|
||||||
|
'RemarkName' # 备注
|
||||||
|
'NickName' # 微信昵称
|
||||||
|
'ShowName' # Log显示用的
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
UnknownPeople = Constant.LOG_MSG_UNKNOWN_NAME + user_id
|
||||||
|
name = {
|
||||||
|
'UserName': user_id,
|
||||||
|
'RemarkName': '',
|
||||||
|
'NickName': '',
|
||||||
|
'ShowName': '',
|
||||||
|
}
|
||||||
|
name['ShowName'] = UnknownPeople
|
||||||
|
|
||||||
|
# yourself
|
||||||
|
if user_id == self.User['UserName']:
|
||||||
|
name['RemarkName'] = self.User['RemarkName']
|
||||||
|
name['NickName'] = self.User['NickName']
|
||||||
|
name['ShowName'] = name['NickName']
|
||||||
|
else:
|
||||||
|
# 联系人
|
||||||
|
for member in self.MemberList:
|
||||||
|
if member['UserName'] == user_id:
|
||||||
|
r, n = member['RemarkName'], member['NickName']
|
||||||
|
name['RemarkName'] = r
|
||||||
|
name['NickName'] = n
|
||||||
|
name['ShowName'] = r if r else n
|
||||||
|
break
|
||||||
|
# 特殊帐号
|
||||||
|
for member in self.SpecialUsersList:
|
||||||
|
if member['UserName'] == user_id:
|
||||||
|
name['RemarkName'] = user_id
|
||||||
|
name['NickName'] = user_id
|
||||||
|
name['ShowName'] = user_id
|
||||||
|
break
|
||||||
|
|
||||||
|
return name
|
||||||
|
|
||||||
|
def get_group_user_by_id(self, user_id, group_id):
|
||||||
|
"""
|
||||||
|
@brief get group user by user id
|
||||||
|
@param user_id The id of user
|
||||||
|
@param group_id The id of group
|
||||||
|
@return Dict: {
|
||||||
|
'UserName' # 微信动态ID
|
||||||
|
'NickName' # 微信昵称
|
||||||
|
'DisplayName' # 群聊显示名称
|
||||||
|
'ShowName' # Log显示用的
|
||||||
|
'AttrStatus' # 群用户id
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
UnknownPeople = Constant.LOG_MSG_UNKNOWN_NAME + user_id
|
||||||
|
name = {
|
||||||
|
'UserName': user_id,
|
||||||
|
'NickName': '',
|
||||||
|
'DisplayName': '',
|
||||||
|
'ShowName': '',
|
||||||
|
'AttrStatus': '',
|
||||||
|
}
|
||||||
|
name['ShowName'] = UnknownPeople
|
||||||
|
|
||||||
|
# 群友
|
||||||
|
if group_id in self.GroupMemeberList:
|
||||||
|
for member in self.GroupMemeberList[group_id]:
|
||||||
|
if member['UserName'] == user_id:
|
||||||
|
n, d = member['NickName'], member['DisplayName']
|
||||||
|
name['NickName'] = n
|
||||||
|
name['DisplayName'] = d
|
||||||
|
name['AttrStatus'] = member['AttrStatus']
|
||||||
|
name['ShowName'] = d if d else n
|
||||||
|
break
|
||||||
|
|
||||||
|
return name
|
||||||
|
|
||||||
|
def get_group_by_id(self, group_id):
|
||||||
|
"""
|
||||||
|
@brief get basic info by group id
|
||||||
|
@param group_id The id of group
|
||||||
|
@return Dict: {
|
||||||
|
'UserName' # 微信动态ID
|
||||||
|
'NickName' # 微信昵称
|
||||||
|
'DisplayName' # 群聊显示名称
|
||||||
|
'ShowName' # Log显示用的
|
||||||
|
'OwnerUin' # 群主ID
|
||||||
|
'MemberCount' # 群人数
|
||||||
|
}
|
||||||
|
"""
|
||||||
|
UnknownGroup = Constant.LOG_MSG_UNKNOWN_GROUP_NAME + group_id
|
||||||
|
group = {
|
||||||
|
'UserName': group_id,
|
||||||
|
'NickName': '',
|
||||||
|
'DisplayName': '',
|
||||||
|
'ShowName': '',
|
||||||
|
'OwnerUin': '',
|
||||||
|
'MemberCount': '',
|
||||||
|
}
|
||||||
|
|
||||||
|
for member in self.GroupList:
|
||||||
|
if member['UserName'] == group_id:
|
||||||
|
group['NickName'] = member['NickName']
|
||||||
|
group['DisplayName'] = member.get('DisplayName', '')
|
||||||
|
group['ShowName'] = member.get('NickName', UnknownGroup)
|
||||||
|
group['OwnerUin'] = member.get('OwnerUin', '')
|
||||||
|
group['MemberCount'] = member['MemberCount']
|
||||||
|
break
|
||||||
|
|
||||||
|
return group
|
||||||
|
|
||||||
|
def get_user_id(self, name):
|
||||||
|
"""
|
||||||
|
@brief Gets the user id.
|
||||||
|
@param name The user nickname or remarkname
|
||||||
|
@return The user id.
|
||||||
|
"""
|
||||||
|
for member in self.MemberList:
|
||||||
|
if name == member['RemarkName'] or name == member['NickName']:
|
||||||
|
return member['UserName']
|
||||||
|
return None
|
||||||
Executable
+278
@@ -0,0 +1,278 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from wechat import WeChat
|
||||||
|
from wechat.utils import *
|
||||||
|
from wx_handler import WeChatMsgProcessor
|
||||||
|
from wx_handler import Bot
|
||||||
|
from db import SqliteDB
|
||||||
|
from db import MysqlDB
|
||||||
|
from config import ConfigManager
|
||||||
|
from config import Constant
|
||||||
|
from config import Log
|
||||||
|
#---------------------------------------------------
|
||||||
|
import flask
|
||||||
|
import threading
|
||||||
|
import traceback
|
||||||
|
import os
|
||||||
|
import logging
|
||||||
|
import time
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
|
||||||
|
cm = ConfigManager()
|
||||||
|
db = SqliteDB(cm.getpath('database'))
|
||||||
|
# db = MysqlDB(cm.mysql())
|
||||||
|
wechat_msg_processor = WeChatMsgProcessor(db)
|
||||||
|
wechat = WeChat(cm.get('wechat', 'host'))
|
||||||
|
wechat.db = db
|
||||||
|
wechat.bot = Bot()
|
||||||
|
wechat.msg_handler = wechat_msg_processor
|
||||||
|
wechat_msg_processor.wechat = wechat
|
||||||
|
|
||||||
|
PORT = int(cm.get('setting', 'server_port'))
|
||||||
|
app = flask.Flask(__name__)
|
||||||
|
app.config['UPLOAD_FOLDER'] = cm.getpath('uploaddir')
|
||||||
|
|
||||||
|
logger = logging.getLogger('werkzeug')
|
||||||
|
log_format_str = Constant.SERVER_LOG_FORMAT
|
||||||
|
formatter = logging.Formatter(log_format_str)
|
||||||
|
flask_log_handler = logging.FileHandler(cm.getpath('server_log_file'))
|
||||||
|
flask_log_handler.setLevel(logging.INFO)
|
||||||
|
flask_log_handler.setFormatter(formatter)
|
||||||
|
logger.addHandler(flask_log_handler)
|
||||||
|
app.logger.addHandler(flask_log_handler)
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/')
|
||||||
|
def index():
|
||||||
|
with open(Constant.SERVER_PAGE_INDEX, 'r') as f:
|
||||||
|
return f.read()
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/qrcode')
|
||||||
|
def qrcode():
|
||||||
|
qdir = cm.getpath('qrcodedir')
|
||||||
|
if not os.path.exists(qdir):
|
||||||
|
os.makedirs(qdir)
|
||||||
|
image_path = '%s/%s_%d.png' % (qdir, wechat.uuid, int(time.time()*100))
|
||||||
|
s = wechat.wx_conf['API_qrcode'] + wechat.uuid
|
||||||
|
str2qr_image(s, image_path)
|
||||||
|
return flask.send_file(image_path, mimetype='image/png')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route("/group_list")
|
||||||
|
def group_list():
|
||||||
|
"""
|
||||||
|
@brief list groups
|
||||||
|
"""
|
||||||
|
result = wechat.db.select(Constant.TABLE_GROUP_LIST())
|
||||||
|
j = {'count': len(result), 'group': result}
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/group_member_list/<g_id>')
|
||||||
|
def group_member_list(g_id):
|
||||||
|
"""
|
||||||
|
@brief list group member
|
||||||
|
@param g_id String
|
||||||
|
"""
|
||||||
|
result = wechat.db.select(Constant.TABLE_GROUP_USER_LIST(), 'RoomID', g_id)
|
||||||
|
j = {'count': len(result), 'member': result}
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/group_chat_log/<g_name>')
|
||||||
|
def group_chat_log(g_name):
|
||||||
|
"""
|
||||||
|
@brief list group chat log
|
||||||
|
@param g_name String
|
||||||
|
"""
|
||||||
|
result = wechat.db.select(Constant.TABLE_GROUP_MSG_LOG, 'RoomName', g_name)
|
||||||
|
j = {'count': len(result), 'chats': result}
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/upload', methods=['GET', 'POST'])
|
||||||
|
def upload_file():
|
||||||
|
if flask.request.method == 'POST':
|
||||||
|
def allowed_file(filename):
|
||||||
|
return '.' in filename and \
|
||||||
|
filename.rsplit('.', 1)[1] in Constant.SERVER_UPLOAD_ALLOWED_EXTENSIONS
|
||||||
|
|
||||||
|
j = {'ret': 1, 'msg': ''}
|
||||||
|
|
||||||
|
# check if the post request has the file part
|
||||||
|
if 'file' not in flask.request.files:
|
||||||
|
j['msg'] = 'No file part'
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
# if user does not select file, browser also
|
||||||
|
# submit a empty part without filename
|
||||||
|
file = flask.request.files['file']
|
||||||
|
if file.filename == '':
|
||||||
|
j['msg'] = 'No selected file'
|
||||||
|
elif file and allowed_file(file.filename):
|
||||||
|
filename = generate_file_name(file.filename)
|
||||||
|
file_path = os.path.join(app.config['UPLOAD_FOLDER'], filename)
|
||||||
|
file.save(file_path)
|
||||||
|
j['ret'] = 0
|
||||||
|
j['msg'] = filename
|
||||||
|
else:
|
||||||
|
j['msg'] = 'File type not support'
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
else:
|
||||||
|
with open(Constant.SERVER_PAGE_UPLOAD, 'r') as f:
|
||||||
|
return f.read()
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/send_msg/<to>/<msg>')
|
||||||
|
def send_msg(to, msg):
|
||||||
|
"""
|
||||||
|
@brief send message to user or gourp
|
||||||
|
@param to: String, user id or group id
|
||||||
|
@param msg: String, words
|
||||||
|
"""
|
||||||
|
j = {'ret': 0 if wechat.send_text(to, msg) else 1}
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/send_img/<to>/<img>')
|
||||||
|
def send_img(to, img):
|
||||||
|
"""
|
||||||
|
@brief send image to user or gourp
|
||||||
|
@param to: String, user id or group id
|
||||||
|
@param img: String, image file name
|
||||||
|
"""
|
||||||
|
img_path = os.path.join(app.config['UPLOAD_FOLDER'], img)
|
||||||
|
j = {'ret': 0 if wechat.send_img(to, img_path) else 1}
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/send_emot/<to>/<emot>')
|
||||||
|
def send_emot(to, emot):
|
||||||
|
"""
|
||||||
|
@brief send emotion to user or gourp
|
||||||
|
@param to: String, user id or group id
|
||||||
|
@param emot: String, emotion file name
|
||||||
|
"""
|
||||||
|
emot_path = os.path.join(app.config['UPLOAD_FOLDER'], emot)
|
||||||
|
j = {'ret': 0 if wechat.send_emot(to, emot_path) else 1}
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/send_file/<to>/<file>')
|
||||||
|
def send_file(to, file):
|
||||||
|
"""
|
||||||
|
@brief send file to user or gourp
|
||||||
|
@param to: String, user id or group id
|
||||||
|
@param file: String, file name
|
||||||
|
"""
|
||||||
|
file_path = os.path.join(app.config['UPLOAD_FOLDER'], file)
|
||||||
|
j = {'ret': 0 if wechat.send_file(to, file_path) else 1}
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
def mass_send(method, data, func):
|
||||||
|
j = {'ret': -1, 'unsend_list':[]}
|
||||||
|
if method == 'POST' and data:
|
||||||
|
to_list = data['to_list']
|
||||||
|
msg = data['msg']
|
||||||
|
media_type = data.get('media_type', '')
|
||||||
|
|
||||||
|
if media_type in ['img', 'emot']:
|
||||||
|
file_path = os.path.join(app.config['UPLOAD_FOLDER'], msg)
|
||||||
|
response = wechat.webwxuploadmedia(file_path)
|
||||||
|
if response is not None:
|
||||||
|
msg = response['MediaId']
|
||||||
|
elif media_type == 'file':
|
||||||
|
file_path = os.path.join(app.config['UPLOAD_FOLDER'], msg)
|
||||||
|
data = {
|
||||||
|
'appid': Constant.API_WXAPPID,
|
||||||
|
'title': msg,
|
||||||
|
'totallen': '',
|
||||||
|
'attachid': '',
|
||||||
|
'type': wechat.wx_conf['APPMSGTYPE_ATTACH'],
|
||||||
|
'fileext': msg.split('.')[-1],
|
||||||
|
}
|
||||||
|
response = wechat.webwxuploadmedia(file_path)
|
||||||
|
if response is not None:
|
||||||
|
data['totallen'] = response['StartPos']
|
||||||
|
data['attachid'] = response['MediaId']
|
||||||
|
else:
|
||||||
|
Log.error('File upload error')
|
||||||
|
msg = data
|
||||||
|
|
||||||
|
for groups in split_array(to_list, 20):
|
||||||
|
for g in groups:
|
||||||
|
r = func(g, msg)
|
||||||
|
if not r:
|
||||||
|
j['unsend_list'].append(g)
|
||||||
|
time.sleep(1)
|
||||||
|
|
||||||
|
j['ret'] = len(j['unsend_list'])
|
||||||
|
|
||||||
|
return j
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/mass_send_msg/', methods=["GET", "POST"])
|
||||||
|
def mass_send_msg():
|
||||||
|
"""
|
||||||
|
@brief send text to mass users or gourps
|
||||||
|
"""
|
||||||
|
j = mass_send(flask.request.method, flask.request.json, wechat.send_text)
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/mass_send_img', methods=["GET", "POST"])
|
||||||
|
def mass_send_img():
|
||||||
|
"""
|
||||||
|
@brief send iamge to mass users or gourps
|
||||||
|
"""
|
||||||
|
j = mass_send(flask.request.method, flask.request.json, wechat.webwxsendmsgimg)
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/mass_send_emot', methods=["GET", "POST"])
|
||||||
|
def mass_send_emot():
|
||||||
|
"""
|
||||||
|
@brief send emoticon to mass users or gourps
|
||||||
|
"""
|
||||||
|
j = mass_send(flask.request.method, flask.request.json, wechat.webwxsendemoticon)
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
@app.route('/mass_send_file', methods=["GET", "POST"])
|
||||||
|
def mass_send_file():
|
||||||
|
"""
|
||||||
|
@brief send file to mass users or gourps
|
||||||
|
"""
|
||||||
|
j = mass_send(flask.request.method, flask.request.json, wechat.webwxsendappmsg)
|
||||||
|
return flask.Response(json.dumps(j), mimetype='application/json')
|
||||||
|
|
||||||
|
|
||||||
|
def run_server():
|
||||||
|
app.run(port=PORT)
|
||||||
|
|
||||||
|
if cm.get('setting', 'server_mode') == 'True':
|
||||||
|
serverProcess = threading.Thread(target=run_server)
|
||||||
|
serverProcess.start()
|
||||||
|
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
wechat.start()
|
||||||
|
except KeyboardInterrupt:
|
||||||
|
echo(Constant.LOG_MSG_QUIT)
|
||||||
|
wechat.exit_code = 1
|
||||||
|
else:
|
||||||
|
Log.error(traceback.format_exc())
|
||||||
|
finally:
|
||||||
|
wechat.stop()
|
||||||
|
|
||||||
|
if wechat.exit_code == 0:
|
||||||
|
echo(Constant.MAIN_RESTART)
|
||||||
|
else:
|
||||||
|
# kill process
|
||||||
|
os.system(Constant.LOG_MSG_KILL_PROCESS % os.getpid())
|
||||||
|
|
||||||
@@ -0,0 +1,5 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
from wechat_msg_processor import WeChatMsgProcessor
|
||||||
|
from bot import Bot
|
||||||
@@ -0,0 +1,47 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from wechat.utils import *
|
||||||
|
from config import Constant
|
||||||
|
#---------------------------------------------------
|
||||||
|
import random, time, json
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
|
||||||
|
class Bot(object):
|
||||||
|
|
||||||
|
def __init__(self):
|
||||||
|
self.emoticons = Constant.EMOTICON
|
||||||
|
self.gifs = []
|
||||||
|
self.last_time = time.time()
|
||||||
|
|
||||||
|
def time_schedule(self):
|
||||||
|
r = ''
|
||||||
|
now = time.time()
|
||||||
|
if int(now - self.last_time) > 3600:
|
||||||
|
self.last_time = now
|
||||||
|
url_latest = Constant.BOT_ZHIHU_URL_LATEST
|
||||||
|
url_daily = Constant.BOT_ZHIHU_URL_DAILY
|
||||||
|
data = get(url_latest)
|
||||||
|
j = json.loads(data)
|
||||||
|
story = j['stories'][random.randint(0, len(j['stories'])-1)]
|
||||||
|
r = story['title'] + '\n' + url_daily + str(story['id'])
|
||||||
|
return r.encode('utf-8')
|
||||||
|
|
||||||
|
def reply(self, text):
|
||||||
|
APIKEY = Constant.BOT_TULING_API_KEY
|
||||||
|
api_url = Constant.BOT_TULING_API_URL % (APIKEY, text, '12345678')
|
||||||
|
r = json.loads(get(api_url))
|
||||||
|
if r.get('code') == 100000 and r.get('text') != Constant.BOT_TULING_BOT_REPLY:
|
||||||
|
p = random.randint(1, 10)
|
||||||
|
if p > 3:
|
||||||
|
return r['text']
|
||||||
|
elif p > 1:
|
||||||
|
# send emoji
|
||||||
|
if random.randint(1, 10) > 5:
|
||||||
|
n = random.randint(0, len(self.emoticons)-1)
|
||||||
|
m = random.randint(1, 3)
|
||||||
|
reply = self.emoticons[n].encode('utf-8') * m
|
||||||
|
return reply
|
||||||
|
return ''
|
||||||
@@ -0,0 +1,246 @@
|
|||||||
|
#!/usr/bin/env python
|
||||||
|
# coding: utf-8
|
||||||
|
|
||||||
|
#===================================================
|
||||||
|
from wechat.utils import *
|
||||||
|
from config import ConfigManager
|
||||||
|
from config import Constant
|
||||||
|
from config import Log
|
||||||
|
#---------------------------------------------------
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
import json
|
||||||
|
import re
|
||||||
|
#===================================================
|
||||||
|
|
||||||
|
|
||||||
|
class WeChatMsgProcessor(object):
|
||||||
|
"""
|
||||||
|
Process fetched data
|
||||||
|
"""
|
||||||
|
|
||||||
|
def __init__(self, db):
|
||||||
|
self.db = db
|
||||||
|
self.wechat = None # recieve `WeChat` class instance
|
||||||
|
# for call some wechat apis
|
||||||
|
|
||||||
|
# read config
|
||||||
|
cm = ConfigManager()
|
||||||
|
[self.upload_dir, self.data_dir, self.log_dir] = cm.setup_database()
|
||||||
|
|
||||||
|
def clean_db(self):
|
||||||
|
"""
|
||||||
|
@brief clean database, delete table & create table
|
||||||
|
"""
|
||||||
|
self.db.delete_table(Constant.TABLE_GROUP_LIST())
|
||||||
|
self.db.delete_table(Constant.TABLE_GROUP_USER_LIST())
|
||||||
|
self.db.create_table(Constant.TABLE_GROUP_MSG_LOG, Constant.TABLE_GROUP_MSG_LOG_COL)
|
||||||
|
self.db.create_table(Constant.TABLE_GROUP_LIST(), Constant.TABLE_GROUP_LIST_COL)
|
||||||
|
self.db.create_table(Constant.TABLE_GROUP_USER_LIST(), Constant.TABLE_GROUP_USER_LIST_COL)
|
||||||
|
self.db.create_table(Constant.TABLE_RECORD_ENTER_GROUP, Constant.TABLE_RECORD_ENTER_GROUP_COL)
|
||||||
|
self.db.create_table(Constant.TABLE_RECORD_RENAME_GROUP, Constant.TABLE_RECORD_RENAME_GROUP_COL)
|
||||||
|
|
||||||
|
def handle_wxsync(self, msg):
|
||||||
|
"""
|
||||||
|
@brief Recieve webwxsync message, saved into json
|
||||||
|
@param msg Dict: webwxsync msg
|
||||||
|
"""
|
||||||
|
fn = time.strftime(Constant.LOG_MSG_FILE, time.localtime())
|
||||||
|
save_json(fn, msg, self.log_dir, 'a+')
|
||||||
|
|
||||||
|
def handle_group_list(self, group_list):
|
||||||
|
"""
|
||||||
|
@brief handle group list & saved in DB
|
||||||
|
@param group_list Array
|
||||||
|
"""
|
||||||
|
fn = Constant.LOG_MSG_GROUP_LIST_FILE
|
||||||
|
save_json(fn, group_list, self.data_dir)
|
||||||
|
cols = [(
|
||||||
|
g['NickName'],
|
||||||
|
g['UserName'],
|
||||||
|
g['OwnerUin'],
|
||||||
|
g['MemberCount'],
|
||||||
|
g['HeadImgUrl']
|
||||||
|
) for g in group_list]
|
||||||
|
self.db.insertmany(Constant.TABLE_GROUP_LIST(), cols)
|
||||||
|
|
||||||
|
def handle_group_member_list(self, group_id, member_list):
|
||||||
|
"""
|
||||||
|
@brief handle group member list & saved in DB
|
||||||
|
@param member_list Array
|
||||||
|
"""
|
||||||
|
fn = group_id + '.json'
|
||||||
|
save_json(fn, member_list, self.data_dir)
|
||||||
|
cols = [(
|
||||||
|
group_id,
|
||||||
|
m['UserName'],
|
||||||
|
m['NickName'],
|
||||||
|
m['DisplayName'],
|
||||||
|
m['AttrStatus']
|
||||||
|
) for m in member_list]
|
||||||
|
self.db.insertmany(Constant.TABLE_GROUP_USER_LIST(), cols)
|
||||||
|
|
||||||
|
def handle_group_list_change(self, new_group):
|
||||||
|
"""
|
||||||
|
@brief handle adding a new group & saved in DB
|
||||||
|
@param new_group Dict
|
||||||
|
"""
|
||||||
|
self.handle_group_list([new_group])
|
||||||
|
|
||||||
|
def handle_group_member_change(self, group_id, member_list):
|
||||||
|
"""
|
||||||
|
@brief handle group member changes & saved in DB
|
||||||
|
@param group_id Dict
|
||||||
|
@param member_list Dict
|
||||||
|
"""
|
||||||
|
self.db.delete(Constant.TABLE_GROUP_USER_LIST(), "RoomID", group_id)
|
||||||
|
self.handle_group_member_list(group_id, member_list)
|
||||||
|
|
||||||
|
def handle_group_msg(self, msg):
|
||||||
|
"""
|
||||||
|
@brief Recieve group messages
|
||||||
|
@param msg Dict: packaged msg
|
||||||
|
"""
|
||||||
|
# rename media files
|
||||||
|
for k in ['image', 'video', 'voice']:
|
||||||
|
if msg[k]:
|
||||||
|
t = time.localtime(float(msg['timestamp']))
|
||||||
|
time_str = time.strftime("%Y%m%d%H%M%S", t)
|
||||||
|
# format: 时间_消息ID_群名
|
||||||
|
file_name = '/%s_%s_%s.' % (time_str, msg['msg_id'], msg['group_name'])
|
||||||
|
new_name = re.sub(r'\/\w+\_\d+\.', file_name, msg[k])
|
||||||
|
Log.debug('rename file to %s' % new_name)
|
||||||
|
os.rename(msg[k], new_name)
|
||||||
|
msg[k] = new_name
|
||||||
|
|
||||||
|
if msg['msg_type'] == 10000:
|
||||||
|
# record member enter in group
|
||||||
|
m = re.search(r'邀请(.+)加入了群聊', msg['sys_notif'])
|
||||||
|
if m:
|
||||||
|
name = m.group(1)
|
||||||
|
col_enter_group = (
|
||||||
|
msg['msg_id'],
|
||||||
|
msg['group_name'],
|
||||||
|
msg['from_user_name'],
|
||||||
|
msg['to_user_name'],
|
||||||
|
name,
|
||||||
|
msg['time'],
|
||||||
|
)
|
||||||
|
self.db.insert(Constant.TABLE_RECORD_ENTER_GROUP, col_enter_group)
|
||||||
|
|
||||||
|
# record rename group
|
||||||
|
n = re.search(r'(.+)修改群名为“(.+)”', msg['sys_notif'])
|
||||||
|
if n:
|
||||||
|
people = n.group(1)
|
||||||
|
to_name = n.group(2)
|
||||||
|
col_rename_group = (
|
||||||
|
msg['msg_id'],
|
||||||
|
msg['group_name'],
|
||||||
|
to_name,
|
||||||
|
people,
|
||||||
|
msg['time'],
|
||||||
|
)
|
||||||
|
self.db.insert(Constant.TABLE_RECORD_RENAME_GROUP, col_rename_group)
|
||||||
|
|
||||||
|
# upadte group in GroupList
|
||||||
|
for g in self.wechat.GroupList:
|
||||||
|
if g['UserName'] == msg['from_user_name']:
|
||||||
|
g['NickName'] = to_name
|
||||||
|
break
|
||||||
|
|
||||||
|
# normal group message
|
||||||
|
col = (
|
||||||
|
msg['msg_id'],
|
||||||
|
msg['group_owner_uin'],
|
||||||
|
msg['group_name'],
|
||||||
|
msg['group_count'],
|
||||||
|
msg['from_user_name'],
|
||||||
|
msg['to_user_name'],
|
||||||
|
msg['user_attrstatus'],
|
||||||
|
msg['user_display_name'],
|
||||||
|
msg['user_nickname'],
|
||||||
|
msg['msg_type'],
|
||||||
|
msg['emoticon'],
|
||||||
|
msg['text'],
|
||||||
|
msg['image'],
|
||||||
|
msg['video'],
|
||||||
|
msg['voice'],
|
||||||
|
msg['link'],
|
||||||
|
msg['namecard'],
|
||||||
|
msg['location'],
|
||||||
|
msg['recall_msg_id'],
|
||||||
|
msg['sys_notif'],
|
||||||
|
msg['time'],
|
||||||
|
msg['timestamp']
|
||||||
|
)
|
||||||
|
self.db.insert(Constant.TABLE_GROUP_MSG_LOG, col)
|
||||||
|
|
||||||
|
text = msg['text']
|
||||||
|
if text and text[0] == '@':
|
||||||
|
n = trans_coding(text).find(u'\u2005')
|
||||||
|
name = trans_coding(text)[1:n].encode('utf-8')
|
||||||
|
if name in [self.wechat.User['NickName'], self.wechat.User['RemarkName']]:
|
||||||
|
self.handle_command(trans_coding(text)[n+1:].encode('utf-8'), msg)
|
||||||
|
|
||||||
|
def handle_user_msg(self, msg):
|
||||||
|
"""
|
||||||
|
@brief Recieve personal messages
|
||||||
|
@param msg Dict
|
||||||
|
"""
|
||||||
|
wechat = self.wechat
|
||||||
|
|
||||||
|
text = trans_coding(msg['text']).encode('utf-8')
|
||||||
|
uid = msg['raw_msg']['FromUserName']
|
||||||
|
|
||||||
|
if text == 'test_revoke': # 撤回消息测试
|
||||||
|
dic = wechat.webwxsendmsg('这条消息将被撤回', uid)
|
||||||
|
wechat.revoke_msg(dic['MsgID'], uid, dic['LocalID'])
|
||||||
|
|
||||||
|
|
||||||
|
def handle_command(self, cmd, msg):
|
||||||
|
"""
|
||||||
|
@brief handle msg of `@yourself cmd`
|
||||||
|
@param cmd String
|
||||||
|
@param msg Dict
|
||||||
|
"""
|
||||||
|
wechat = self.wechat
|
||||||
|
g_id = ''
|
||||||
|
for g in wechat.GroupList:
|
||||||
|
if g['NickName'] == msg['group_name']:
|
||||||
|
g_id = g['UserName']
|
||||||
|
|
||||||
|
cmd = cmd.strip()
|
||||||
|
if cmd == 'test':
|
||||||
|
# wechat.send_file(g_id, 'test/Data/upload/shake.wav')
|
||||||
|
wechat.send_img(g_id, 'test/emotion/7.gif')
|
||||||
|
elif cmd == 'runtime':
|
||||||
|
wechat.webwxsendmsg(wechat.get_run_time(), g_id)
|
||||||
|
else:
|
||||||
|
# reply bot
|
||||||
|
# ---------
|
||||||
|
# if wechat.bot:
|
||||||
|
# r = wechat.bot.reply(cmd)
|
||||||
|
# if r:
|
||||||
|
# wechat.webwxsendmsg(r, g_id)
|
||||||
|
# else:
|
||||||
|
# pass
|
||||||
|
img_name = [
|
||||||
|
'0.jpg', '1.jpeg', '2.gif', '3.jpg', '4.jpeg',
|
||||||
|
'5.gif', '6.gif', '7.gif', '8.jpg', '9.jpg'
|
||||||
|
]
|
||||||
|
name = img_name[int(time.time()) % 10]
|
||||||
|
emot_path = os.path.join('test/emotion/', name)
|
||||||
|
wechat.send_emot(g_id, emot_path)
|
||||||
|
|
||||||
|
def check_schedule_task(self):
|
||||||
|
# update group member list at 00:00 am every morning
|
||||||
|
t = time.localtime()
|
||||||
|
if t.tm_hour == 0 and t.tm_min <= 1:
|
||||||
|
# update group member
|
||||||
|
Log.debug('update group member list everyday')
|
||||||
|
self.db.delete_table(Constant.TABLE_GROUP_LIST())
|
||||||
|
self.db.delete_table(Constant.TABLE_GROUP_USER_LIST())
|
||||||
|
self.db.create_table(Constant.TABLE_GROUP_LIST(), Constant.TABLE_GROUP_LIST_COL)
|
||||||
|
self.db.create_table(Constant.TABLE_GROUP_USER_LIST(), Constant.TABLE_GROUP_USER_LIST_COL)
|
||||||
|
self.wechat.fetch_group_contacts()
|
||||||
|
|
||||||
Reference in New Issue
Block a user