-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest.py
179 lines (164 loc) · 6.97 KB
/
test.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
#!/usr/bin/env python
# coding: utf-8
#
from wxbot import *
import datetime
superAdmin = {
u'标admin', # name of super users who want to get notify
u'果果admin'
};
cmdGroupName = u"test群"
notifyGroupName = u"金智宝贝内勤队"
fowordMsg = u"%s 说：\n---------\n%s"
fowordPic = u"%s 发来图片："
fowordVoice = u"%s 发来语音："
fowordFile = u"%s 发来文件："
fowordOther = u"%s 发来：%s"
class MyWXBot(WXBot):
def got_contact(self):
self.superUser = []
for name in superAdmin:
uid = self.get_user_id(name)
if uid:
self.superUser.append(uid);
print "Super user:", name.encode("utf8"), uid
else :
print name.encode("utf8"), "NOT FOUND";
pass
pass
self.notifyToGroup = self.get_user_id(notifyGroupName)
if self.notifyToGroup:
print "Our Group:", notifyGroupName.encode("utf8"), self.notifyToGroup
self.cmdFromGroup = self.get_user_id(cmdGroupName)
if self.cmdFromGroup:
print "CMD Group:", cmdGroupName.encode("utf8"), self.cmdFromGroup
def handle_msg_all(self, msg):
# print "Content", msg['content']
# print "User:", msg['user']
# print "Msg", msg
if not self.handleMoneyInfo(msg):
if msg['msg_type_id'] == 4:
# if msg['user']['id'] in self.superUser :
# print "!!!!! from ADMIN !!!!!";
if not self.handleCommad(msg):
self.handleFoword(msg);
elif msg['msg_type_id'] == 3:
fromGroupId = msg['user']['id'];
content = msg['content']['data'];
contentType = msg['content']['type'];
fromUid = msg['content']['user']['id'];
fromUname = msg['content']['user']['name'];
if fromGroupId == self.cmdFromGroup and contentType == 0:
print fromUname , "CMD"
# print content
# print "================="
self.handleCommadContent(fromGroupId, content);
pass
# if msg['msg_type_id'] == 4 and msg['content']['type'] == 0:
# self.send_msg_by_uid(u'hi', msg['user']['id'])
# self.send_img_msg_by_uid("img/1.png", msg['user']['id'])
# self.send_file_msg_by_uid("img/1.png", msg['user']['id'])
def handleCommad(self, msg):
fromUid = msg['user']['id'];
content = msg['content']['data'];
return False
return self.handleCommadContent(fromUid, content);
def handleCommadContent(self, uid, content):
if content == "auto" :
t = datetime.datetime.now().strftime("%m-%d %H:%M")
self.send_msg_by_uid("我在线哦。\n" + t, uid)
else :
self.send_msg_by_uid("我在线。", uid)
return True;
def handleFoword(self, msg):
fromUid = msg['user']['id'];
fromName = msg['user']['name']
content = msg['content']['data'];
contentType = msg['content']['type'];
file_name = msg['content']['file_name'] if "file_name" in msg['content'] else None;
if contentType == 0:
self.sendToAdmin(fromUid, fowordMsg % (fromName, content))
elif contentType in [3, 4, 6, 8, 13]:
if (contentType == 3 or contentType == 6):
if file_name:
self.sendToAdmin(fromUid, fowordPic % (fromName), file_name, True)
else:
self.sendToAdmin(fromUid, fowordPic % (fromName) + "动画表情")
elif contentType == 4 :
self.sendToAdmin(fromUid, fowordVoice % (fromName), file_name)
pass
elif file_name:
self.sendToAdmin(fromUid, fowordFile % (fromName), file_name)
else :
self.sendToAdmin(fromUid, fowordOther % (fromName, msg['content']))
pass
elif contentType == 7 :
msg = u"%s分享了一个%s链接（来源：%s）\n%s\n%s" % \
(fromName, content['type'], content['from'], content['title'], content['desc'])
self.sendToAdmin(fromUid, msg)
self.sendToAdmin(fromUid, content['url'])
pass
elif contentType == 5 :
msg = u"%s分享了一张名片\n昵称：%s 性别：%s" % \
(fromName, content['nickname'], content['genderCN'])
self.sendToAdmin(fromUid, msg)
pass
# self.send_img_msg_by_uid("img/1.png", msg['user']['id'])
# self.send_file_msg_by_uid("img/1.png", msg['user']['id'])
pass
return False;
def sendToAdmin(self, fromUid, text=None, file_name = None, is_pic = False, to_group = False):
# fromUid = None
if text:
if to_group:
self.send_msg_by_uid(text, self.notifyToGroup)
else :
for au in self.superUser:
if (fromUid != au):
self.send_msg_by_uid(text, au)
pass
if file_name:
file_name = os.path.join(self.temp_pwd,file_name)
if to_group:
if is_pic:
self.send_img_msg_by_uid(file_name, self.notifyToGroup)
else :
self.send_file_msg_by_uid(file_name, self.notifyToGroup)
else :
for au in self.superUser:
if (fromUid != au):
if is_pic:
self.send_img_msg_by_uid(file_name, au)
else:
self.send_file_msg_by_uid(file_name, au)
pass
pass
def handleMoneyInfo(self, msg):
fromUid = msg['user']['id'];
fromName = msg['user']['name'];
say = None
if 'sub_type' in msg['content']:
if msg['content']['sub_type'] == 'payf2f':
say = "\n".join(msg['content']['sub_data'])
if msg['content']['sub_type'] == 'transfer_money':
say = "\n".join(msg['content']['sub_data'])
say = u'%s发起%s\n需要在手机上点击的确认。' %(fromName, say);
if msg['content']['type'] == 12 and u'红包' in msg['content']['data']:
say = u'%s发来红包，需要在手机上点击领取。' %(fromName);
pass
if say :
self.sendToAdmin(fromUid, say, to_group=True)
return True;
return False;
def schedule(self):
print "schedule"
self.handleCommadContent(self.cmdFromGroup, "auto");
return 60 * 60 * 1;
def main():
bot = MyWXBot()
bot.DEBUG = True
bot.conf['qr'] = 'tty'
bot.is_big_contact = False #如果确定通讯录过大，无法获取，可以直接配置，跳过检查。假如不是过大的话，这个方法可能无法获取所有的联系人
bot.run()
if __name__ == '__main__':
main()