class WXHandler():
bot = None
def login(self):
self.bot = Bot(cache_path=True)
# 搜索名称含有 "游否" 的男性深圳好友
def sendFriend(self,msg,friendName):
my_friend = self.bot.friends().search(unicode( '文件传输助手' ))
# # 发送文本给好友
# my_friend[0].send('Hello WeChat!')
# print my_friend
# for i in range(1,101):
# my_friend[0].send('宝儿'+str(i))
# time.sleep(0.5)
my_groups = bot.groups().search(unicode( 'wxpy' ))
print my_groups
for i in range(1,101):
my_groups[0].send(i)
time.sleep(0.5)
class ServerController(object):
def __init__(self, init=False):
#初始化机器人
self.bot = Bot(init)
self.bot.enable_puid("wxpy_puid.pkl")
self.friend_manager = friends.FriendManager()
self.common = function.CommonFn()
#存储所有朋友
self.friends = []
#获取朋友列表
self._get_friend()
def _get_friend(self):
friends = self.bot.friends()
friendpUid = set()
friendInfo = []
for friend in friends:
#判断重复的puid
if friend.puid in friendpUid:
continue
friendpUid.add(friend.puid)
tmpFriends = self.bot.friends().search(puid=friend.puid)
for tmpFriend in tmpFriends:
#打印所有对象集合
#self.common.prn_obj(tmpFriend)
friendInfo.append({
"user_name": tmpFriend.raw["UserName"],
"nick_name": tmpFriend.raw["NickName"],
"puid": tmpFriend.puid,
"remark_name": tmpFriend.raw["RemarkName"]
})
self.friends.append(tmpFriend)
self.friend_manager.saveFriends(friendInfo)
def start(self):
print("【" + self.bot.self.raw["NickName"] + "】登录成功")
listen(self.bot, self.friends)
self.bot.join()
def replay(self):
@self.bot.register()
def print_others(msg):
print(msg)
@self.bot.register(self.friends)
def reply_my_friend(msg):
if msg.type != "Text":
return "暂时支持文本格式的"
#连接斗鱼数据查询
douyuSer = douyuServer.DouyuServer(msg)
#根据roomID号爬取斗鱼直播情况
return douyuSer.liveDataByRoomId()
@self.bot.register(msg_types="Friends")
def auto_accept_friends(msg):
if msg.text == "top":
new_friend = msg.card.accept()
#更新朋友列表
self._get_friend()
new_friend.send(
'您好，已经接受好友请求了\n访问 https://github.com/Topthinking/wxPython 查看更多'
)
class ServerController(object):
def __init__(self, init=False):
# 初始化机器人
self.bot = Bot(init)
self.bot.enable_puid("wxpy_puid.pkl")
self.common = function.CommonFn()
self.userdbser = userDb.User()
self.textmsg = TextMsgHandle.TextMsg()
# 存储所有朋友
self.friends = []
# 获取朋友列表
self._get_friend()
def _get_friend(self):
friends = self.bot.friends()
# 遍历所有好友，进行存储和更新好友信息
friendpuid = set()
for friend in friends:
# 判断重复的puid
if friend.puid in friendpuid:
continue
friendpuid.add(friend.puid)
tmp_friends = self.bot.friends().search(puid=friend.puid)
for tmp_friend in tmp_friends:
# 机器人自己不加入
if tmp_friend.raw["UserName"] == self.bot.self.raw["UserName"]:
continue
# 刷新当前数据库好友信息
self._actionUserInfo(tmp_friend)
def start(self):
print("【" + self.bot.self.raw["NickName"] + "】登录成功")
listen(self.bot, self.friends)
self.bot.join()
def replay(self):
@self.bot.register()
def print_others(msg):
print(msg)
@self.bot.register(self.friends)
def reply_my_friend(msg):
# 添加好友的提示
if msg.type == "Note":
return ''
if msg.type != "Text":
return "暂时支持文本格式的"
# 发送的文本信息
if msg.type == "Text":
self.textmsg.start(msg)
@self.bot.register(msg_types="Friends")
def auto_accept_friends(msg):
if "top" in msg.text.lower():
new_friend = msg.card.accept()
# 对添加的用户进行保存
self._actionUserInfo(new_friend)
new_friend.send(
'您好，已经接受好友请求了\n访问 https://github.com/Topthinking/wxPython 查看更多'
)
def _actionUserInfo(self, friend):
"""
添加用户
nick_name,user_name,puid,add_time
更新用户
nick_name,user_name,puid,update_time,id
"""
curTime = int(time.mktime(datetime.datetime.now().timetuple()))
# 判定该添加的朋友之前是否存在备注
if friend.raw["RemarkName"] == '':
# 添加新的朋友到数据库
param = (friend.raw["NickName"], friend.raw["UserName"],
friend.puid, curTime)
remarkName = self.userdbser.insertUserGetInertId(param)
self.bot.core.set_alias(userName=friend.raw["UserName"],
alias=remarkName)
else:
# 更新
# 1.先查询是否存在该用户 存在就更新 否则就添加
param = (friend.raw["RemarkName"], )
info = self.userdbser.isExistUser(param)
if info is None:
# 添加新的朋友到数据库
param = (friend.raw["NickName"], friend.raw["UserName"],
friend.puid, curTime)
remarkName = self.userdbser.insertUserGetInertId(param)
self.bot.core.set_alias(userName=friend.raw["UserName"],
alias=remarkName)
else:
param = (friend.raw["NickName"], friend.raw["UserName"],
friend.puid, curTime, friend.raw["RemarkName"])
self.userdbser.updateUserInfo(param)
self.friends.append(friend)