def create_chatroom(self, v_user, wx_id_list):
"""
建群 private void btn_CreateChatRoom_Click(object sender, EventArgs e)
:param v_user:
:param wx_id_list:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
payload_json = "{\"Membernames\":\"" + wx_id_list + "\"}"
create_chatroom_req = WechatMsg(
token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=119,
user=v_user,
payloads=payload_json.encode('utf-8')))
create_chatroom_rsp = grpc_client.send(create_chatroom_req)
(buffers, seq) = grpc_utils.get_seq_buffer(create_chatroom_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
check_buffer_16_is_191(buffers)
create_chatroom_rsp.baseMsg.cmd = -119
create_chatroom_rsp.baseMsg.payloads = char_to_str(buffers)
payloads = grpc_client.send(create_chatroom_rsp)
chatroom_detail = json.loads(payloads.baseMsg.payloads)
chatroom_id = chatroom_detail['Roomeid']
print('新建的群id是{}'.format(chatroom_id))
self.send_text_msg(chatroom_id, "欢迎进群", v_user)
def get_contact(self, v_user, wx_id_list):
"""
TODO 根据联系人wxid，获取contact
private void btn_GetContact_Click(object sender, EventArgs e)
:param wx_id_list:
:param v_user:
:return:
"""
payLoadJson = "{\"UserNameList\":\"" + wx_id_list + "\"}"
contacts_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=182,
user=v_user,
payloads=payLoadJson))
get_contacts_rsp = grpc_client.send(contacts_req)
(buffers, seq) = grpc_utils.get_seq_buffer(get_contacts_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
if not check_buffer_16_is_191(buffers):
print("未知包 init")
else:
get_contacts_rsp.baseMsg.cmd = -182
get_contacts_rsp.baseMsg.payloads = char_to_str(buffers)
get_contacts_rsp = grpc_client.send(get_contacts_rsp)
print(get_contacts_rsp.baseMsg.payloads)
print('获取联系人成功')
def add_chatroom_member(self, v_user, chatroom_id, wx_id):
"""
添加微信群成员为好友 btn_AddChatRoomMember_Click
:param v_user:
:param wxid:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
payloadJson = "{\"Roomeid\":\"" + chatroom_id + "\",\"Membernames\":\"" + wx_id + "\"}"
req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=120,
user=v_user,
payloads=payloadJson.encode('utf-8')))
rsp = grpc_client.send(req)
(buffers, seq) = grpc_utils.get_seq_buffer(rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
check_buffer_16_is_191(buffers)
rsp.baseMsg.cmd = -120
rsp.baseMsg.payloads = char_to_str(buffers)
payloads = grpc_client.send(rsp)
def get_chatroom_detail(self, v_user, room_id):
"""
获取群的信息
:param room_id:
:param v_user:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.short_host = bot_param.ShortHost
pay_load_json = "{\"Chatroom\":\"" + room_id + "\"}"
get_room_detail_req = WechatMsg(
token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=551,
user=v_user,
payloads=pay_load_json.encode('utf-8')))
get_room_detail_rsp = grpc_client.send(get_room_detail_req)
# (buffers, seq) = grpc_utils.get_seq_buffer(get_room_detail_rsp)
body = get_room_detail_rsp.baseMsg.payloads
buffers = requests.post(
"http://" + self.short_host + get_room_detail_rsp.baseMsg.cmdUrl,
body)
get_room_detail_rsp.baseMsg.cmd = -551
get_room_detail_rsp.baseMsg.payloads = buffers.content
get_room_detail_rsp = grpc_client.send(get_room_detail_rsp)
buffers = get_room_detail_rsp.baseMsg.payloads
print buffers.encode('utf-8')
def auto_auth(self, v_user, uuid, device_type, new_socket=True):
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
if new_socket:
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
pay_load = "{\"UUid\":\"" + uuid + "\",\"DeviceType\":\"" + device_type + "\"}"
auto_auth_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(
cmd=702,
user=v_user,
payloads=pay_load.encode('utf-8')))
auto_auth_rsp = grpc_client.send(auto_auth_req)
(buffers, seq) = grpc_utils.get_seq_buffer(auto_auth_rsp)
buffers = self.wechat_client.sync_send_and_return(
buffers, close_socket=new_socket)
self.wechat_client.check_buffer_16_is_191(buffers)
auto_auth_rsp.baseMsg.cmd = -702
auto_auth_rsp.baseMsg.payloads = buffers
auto_auth_rsp_2 = grpc_client.send(auto_auth_rsp)
if auto_auth_rsp_2.baseMsg.ret == 0:
user = auto_auth_rsp_2.baseMsg.user
print("二次登陆成功")
v_user_pickle = pickle.dumps(user)
red.set('v_user_' + v_user.userame, v_user_pickle)
return True
elif auto_auth_rsp_2.baseMsg.ret == -100 or auto_auth_rsp_2.baseMsg.ret == -2023:
print("二次登陆失败，重新扫码吧朋友")
ret_reason = ''
try:
payload = auto_auth_rsp_2.baseMsg.payloads
start = "<Content><![CDATA["
end = "]]></Content>"
ret_reason = payload[payload.find(start) +
len(start):payload.find(end)]
except Exception as e:
ret_reason = "未知"
# oss_utils.beary_chat("淘宝客：{0} 已掉线,原因:{1}".format(v_user.nickname, ret_reason))
self.wechat_client.close_when_done()
return False
else:
print("二次登陆未知返回码")
ret_code = auto_auth_rsp_2.baseMsg.ret
# oss_utils.beary_chat("淘宝客：{0} 已掉线,未知返回码:{1}".format(v_user.nickname, ret_code))
self.wechat_client.close_when_done()
return False
def decode_cmd_318(self, v_user, data):
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.short_host = bot_param.ShortHost
self.long_host = bot_param.LongHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
# send notify_req
notify_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=-318,
user=v_user,
payloads=data))
notify_rsp = grpc_client.send(notify_req)
body = notify_rsp.baseMsg.payloads.encode('utf-8')
add_msg_digest = json.loads(body)
print add_msg_digest
payload_dict = {
"ChatroomId": add_msg_digest['ChatRoomId'],
"MsgSeq": add_msg_digest['NewMsgSeq']
}
payload_dict_json = json.dumps(payload_dict)
# get chatroom req
get_chatroom_msg_req = WechatMsg(
token=CONST_PROTOCOL_DICT['machine_code'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=805,
user=v_user,
payloads=payload_dict_json.encode('utf-8')))
get_chatroom_msg_rsp = grpc_client.send(get_chatroom_msg_req)
body = get_chatroom_msg_rsp.baseMsg.payloads
upload_url = get_chatroom_msg_rsp.baseMsg.cmdUrl
buffers = requests.post("http://" + self.short_host + upload_url,
data=body)
print buffers.text
if buffers is None or buffers.text.encode('utf-8')[0] == 191:
# TODO:hextostr
print("unknown package:{0}".format(buffers))
return False
# send grpc and decode again
get_chatroom_msg_rsp.baseMsg.cmd = -805
get_chatroom_msg_rsp.baseMsg.payloads = buffers.text.encode('utf-8')
get_chatroom_msg_rsp = grpc_client.send(get_chatroom_msg_rsp)
buffers = get_chatroom_msg_rsp.baseMsg.payloads
print buffers
return True
def new_init(self, v_user):
"""
登陆初始化
:param v_user:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
new_init_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=1002, user=v_user))
new_init_rsp = grpc_client.send(new_init_req)
(buffers, seq) = grpc_utils.get_seq_buffer(new_init_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers, time_out=3)
if not check_buffer_16_is_191(buffers):
print("未知包 init")
self.wechat_client.close_when_done()
return False
else:
new_init_rsp.baseMsg.cmd = -1002
new_init_rsp.baseMsg.payloads = char_to_str(buffers)
new_init_rsp = grpc_client.send(new_init_rsp)
# 打印出同步消息的结构体
msg_list = json.loads(new_init_rsp.baseMsg.payloads)
if msg_list is not None:
for msg_dict in msg_list:
# save msg
# print(msg_dict)
if msg_dict['MsgType'] == 2:
with app.app_context():
model.save_contact(msg_dict)
else:
print(msg_dict)
v_user = new_init_rsp.baseMsg.user
v_user_pickle = pickle.dumps(v_user)
red.set('v_user_' + v_user.userame, v_user_pickle)
if new_init_rsp.baseMsg.ret == 8888:
print("同步完成")
self.wechat_client.close_when_done()
else:
self.wechat_client.close_when_done()
self.new_init(v_user)
return True
def invite_chatroom(self, v_user, chatroom_id, wx_id):
"""
邀请进群:
向指定用户发送群名片进行邀请
:param v_user:
:param chatroom_id:
:param wx_id:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.short_host = bot_param.ShortHost
payloadJson = "{\"ChatRoom\":\"" + chatroom_id + "\",\"Username\":\"" + wx_id + "\"}"
req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=610,
user=v_user,
payloads=payloadJson.encode('utf-8')))
rsp = grpc_client.send(req)
(buffers, seq) = grpc_utils.get_seq_buffer(rsp)
buffers = requests.post("http://" + self.short_host +
rsp.baseMsg.cmdUrl,
data=buffers)
# 似乎每次登陆后的返回都不一样
if not ord(buffers.text[0]) == 191:
print "invite_chatroom_member 返回码{0}".format(ord(buffers.text[0]))
return True
def delete_chatroom_member(self, v_user, chatroom_id, wx_id):
"""
踢出微信群 btn_DelChatRoomUser_Click
:param v_user:
:param wxid:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.short_host = bot_param.ShortHost
payloadJson = "{\"ChatRoom\":\"" + chatroom_id + "\",\"Username\":\"" + wx_id + "\"}"
req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=179,
user=v_user,
payloads=payloadJson.encode('utf-8')))
rsp = grpc_client.send(req)
(buffers, seq) = grpc_utils.get_seq_buffer(rsp)
buffers = requests.post(
"http://" + self.short_host + rsp.baseMsg.cmdUrl, buffers)
# 检测buffers[0] == 191
if not ord(buffers.text.encode('utf-8')[0]):
print "delete_chatroom_member 未知包"
return False
else:
return True
def heart_beat(self, v_user):
"""
30到60s的定时心跳包
主要用来维持socket的链接状态
:param v_user:
:return:
"""
sync_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=205, user=v_user))
sync_rsp = grpc_client.send(sync_req)
(buffers, seq) = grpc_utils.get_seq_buffer(sync_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
if buffers is None:
return False
ret_code = ord(buffers[16])
if ret_code is not 191:
selector = read_int(buffers, 16)
print "--heartbeat() selector is:{}--".format(selector)
# if selector == -1:
# print "user logout when heartbeat"
#
# if selector > 0:
# v_user = pickle.loads(red.get('v_user_' + v_user.username))
# self.async_check(v_user)
return True
def sns_post(self, v_user, xml):
"""
TODO 发送朋友圈 btn_SnsPost_Click
:param v_user:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
self.short_host = bot_param.ShortHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
pay_load = "{\"Content\":\"" + xml + "\"}"
sns_post_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(
cmd=209,
user=v_user,
payloads=pay_load.encode('utf-8')))
sns_post_rsp = grpc_client.send(sns_post_req)
(buffers, seq) = grpc_utils.get_seq_buffer(sns_post_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
if not self.wechat_client.check_buffer_16_is_191(buffers):
print "sns post 未知包"
return False
else:
return True
def search_contact(self, user_name, v_user):
payLoadJson = "{\"Username\":\"" + user_name + "\"}"
search_contact_req = WechatMsg(
token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=106,
user=v_user,
payloads=payLoadJson.encode('utf-8')))
search_contact_rsp = grpc_client.send(search_contact_req)
(buffers, seq) = grpc_utils.get_seq_buffer(search_contact_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
check_buffer_16_is_191(buffers)
search_contact_rsp.baseMsg.cmd = -106
search_contact_rsp.baseMsg.payloads = char_to_str(buffers)
payloads = grpc_client.send(search_contact_rsp)
print(payloads)
def get_qrcode(self, md_username):
"""
获取qrcode
:return:
"""
# session_key = '5326451F200E0D130CE4AE27262B5897'.decode('hex')
# 构造qrcode请求
self.wechat_client = WechatClient.WechatClient(self.long_host, 80,
True)
self.deviceId = get_md5(md_username)
qrcode_req = WechatMsg(
token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=502,
user=User(sessionKey=int_list_convert_to_byte_list(
CONST_PROTOCOL_DICT['random_encry_key']),
deviceId=self.deviceId)))
qrcode_rsp = grpc_client.send(qrcode_req)
check_grpc_response(qrcode_rsp.baseMsg.ret)
(buffers, seq) = grpc_utils.get_seq_buffer(qrcode_rsp)
data = self.wechat_client.sync_send_and_return(buffers)
qrcode_req.baseMsg.cmd = -502
qrcode_req.baseMsg.payloads = char_to_str(data)
qrcode_rsp = grpc_client.send(qrcode_req)
buffers = qrcode_rsp.baseMsg.payloads
# 保存二维码图片
qr_code = json.loads(buffers)
imgData = base64.b64decode(qr_code['ImgBuf'])
uuid = qr_code['Uuid']
with app.app_context():
model.save_qr_code(qr_code)
try:
oss_path = oss_utils.put_object_to_oss(uuid + ".png", imgData)
print("oss_path is: {}".format(oss_path))
except Exception as e:
print(e)
print('upload oss err by uuid:{}'.format(uuid))
return oss_path, qrcode_rsp, self.deviceId
def send_voice_msg(self, v_user, to_user_name, url):
"""
发送语音 btn_SendVoice_Click
:param v_user:
:param url:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
voice_path = 'img/test.mp3'
with open(voice_path, 'rb') as voice_file:
data = voice_file.read()
payload = {
'ToUserName': to_user_name,
'Offset': 0,
'Length': len(data),
'EndFlag': 1,
'Data': data,
'VoiceFormat': 0
}
payload_json = json.dumps(payload)
send_voice_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(
cmd=127,
user=v_user,
payloads=payload_json.encode('utf-8')))
send_voice_rsp = grpc_client.send(send_voice_req)
(buffers, seq) = grpc_utils.get_seq_buffer(send_voice_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
self.wechat_client.close_when_done()
return check_buffer_16_is_191(buffers)
def modify_chatroom_name(self, v_user, chatroom_id, room_name):
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.short_host = bot_param.ShortHost
payload_json = "{\"Cmdid\":27,\"ChatRoom\":\"" + chatroom_id + "\",\"Roomname\":\"" + room_name + "\"}"
modify_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(
cmd=681,
user=v_user,
payloads=payload_json.encode('utf-8')))
modify_rsp = grpc_client.send(modify_req)
(buffers, seq) = grpc_utils.get_seq_buffer(modify_rsp)
buffers = requests.post("http://" + self.short_host +
modify_rsp.baseMsg.cmdUrl,
data=buffers)
self.wechat_client.close_when_done()
return check_buffer_16_is_191(buffers)
def send_text_msg(self, user_name, content, v_user):
"""
参考btn_SendMsg_Click
:param user_name:
:param content:
:param at_user_list:
:param v_user:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
payLoadJson = "{\"ToUserName\":\"" + user_name + "\",\"Content\":\"" + content + "\",\"Type\":0,\"MsgSource\":\"\"}"
send_text_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(
cmd=522,
user=v_user,
payloads=payLoadJson.encode('utf-8')))
send_text_rsp = grpc_client.send(send_text_req)
(buffers, seq) = grpc_utils.get_seq_buffer(send_text_rsp)
import binascii
print(binascii.b2a_hex(buffers))
buffers = self.wechat_client.sync_send_and_return(buffers)
if not check_buffer_16_is_191(buffers):
if read_int(buffers, 18) == -13:
print("Session Time out 离线或用户取消登陆 需执行二次登录")
return False
self.wechat_client.close_when_done()
return True
def confirm_qrcode_login(self, qr_code, keep_heart_beat):
# 重置longHost
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=qr_code['Username']).first()
if bot_param and bot_param.LongHost is not None and bot_param.LongHost != "":
self.long_host = bot_param.LongHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
# 微信确认登陆模块
UUid = u"667D18B1-BCE3-4AA2-8ED1-1FDC19446567"
DeviceType = u"<k21>TP_lINKS_5G</k21><k22>中国移动</k22><k24>c1:cd:2d:1c:5b:11</k24>"
payLoadJson = "{\"Username\":\"" + qr_code[
'Username'] + "\",\"PassWord\":\"" + qr_code[
'Password'] + "\",\"UUid\":\"" + UUid + "\",\"DeviceType\":\"" + DeviceType + "\"}"
qrcode_login_req = WechatMsg(
token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=1111,
user=User(sessionKey=int_list_convert_to_byte_list(
CONST_PROTOCOL_DICT['random_encry_key']),
deviceId=self.deviceId),
payloads=payLoadJson.encode('utf-8')))
qrcode_login_rsp = grpc_client.send(qrcode_login_req)
(buffers, seq) = grpc_utils.get_seq_buffer(qrcode_login_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
check_buffer_16_is_191(buffers)
# 解微信包
qrcode_login_rsp.baseMsg.cmd = -1001
qrcode_login_rsp.baseMsg.payloads = char_to_str(buffers)
qrcode_login_rsp = grpc_client.send(qrcode_login_rsp)
with app.app_context():
bot_param_db = model.BotParam.query.filter_by(
Username=qr_code['Username']).first()
if bot_param_db is None:
model.save_bot_param(qr_code['Username'], self.deviceId,
qrcode_login_rsp.baseMsg.longHost,
qrcode_login_rsp.baseMsg.shortHost)
else:
model.update_bot_param(bot_param_db, qr_code['Username'],
self.deviceId,
qrcode_login_rsp.baseMsg.longHost,
qrcode_login_rsp.baseMsg.shortHost)
if qrcode_login_rsp.baseMsg.ret == -301:
# 返回-301代表重定向
self.wechat_client.close_when_done()
# 在user表写上gg，麻烦重新登录吧
return False
elif qrcode_login_rsp.baseMsg.ret == 0:
# 返回0代表登陆成功
print('login successful')
# 将User赋值
v_user = qrcode_login_rsp.baseMsg.user
# 存下v_user
with app.app_context():
model.save_user(v_user)
with app.app_context():
try:
user_db = model.User.query.filter_by(
userame=v_user.userame).first()
user_db.login = 1
db.session.commit()
except Exception as e:
print(e)
# if keep_heart_beat:
# 登陆成功，维持心跳
# asyn_rec_thread = threading.Thread(target=self, WXBot.heart_beat(v_user))
# asyn_rec_thread.start()
red.set('v_user_' + str(v_user.userame), pickle.dumps(v_user))
self.wechat_client.close_when_done()
return True
else:
print("qrcode_login_rsp.baseMsg.ret is {}".format(
qrcode_login_rsp.baseMsg.ret))
print("原因是{}".format(qrcode_login_rsp.baseMsg.payloads))
return False
def async_check(self, v_user, new_socket=True):
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
if new_socket:
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
sync_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(cmd=138, user=v_user))
sync_rsp = grpc_client.send(sync_req)
(buffers, seq) = grpc_utils.get_seq_buffer(sync_rsp)
buffers = self.wechat_client.sync_send_and_return(
buffers, close_socket=new_socket)
if not check_buffer_16_is_191(buffers):
try:
# TODO:uuid 和 devicetype 存起来？
UUid = u"667D18B1-BCE3-4AA2-8ED1-1FDC19446567"
DeviceType = u"<k21>TP_lINKS_5G</k21><k22>中国移动</k22><k24>c1:cd:2d:1c:5b:11</k24>"
if self.auto_auth(v_user,
UUid,
DeviceType,
new_socket=new_socket):
v_user = pickle.loads(red.get('v_user_' + v_user.userame))
self.async_check(v_user, new_socket=new_socket)
return True
self.logout_bot(v_user)
print(read_int(buffers, 18))
if read_int(buffers, 18) == -13:
print("Session Time out 离线或用户取消登陆 需执行二次登录")
except Exception as e:
print(e)
return False
else:
sync_rsp.baseMsg.cmd = -138
sync_rsp.baseMsg.payloads = char_to_str(buffers)
sync_rsp = grpc_client.send(sync_rsp)
# 刷新用户信息
v_user = sync_rsp.baseMsg.user
v_user_pickle = pickle.dumps(v_user)
red.set('v_user_' + v_user.userame, v_user_pickle)
msg_list = json.loads(sync_rsp.baseMsg.payloads)
if msg_list is not None:
for msg_dict in msg_list:
try:
if msg_dict['MsgType'] == 2:
with app.app_context():
model.save_contact(msg_dict)
elif msg_dict['Status'] is not None:
try:
action_rule.filter_keyword_rule(
v_user.userame, msg_dict)
except Exception as e:
print(e)
with app.app_context():
model.save_message(msg_dict)
else:
print(msg_dict)
except Exception as e:
print(e)
print(msg_dict)
self.async_check(v_user, new_socket=new_socket)
else:
print "sync 完成"
def send_img_msg(self, user_name, v_user, url):
"""
btn_SendMsgimg_Click
:param user_name:
:param v_user:
:return:
"""
with app.app_context():
bot_param = model.BotParam.query.filter_by(
Username=v_user.userame).first()
if bot_param:
self.long_host = bot_param.LongHost
self.wechat_client = WechatClient.WechatClient(
self.long_host, 80, True)
data = urllib2.urlopen(url).read()
# img_path = 'img/no.png'
# with codecs.open(img_path, 'rb') as img_file:
# data = img_file.read()
# with open(img_path, 'rb') as img_file:
# data = img_file.read()
# 起始位置
start_pos = 0
# 数据分块长度
data_len = 16535
# 总长度
data_total_length = len(data)
print('preparing sent img length length {}'.format(data_total_length))
# 客户图像id
client_img_id = v_user.userame + "_" + str(get_time_stamp())
while start_pos != data_total_length:
# 每次最多只传65535
count = 0
if data_total_length - start_pos > data_len:
count = data_len
else:
count = data_total_length - start_pos
upload_data = base64.b64encode(data[start_pos:start_pos + count])
payLoadJson = {
'ClientImgId': client_img_id.encode('utf-8'),
'ToUserName': user_name.encode('utf-8'),
'StartPos': start_pos,
'TotalLen': data_total_length,
'DataLen': len(data[start_pos:start_pos + count]),
'Data': upload_data
}
pay_load_json = json.dumps(payLoadJson)
start_pos = start_pos + count
print("Send Img Block {}".format(count))
print("start_pos is {}".format(start_pos))
print("data_total_length is {}".format(data_total_length))
img_msg_req = WechatMsg(token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(
cmd=110,
user=v_user,
payloads=pay_load_json.encode('utf-8'),
))
img_msg_rsp = grpc_client.send(img_msg_req)
(buffers, seq) = grpc_utils.get_seq_buffer(img_msg_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers,
time_out=3)
check_buffer_16_is_191(buffers)
self.wechat_client.close_when_done()
return True
def check_qrcode_login(self, qrcode_rsp, device_id):
"""
检测扫描是否登陆
:param qr_code:
:return:
"""
buffers = qrcode_rsp.baseMsg.payloads
qr_code = json.loads(buffers)
uuid = qr_code['Uuid']
notify_key_str = base64.b64decode(qr_code['NotifyKey'].encode('utf-8'))
long_head = qrcode_rsp.baseMsg.longHead
start_time = datetime.now()
self.deviceId = device_id
while qr_code['Status'] is not 2:
# 构造扫描确认请求
check_qrcode_grpc_req = WechatMsg(
token=CONST_PROTOCOL_DICT['machine_code'],
version=CONST_PROTOCOL_DICT['version'],
timeStamp=get_time_stamp(),
iP=get_public_ip(),
baseMsg=BaseMsg(
cmd=503,
longHead=long_head,
payloads=str(uuid),
user=User(sessionKey=int_list_convert_to_byte_list(
CONST_PROTOCOL_DICT['random_encry_key']),
deviceId=self.deviceId,
maxSyncKey=notify_key_str)))
checkqrcode_grpc_rsp = grpc_client.send(check_qrcode_grpc_req)
(buffers, seq) = grpc_utils.get_seq_buffer(checkqrcode_grpc_rsp)
buffers = self.wechat_client.sync_send_and_return(buffers)
check_buffer_16_is_191(buffers)
# 将微信包传给grpc服务器，进行一次解包操作
checkqrcode_grpc_rsp.baseMsg.cmd = -503
checkqrcode_grpc_rsp.baseMsg.payloads = char_to_str(buffers)
checkqrcode_grpc_rsp_2 = grpc_client.send(checkqrcode_grpc_rsp)
payloads = checkqrcode_grpc_rsp_2.baseMsg.payloads
if 'unpack err' not in payloads:
qr_code = json.loads(payloads)
if qr_code['Status'] is 2:
# save qr_code
with app.app_context():
qr_code_db = model.Qrcode.query.filter_by(
Uuid=uuid).order_by(model.Qrcode.id.desc()).first()
model.update_qr_code(qr_code, qr_code_db)
# 成功登陆
return qr_code
elif qr_code['Status'] is 0:
# 未扫描等待扫描
pass
elif qr_code['Status'] is 1:
# 已扫描未确认
pass
elif qr_code['Status'] is 4:
# 已取消扫描
pass
# 等待5s再检测
time.sleep(5)
# 如果3分钟都没有正常返回status 2 返回False
if (datetime.now() - start_time).seconds >= 60 * 3:
return False