class AutoSophia:
def __init__(self, name="0000000",flowName="", count = 0, makeRooms=[]):
self.robot = Robot()
self.http = Http()
self.name = name
self.flowName = flowName
self.count = count #编号
self.listMsgQue = [] #消息发送队列
self.timeDetaMsgSend = 1.5 #最小发送消息间隔s
self.makeRooms = makeRooms
self.userIndex = {} # 用户name -> 用户id
self.userIndexRe = {} #id -> name
self.roomIndex = {} #房间号 及其<用户>信息
self.roomMsg = {} #消息 记录
self.roomId = "" #当前房号
self.ifWelcom = False #是否迎客
self.ifTime = False #是否存货确认
self.admins = {}
self.tripcodeIndex = {} #上次房间记录 的 用户名 绑定的 tc code
self.linkStart = 0 #链接状态
self.init()
self.tail = " の... "
def init(self):
############### 心情模块
self.statusMin = 5
self.statusMax = 95
self.statusDefault = 80
self.status = 10 #说话欲望值 0-100
self.statusOnDeta = 15 #开心
self.statusOffDeta = 15 #难过
self.statusDownDeta = 40 #闭嘴
self.lastOtherSay = tool.getNowTime() #上次其他人说话时间
self.getMsgDetaTime = 1 #抓取消息间隔
self.lastMsgTime = int(time.time() * 10000 ) * 1.0 / 10000 #上一次更新房间聊天记录时间
self.lastEchoTime = tool.getNowTime() #上次说话时间
self.lastEchoTimeQuene = tool.getNowTime()
self.maxDetaOtherSay = 1000 * 60 * 15 #最大没人说话时间 换房
self.maxDetaTime = 1000 * 60 * 5 # 最大沉默时间
self.lastMusicTime = tool.getNowTime() #上次放歌时间
self.maxMusicTime = 1000 * 60 * 4 #音乐间隔 暂不解析音乐文件时长控制
self.maxMusicTimeDefault = 1000 * 60 * 4 #默认时间
self.musicNow = {}
self.nextNames = [] #下一首序列
self.musicPlayType = -1
self.ifOnMusic = False
self.notWait = True
self.adminRes = ""
self.adminDeta = 2
self.adminDetaDefault = 2
self.toIdNow = "" #当前处理消息
self.toNameNow = "" #当前处理消息
self.fromIdNow = "" #当前处理消息
self.fromNameNow = "" #当前处理消息
def out(self, obj):
print(time.strftime("%Y%m%d %H:%M:%S", time.localtime()) + "." + self.name + "." + str(obj))
return
def login(self):
# tool.line()
self.out("访问主页 获取 token session")
responce = self.http.doGet('http://drrr.com/')
re = responce.read()
soup =BeautifulSoup.BeautifulSoup(re)
# self.out soup.prettify()
nameList = soup.findAll('input',{'name':{'token'}})
if(len(nameList) > 0):
token = nameList[0]['data-value']
token = tool.encode(token)
self.out("抓取成功: ")
self.out("token\t " + token)
self.out("cookie\t " + tool.toString(self.http.getCookie()))
# tool.line()
self.out("模拟登录")
responce=self.http.doPost('http://drrr.com/', {
"name":self.name,
"login":"******",
"token":token,
"direct-join":"",
"language":"zh-CN",
"icon":"tanaka",
})
else:
self.out("error！ 没能抓取到token")
def showHelp(self):
self.send("@" + self.name + " \n 0.help \n 0.play/next name \n 0.play music/stop music \n 1.del/rm xxx \n 2.host \n <0.admin> ")
self.help()
def nobody(self):
self.showHelp()
def showAdmin(self):
res = "#Admins:"
items = sorted(self.admins.items(), cmp=lambda x,y: cmp(y[1], x[1]) )
i = 0
for key,value in items: # id value
res = res + "\na" + str(i) + " " + str(self.userIndexRe.get(key)) + " " + str(value)
i = i+1
self.send("" + res)
def addAdmin(self, fromId, value=0):
if(self.admins.get(fromId, 0) > 2**31):
self.send(self.name + "被玩坏了orz" + self.tail)
self.admins = {}
return
if(value>0):
self.adminDeta = value
self.admins[fromId] = int(self.admins.get(fromId,0)) + int(self.adminDeta) #认证加权
if(self.admins[fromId] <= 0):
self.admins.pop(fromId)
self.adminRes = ""
self.adminDeta = self.adminDetaDefault
# self.showAdmin()
def getAdmin(self, name=""):
level = self.admins.get(self.userIndex.get(name, ""), 0)
# self.adminRes = 0
# self.adminDeta = 2
# self.adminDetaDefault = 2
# <2 <4 <8 <16 <32
#i:1 2 3 4 5
# +2 +4 +8 +16 +32
i = 1
weight = 1
while(True):
weight = weight * self.adminDetaDefault # *= 2
if(level < weight):
break
i = i + 1
# level=0-2 -> i=1 weight=2
res = ""
ranres = ""
for j in range(i):
ran = tool.getRandom(1000, 9999) # 2843 = 287 = 35 = 8
ans = ran
while(ans >= 10):
ans = ans % 10 + ans / 10
res = res + str(ans)
ranres = ranres + str(ran) + " "
self.adminRes = res
self.adminDeta = weight
# self.showAdmin()
self.send("#Admin认证 lv." + str(i) + "\nF(" + str(ranres) + ") = ? " )
#管理员权限认证 10次
def ifAdminName(self, name="", lev=0, ifRedu=True):
return self.ifAdmin(self.userIndex.get(name, ""), lev, ifRedu)
def ifAdmin(self, id="", lev=0, ifRedu=True):
if(self.admins.get(id, 0) > 2**lev-1):
if(ifRedu):
self.admins[id] = self.admins[id] * 2 / 3 # 每次认证消减1/3生命值
if(self.admins[id] <= 0):
self.admins.pop(id)
return True
return False
def host(self, name=""):
# new_host:5a1da324d5e68e6712725a50046f4b75
name = str(name)
if(name == ""):
return
self.getRooms() # 刷新最新房间信息
if(not self.ifHost):
self.send("不是房主,没有权限")
return
userId = self.userIndex.get(name, "")
if(self.ifAdmin(userId) ):
if(self.getRoomUser(name).get("name","") != ""):
self.send("转移房主权限给[" + name + "] ")
responce=self.http.doPost('http://drrr.com/room/?ajax=1', {
"new_host":userId
})
else:
self.send("用户[" + name + "]不在当前房间")
else:
self.send("用户[" + name + "]未认证admin权限")
self.getAdmin(name)
def rm(self, name="", pwd=""):
name = str(name)
if(name == ""):
return
self.getRooms() # 刷新最新房间信息
if(not self.ifHost):
self.send("不是房主,没有权限")
return
userId = self.userIndex.get(name,"")
if(self.ifAdmin(userId) ):
self.send("禁止踢出admin")
else:
if(self.getRoomUser(name).get("name","") != ""):
self.send("踢出[" + name + "]" ) # id:" + str(self.userIndex.get(name, "")) + " tripcode:" + str(item.get("tripcode", "")) + " device:" + str(item.get("device", ""))
responce=self.http.doPost('http://drrr.com/room/?ajax=1', {
"kick":userId
})
else:
self.send("用户[" + name + "]不在当前房间")
return
def help(self):
self.out(dir(self))
def showUser(self, user, show=True):
userInfo ="U " + user.get("device", "")[0:4] + " " + user.get("icon", "")[0:4] + " " + user.get("name", "")[0:6] + " " + user.get("id", "")[0:4] + " "+ user.get("tripcode", "")[0:4]
if(show):
self.out(userInfo)
return userInfo
def showRoom(self, roomId="", show=True, i=0):
if(roomId == ""):
roomId = self.roomId
room = self.roomIndex.get(roomId, "")
if(room == ""):
self.getRooms()
room = self.roomIndex.get(roomId, "")
info = ""
if(room != ""):
info = ("##" + tool.fill(str(i), '#', 10) + "\n--G " + tool.fill(room["id"], ' ', 10) + " " + tool.fill(str(room["total"]) + "/" + str(room["limit"]), ' ', 7) + " " + room["name"][0:10]) + "\n"
info = info + "music: " + str(room.get("music", False)) + " language:" + room.get("language","")+" game:"+ str(room.get("gameRoom", "")) +" adult:"+ str(room.get("adultRoom", ""))+"\n" # " 静态房间: " + str(room.get("staticRoom", "")) + ""
# info = info + " hiddenRoom: " + str(room.get("staticRoom", "")) + " 游戏房间: " + str(room.get("gameRoom", "")) + " 成人房间: " + str(room.get("adultRoom", "")) + "\n"
info = info + "Host:" + room.get("host", {}).get("name", "") + "\n"
info = info + "Users: " + "\n"
for item in room.get("users", []):
# print(item)
info = info + "--" + self.showUser(item, False) + "\n"
if(show):
self.out(info)
return info
def showRoomList(self, roomId=""):
if(roomId == ""):
rooms = self.roomIndex.values()
rooms = sorted(rooms, cmp=lambda y,x: cmp(x.get("total",0), y.get("total", 0)))
i = 0
res = "#Rooms list:"
for room in rooms:
i = i + 1 #+ "" + room["id"]
res = res + "\n" + tool.fill(str(i)," ",2) + "." + tool.fill(str(room["total"]) + "/" + str(room["limit"]), ' ', 4) + " " + room["name"][0:6] + " " + room["id"]
else:
room = self.roomIndex.get(roomId, "")
if(room != ""):
res = "" #Rooms info:
res = res + "-G." + tool.fill(str(room["total"]) + "/" + str(room["limit"]), ' ', 4) + " " + room["name"][0:6] + " " + room["id"]
res = res + "\n-Music: " + str(room.get("Music", False)) + " " + room.get("language","") # +" game:"+ str(room.get("gameRoom", "")) +" adult:"+ str(room.get("adultRoom", "")) # " 静态房间: " + str(room.get("staticRoom", "")) + ""
# info = info + " hiddenRoom: " + str(room.get("staticRoom", "")) + " 游戏房间: " + str(room.get("gameRoom", "")) + " 成人房间: " + str(room.get("adultRoom", "")) + "\n"
res = res + "\n-Host:" + room.get("host", {}).get("name", "")
# res = res + "\n Users: " + "\n"
i = 0
for item in room.get("users", []):
res = res + "\n -" + self.showUser(item, False) # item.get("name", "")[0:6]
i = i + 1
self.send(res)
def showAllRoom(self):
if(self.roomIndex is None or self.roomIndex == "" or self.roomIndex == {}):
self.getRooms()
tool.line()
self.out("展示所有房间信息")
i = 0
for key in self.roomIndex:
# room = self.roomIndex[key]
self.showRoom(key, True, i)
i = i+1
tool.line()
# 获取当前房间人员列表
def getRoomUsers(self, roomId=""):
if(roomId == ""):
roomId = self.roomId
room = self.roomIndex.get(roomId, {})
users = room.get("users", [])
return users
# 按照名字获取当前房间用户
def getRoomUser(self, name=""):
users = self.getRoomUsers()
for user in users:
if(user.get("name", "") == name):
return user
return {}
# 获取用户所在房间
def getUserRoom(self, userName="******", userId="8f1b61e25098b0427f01d724716b70cb"):
i=0
res = []
for key in self.roomIndex:
room = self.roomIndex[key]
users = room.get("users", [])
for user in users:
if(user.get("name", "") == userName):
# self.showRoom(key, True, i)
res.append(room)
if(user.get("id", "") == userId):
self.showRoom(key, True, i)
i = i+1
if(len(res) <= 0):
self.out("用户不在线"+userName)
if(len(res) >= 2):
self.out("该用户多次出现？？？？？")
for item in res:
self.out(item)
tool.line()
return res
def goRoomName(self, roomName):
if(self.roomIndex is None or self.roomIndex == "" or self.roomIndex == {}):
self.getRooms()
tool.line()
self.out("查找房间名字加入" + roomName)
i = 0
for key in self.roomIndex:
room = self.roomIndex[key]
name = room.get("name", "")
if(re.search(roomName, name) != None):
self.goRoom(key)
break
i = i+1
tool.line()
def goRoom(self, roomId):
if(self.roomId == roomId):
self.out("已经处于当前房间")
return
if(self.roomId != ""):
self.outRoom()
time.sleep(10)
# tool.line()
self.out("加入房间:" + roomId)
# self.showRoom(roomId)
responce=self.http.doGet("http://drrr.com/room/?id=" + roomId)
if(responce != "error"):
self.roomId = roomId
self.init()
# self.send("大家好 我是暖手宝" + self.name + " 可以@ [点歌/turn/prev](*^_^*) @不一定会回 不@也不一定不会回(∩_∩) ")
self.initRoom()
return
def initRoom(self, roomId=""):
if(roomId == ""):
roomId = self.roomId
room = self.roomIndex.get(roomId, "")
if(room != ""):
for item in room.get("users", []):
# print(item)
id = item.get("id", "")
if(id != ""):
self.userIndex[item.get("name", "")] = item.get("id", "")
self.userIndexRe[item.get("id")] = item.get("name", "")
self.tripcodeIndex[item.get("name", "")] = item.get("tripcode", "")
# self.out(str(self.tripcodeIndex))
def outRoom(self):
self.out("离开房间:" + self.roomId)
# self.send("" + self.name + "好无聊啊 "+self.name +"要出去溜达一会儿" + self.tail)
# self.send(""+self.name+"一定会回来的" + self.tail)
# self.send("出去一下，马上回来" + self.tail)
self.showRoom(self.roomId)
time.sleep(self.timeDetaMsgSend * len(self.listMsgQue) + 1) #等待一会儿消息发送
responce=self.http.doPost("http://drrr.com/room/?ajax=1", {
"leave":"leave",
})
if(responce == "error"):
return False
self.roomId = ""
time.sleep(1)
return True
def createRoom(self, room=""):
# tool.line()
self.out("创建房间")
if(room == ""):
room = {}
room["name"] = self.name + "'s room"
responce=self.http.doPost("http://drrr.com/create_room/?", {
"name":room.get("name",""),
"description":room.get("description",""),
"limit":room.get("limit",10),
"music":room.get("music",True),
"submit":room.get("music","创建房间"),
})
if(responce == "error"):
return False
self.roomId = ""
time.sleep(1)
def getRooms(self, detail=False):
tool.line()
self.out("获取房间列表")
responce=self.http.doGet("http://drrr.com/lounge?api=json")
jsonObj = tool.makeObj(json.loads(responce.read()))
rooms = jsonObj["rooms"]
makeRooms = []
if(len(rooms) > 0):
self.roomIndex.clear()
i = 0
count = 0
userCount = 0
for room in rooms:
id = room.get("id","")
if(room.get("language","") == "zh-CN"):
# root.showRoom(id, show=True, i=i)
makeRooms.append(room)
self.roomIndex[room["id"]] = room
count = count + 1
userCount = userCount + int(room.get("total", 0))
self.out("#" + tool.fill(str(i),' ',4) + "" + room["id"] + " " + str(room["total"]) + "/" + str(room["limit"]) + "\t " + room["name"])
# for item in room.get("users", []):
# if(item.get("id", "") != ""):
# self.userIndex[item.get("name", "")] = item.get("id", "")
# self.userIndexRe[item.get("id")] = item.get("name", "")
i = i + 1
self.out("共计房间" + tool.fill(str(count), ' ', 5) + " 用户" + tool.fill(str(userCount), ' ', 5) )
self.out("解析完毕")
roomId = self.roomId
room = self.roomIndex.get(roomId, {})
if(room.get("host", {}).get("name", "") == self.name):
self.ifHost = True
else:
self.ifHost = False
self.initRoom()
return makeRooms
# 太久没人发言 时 退出 并 进入一个新的 活跃的房间
def goARoom(self):
# self.out("#" + str(i) + "\t" + room["id"] + " " + str(room["total"]) + "/" + str(room["limit"]) + "\t " + room["name"])
lastRoomId = self.roomId
nowRoom = self.roomIndex.get(self.roomId, {})
if(nowRoom.get("total", 0) > 1): #当前房间人数 还有其他人
# self.send(self.name + "也无法忍受这样的寂静了 债见了各位" + self.tail)
pass
if(self.roomId != ""):
self.outRoom()
self.getRooms()
self.out("选择最活跃房间")
i = 0
maxNum = 0
maxKey = ""
for key in self.roomIndex:
exist = True
room = self.roomIndex[key]
total = room.get("total", 0)
limit = room.get("limit", 0)
music = room.get("music", False)
for item in room.get("users", []):
if(item.get("name", "") == self.name):
tool.line()
self.out("异常! 该房间存在同名用户 无法加入 ")
self.showRoom(room.get("id", ""))
exist = False
break
if(item.get("name", "") == self.flowName): #跟随
self.out("跟随触发 增大权重选中")
maxNum = 99 + int(total)
maxKey = key
i = i+1
if(maxKey != ""):
self.out("选中房间:")
self.showRoom(maxKey)
tool.line()
self.goRoom(maxKey)
else:
tool.line()
self.out("异常！！！！！！！！！ 居然无可用房间？")
time.sleep(2)
return
# 定时消息发送队列
def doHello(self):
while(True):
if(self.roomId != ""):
self.out("开启消息发送队列 deta=" + str(self.timeDetaMsgSend) + "ms")
# self.listMsgQue = []
while(self.roomId != ""):
try:
detaTime = tool.getNowTime() - self.lastEchoTime
if(detaTime > self.timeDetaMsgSend): # 发送时差大于最小发送间隔
if(len(self.listMsgQue) > 0):
msg = self.listMsgQue.pop(0)
if(type(msg) == tuple):
self.doSendId(msg[0], msg[1])
else:
self.doSend(msg)
time.sleep(self.timeDetaMsgSend)
except Exception as e:
self.out("消息发送异常 消息队列:")
self.out(self.listMsgQue)
self.out(traceback.format_exc())
# self.out("当前房间roomId:" + self.roomId + " 未加入房间 暂时停止sayHello ")
time.sleep(3)
# 定时操作
def sayHello(self):
while(True):
if(self.roomId != ""):
self.out("开启定时发言，最大发言间隔" + str(self.maxDetaTime / 1000) + "s")
sleepTime = 3
dt = 0
theI = 0
self.lastEchoTimeQuene = tool.getNowTime()
while(self.roomId != ""):
try:
# message = "Now Time is "+ time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
detaTime = tool.getNowTime() - self.lastEchoTime # ms
if(detaTime > self.maxDetaTime):
# message = "存活确认." + str(theI) + "." +
ttt = time.strftime("%H時%M分%S秒")
message = "現在時間是" + ttt + "，每隔5分鐘免費派送一座冰山：△，一間冰窟酒吧：♜，一棟冰窟豪宅：▌，一缸冰：，一缸涼水：，附加贈送一個月的南極大陸度假旅遊服務(可免費觀看地下冰花：✹✺✻✼✽✾✿❀❁❂❃❄❅❆❇❈❉❊❋。)。"
self.doSendId(message)
self.out(str(theI) + "\t" + message)
theI = theI + 1
self.lastEchoTime = tool.getNowTime()
detaTime = tool.getNowTime() - self.lastOtherSay # ms
if(detaTime > self.maxDetaOtherSay): #不不停留True
self.goARoom() #10分钟没处理过消息 互动 则换房间
if(detaTime > self.maxDetaOtherSay * 2): #若一小时没信息 则 是否掉线?
self.shutdown() #等待系统重启
if(dt % 600 == 0):
self.getRooms() #定时5分钟获取房间最新信息
if(dt % 120 == 0):
roomsAdmin = self.getUserRoom(self.flowName);
if(len(roomsAdmin) > 0):
self.out("跟随 触发")
self.goRoom(roomsAdmin[tool.getRandom(0, len(roomsAdmin))].get("id", ""))
time.sleep(sleepTime)
dt = dt + sleepTime
dt = dt % 3600
except Exception as e:
self.out(traceback.format_exc())
# self.out("当前房间roomId:" + self.roomId + " 未加入房间 暂时停止sayHello ")
time.sleep(10)
if(self.roomId == ""): #无房间则自动加入
self.goARoom()
# 定时抓取消息##########################
def getHello(self):
tt = self.getMsgDetaTime
while(True):
if(self.roomId != ""):
self.out("开启抓取发言，" + str(tt) + "s/次")
while(self.roomId != ""):
try:
obj = self.rece()
if(obj != ""):
self.makeHello(obj)
except Exception as e:
self.out(traceback.format_exc())
time.sleep(tt)
# self.out("当前房间roomId:" + self.roomId + " 未加入房间 暂时停止getHello ")
time.sleep(3)
# 抓取发言 json Obj
def rece(self):
# 获取最新时间的消息1530004210 157 s秒
res = ""
url = ""
if(self.linkStart == 0): #正常
url = "http://drrr.com/json.php?update="+str(self.lastMsgTime)
else:
url = "http://drrr.com/json.php?fast=1&update="+str(self.lastMsgTime)
url = "http://drrr.com/json.php?update="+str(self.lastMsgTime)
# self.out(url)
responce=self.http.doGet(url)
if(responce != "" and type(responce) != str ):
jsonStr = responce.read()
if(jsonStr != ""):
res = tool.makeObj(json.loads(jsonStr))
else:
res = ""
self.linkStart = 0
else:
self.out("请求异常:" + str(responce) )
self.linkStart = 1
return res
# 发送消息-添加队列
def send(self, message, id=""):
if(message != None and message != ""):
if(id == "" and self.toIdNow != ""): #若是别人发给自己
id = self.fromIdNow
msg = message
# maxl = 131
# while(len(msg) > maxl):
# sm = msg[0:maxl]
# msg = msg[maxl:99999]
# self.listMsgQue.append( (sm + "", id) )
self.listMsgQue.append( (msg,id))
self.lastEchoTimeQuene = tool.getNowTime()
return
# 发送消息
def doSend(self, message=""):
if(message == ""):
return
self.out("Send." + message)
responce=self.http.doPost("http://drrr.com/room/?ajax=1", {
"message":message, # [0:4],
"url":"",
# to:5a1da324d5e68e6712725a50046f4b75 私聊
})
# self.out("发送[" + message + "]" + re[0:66])
self.lastEchoTime = tool.getNowTime()
return
def doSendId(self, message="", id=""):
if(message == ""):
return
if(id == ""):
self.doSend(message)
return
self.out("SendTo." + self.userIndexRe.get(id,"") + "."+ message)
responce=self.http.doPost("http://drrr.com/room/?ajax=1", {
"message":message, # [0:4],
"url":"",
"to":id
})
# self.out("发送[" + message + "]" + re[0:66])
self.lastEchoTime = tool.getNowTime()
return
def addNextMusic(self, name="", fromName=""):
if(fromName == ""):
fromName = self.fromNameNow
self.nextNames.append( (name,fromName) )
def showNexts(self):
res = ""
if(len(self.nextNames) > 0):
res = "#Play list: "
i = 1
for name,fromName in self.nextNames:
res = res + " \nNo." + str(i) + " " + str(name) + " by " + str(fromName)
i = i+1
else:
res = "接下来随机播放"
self.send(res)
def play(self, name=""):
self.playMusic("aaa", name)
def playurl(self, url=""):
self.playMusic(url)
# 分享音乐
def playMusic(self, url="", name="", fromName=""):
if(self.roomIndex.get(self.roomId, {}).get("music", False) == False):
self.send("当前房间禁止音乐播放" + self.tail)
return
duration = self.maxMusicTimeDefault
music = {}
if(url[0:4] != "http"): #无地址url则是定向点播
if(name == ""): #无名字 则自动换 则考虑下一首序列 (name,fromName)
while(len(self.nextNames) > 0):
(name,fromName) = self.nextNames.pop(0)
if(name != ""):
self.send("接下来播放歌曲[" + name + "]" + self.tail)
music = self.robot.getMusic(name, fromName)
if(music.get("url","") != ""):
break
else:
self.send("没有找到可播放歌曲[" + str(name) + "]")
if(music.get("url", "") == ""):
music = self.robot.turnMusic(self.musicPlayType)
else:
self.send("正在搜索歌曲[" + name + "]" + self.tail)
music = self.robot.getMusic(name, fromName)
url = music.get("url", "")
name = music.get("name", "")
fromName = music.get("fromName", "")
duration = int(music.get("duration", 0))
if(duration < 100):
duration = self.maxMusicTimeDefault
self.musicPlayType = 0 #重置为随机播放
if(fromName != ""):
msg = ""
rooms = self.getUserRoom(fromName)
if(len(rooms) > 0):
room = rooms[0]
if(room.get("id","") == self.roomId): #在当前房间
msg = "一首[" + name + "]送给" + fromName + "" + self.tail
else:
pass
# msg = "Share " + room.get("name")[0:4] + "/" + fromName + "'s[" + name + "]" + "" + self.tail
else: #不在线
pass
# msg = "Then play" + fromName + " ordered [" + name + "]" + "" + self.tail
self.send(msg)
self.out("分享歌曲url=" + url + " name=" + name + " fromName=" + fromName + " time=" + str(tool.calcTime(duration)) )
if(url == ""):
self.send("怼不起,没有找到类似的歌曲,瑟瑟发抖"+self.tail)
return
responce=self.http.doPost("http://drrr.com/room/?ajax=1", {
"music":"music",
"name":name,
"url":url,
})
self.musicNow = {"url":url, "name":name, "fromName":fromName, "duration":duration}
self.maxMusicTime = duration - 6000
self.lastMusicTime = tool.getNowTime()
return
def listMusic(self):
self.showNexts()
# 切歌控制 on/off/turn/prev/next/remove
def music(self, cmd="on"):
self.out("music:" + cmd)
if(cmd == "on"):
self.ifOnMusic = True
self.send("已经打开音乐点播" + self.tail)
elif(cmd == "off"):
self.ifOnMusic = False
self.send("已经关闭音乐点播" + self.tail)
elif(cmd == "turn"):
# self.ifOnMusic = True
self.musicPlayType = 0
self.playMusic()
elif(cmd == "prev"):
# self.ifOnMusic = True
self.musicPlayType = -1
self.playMusic()
elif(cmd == "next"):
# self.ifOnMusic = True
self.musicPlayType = 1
self.playMusic()
elif(cmd == "remove"):
self.robot.removeMusic(self.musicNow.get("url", ""))
self.send("降低音乐" + self.musicNow.get("name") + "播放频率 " + self.tail)
return
# 手动控制
def inputHello(self):
self.out("开启输入监控！")
self.help()
while(True):
try:
cmd=raw_input("")
if(cmd != ""):
if(not self.doMethod(cmd)):
self.out("手动发送:" + cmd)
self.send(cmd)
time.sleep(1)
except Exception as e:
self.out(traceback.format_exc())
return
# 抓取到消息的auto回复
def makeHello(self, obj):
res = ""
try:
# tool.line()
# self.out("抓取到消息obj")
# self.out(obj)
newTime = obj.get("update", self.lastMsgTime)
newTime = int(newTime)
# print(newTime)
self.lastMsgTime = newTime
talks = obj.get('talks', "")
users = obj.get('users', "")
if(users != ""):
room = self.roomIndex.get(self.roomId, "")
if(room != ""):
self.roomIndex[self.roomId]['users'] = users
else:
self.roomIndex[self.roomId] = obj
if(talks != ""):
onceDocount = 0
for item in talks:
# self.out(item)
msgTime = item.get("time", tool.getNowTime())
msgId = item.get('id', " ")
secret = item.get('secret', False)
toId = item.get('to', {}).get('id', '')
toName = item.get('to', {}).get('name', '')
msgType = item.get('type', 'message')
msgData = ""
msgFromName = item.get('from', {}).get('name', "")
fromId = item.get('from', {}).get('id', "")
fromCode = item.get('from', {}).get('tripcode', '')
if(msgFromName == ""):
msgFromName = item.get('user', {}).get('name', "")
fromId = item.get('user', {}).get('id', "")
fromCode = item.get('user', {}).get('tripcode', '')
turnFlag = 0
if(msgFromName != "" and fromId != "" and self.roomMsg.get(msgId, "") == ""):
self.userIndex.get(msgFromName)
# 处理同名 tripcode认证继承 异id问题 名字对应id不一样了 该房间里的cc不是原来记录的了 则删除原来的admin 顶替 n:id--1:名字
if(self.userIndex.get(msgFromName, fromId) != fromId): #非同id
oldId = self.userIndex.pop(msgFromName)
if(self.userIndexRe.get(oldId, "") != ""):
self.userIndexRe.pop(oldId)
# 判定和上次记录 的 zk#aa zk# 是否匹配 绑定用户名和code
lastCode = self.tripcodeIndex.get(msgFromName, "")
if(lastCode == fromCode): #但是 tc code 正常 则转移
if(self.admins.get(oldId,"") != ""): #若有admin则继承
oldValue = self.admins.pop(oldId)
self.addAdmin(fromId, int(oldValue))
turnFlag = 1 #欢迎回来
else: # 否则 没有tc 同名 覆盖 都有tc 但不同 则也覆盖
if(self.admins.get(oldId,"") != ""): #若有admin则继承
oldValue = self.admins.pop(oldId)
self.tripcodeIndex[msgFromName] = fromCode
self.send("[" + msgFromName + "] tc变更[" + str(lastCode) + " - " + str(fromCode) )
self.userIndex[msgFromName] = fromId
self.userIndexRe[fromId] = msgFromName
if(msgType == 'me'):
msgData = item.get('content', "")
elif(msgType == 'message'):
msgData = item.get('message', "")
elif(msgType == 'join' and self.ifWelcom):
# msgFromName = item.get('user', {}).get('name', "")
if(turnFlag == 0):
msgData = '欢迎' + msgFromName + self.tail
else:
msgData = '欢迎回来 ' + msgFromName + self.tail
elif(msgType == 'leave'):
msgData = '' + msgFromName + '' + self.tail
msgData = ''
elif(msgType == 'music'):
music = item.get('music', {})
name = music.get('name', '')
url = music.get('url', '')
# msgData = '悄悄的的把[' + name + ']给记在小本子上 ' + self.tail
######################################################## 不处理
if( self.roomMsg.get(msgId, "") != ""): #已经处理过 或者是自己发送的 或者取出发送者失败
# self.out("旧消息 " + msgId + " type:" + msgType + " data:" + msgData)
continue
self.roomMsg[msgId] = item #标记未已经处理 历史消息
if( msgFromName == self.name or msgFromName == ""):
continue
if(msgType == "me" or msgType == "message"): #只记录聊天消息
self.robot.addMsg(msgId, msgFromName, msgData, msgTime)
#############################################################
#当前消息信息记录
self.fromNameNow = msgFromName
self.fromIdNow = fromId
self.fromNameNow = msgFromName
self.toIdNow = toId
self.toNameNow = toName
if(msgType == 'music'):
music = { "name":name, "url":url, "fromName":msgFromName }
res = self.robot.addMusic(music) #添加用户分享记录
if(res):
self.robot.addHistory(music)
self.musicNow = music
self.lastMusicTime = tool.getNowTime()
# self.roomMsg[msgId] = item #标记未已经处理 历史消息
if(self.status>self.statusMax):
self.status = self.statusMax
elif(self.status < self.statusMin):
self.status = self.statusMin
detaTime = tool.getNowTime() - self.lastEchoTimeQuene # ms 60s
olRan = tool.getRandom(0,self.maxDetaTime) / 1000 #0-180 过于久没有发过消息了 权重高则可能自回复
weight = (self.maxDetaTime - detaTime) / 1000 #多久没说话了 最大多长时间必须说话
ran = int(1.0 * olRan * (1+ 1.0 * (self.status-90) / 100) )
# self.out("Msg." + msgId[0:4] + "." + tool.fill(str(weight) + "" , ' ', 5) + " " + tool.fill(str(olRan) + "->" + str(ran),' ', 5) + "." + tool.fill(msgFromName,' ',8) + "."+tool.fill(msgType,' ',4) + "." + msgData + " ." + str(fromId))
self.out("Msg." + msgId[0:4] + " " + msgFromName[0:10] + "->" + toName[0:10]+ " "+msgType[0:4] + " " + msgData )
msgData = msgData.strip()
flag = 0 #不回复
if(msgType == 'message' or msgType == 'me' ): #普通聊天消息
if( re.search('@' + self.name + " ", msgData) != None): #有@自己 且权重不太低
msgData = re.sub('@' + self.name + "", "", msgData) #摘除@自己
flag = 1
msgData = msgData.strip()
# else:
# self.out("@me 随机数=" + str(ran) + " 小于 说话欲望=" + str(self.status) + " ")
# flag = 2
# msg = "生气程度:" + str(100-self.status) + "%,不想搭理"+self.tail
elif(ran > weight and re.search('@', msgData) == None): # 没有@ 且 权重高 主动搭话概率
flag = 1
elif(toId != ""): #私聊
flag = 1
else:
flag = 10
#admin权限认证
if(self.adminRes != "" and str(msgData) == str(self.adminRes)):
self.out("触发权限admin认证." + str(self.adminRes) + "=" + str(msgData) + "." + msgFromName + "." + fromId)
self.send("认证成功[" + str(msgFromName) + "]")
self.addAdmin(fromId)
flag = 0
msgData = ""
else: #事件
flag = 2
res = ""
self.lastOtherSay = tool.getNowTime() #重置处理时间 黑名单消息不计入消息
if(self.filterFlag(msgData, msgFromName)): #最高级 权限是否黑名单过滤
if(flag == 1 or flag == 10):
if(flag == 1 and self.robot.getUser(msgFromName).get("flag", "0") != "0"):
self.out("不想搭理" + msgFromName)
else:
if(flag == 1 and self.filterCmd(msgData, msgFromName)): #若过滤器未处理 则继续交由下面处理
pass
elif(flag == 10): #让普通消息也接入 cmd 不过没有后续处理
self.filterCmd(msgData, msgFromName)
res = ""
elif(flag == 2):
res = msgData
if(res != "" and flag != 0 and onceDocount < 6): # 最多一次抓取发送3个
res = '' + res
onceDocount = onceDocount + 1
# self.send(res)
#当前消息信息清空
self.fromNameNow = ""
self.fromIdNow = ""
self.fromNameNow = ""
self.toIdNow = ""
self.toNameNow = ""
except Exception as e:
self.out("Exception:" + str(e))
# tool.line()
return res
# /do help 指令控制行为 /do send 你们好
def filterCmd(self, msgData="", fromName=""):
res = True
msgData = msgData.strip()
flag = False
size = len(msgData)
# self.out("filterCmd." + msgData + "." + fromName)
if( re.search('/do', msgData) != None and self.toIdNow != "" ): # /do指令1阶 且只接受私聊控制
if( self.ifAdminName(fromName, 0) ):
res = False
cmd = msgData[3:9999]
self.out(" do method." + str(cmd))
if(not self.doMethod(cmd)):
self.help()
# res = self.doControl(res, msgData, ['管理员', 'admin', 'Admin'], self.getAdmin, fromName, False, 0)
# res = self.doControl(res, msgData, ['help', '帮助'], self.showHelp, fromName, False, 0)
# res = self.doControl(res, msgData, ['nexts', '歌单', '歌单列表'], self.showNexts, fromName, False, 0)
# res = self.doControl(res, msgData, ['next', '接下来播放', '下一首播放'], self.addNextMusic, fromName, True, 0)
#
# res = self.doControl(res, msgData, ['rooms', '房间列表'], self.showRoomList, fromName, True, 0)
#
# res = self.doControl(res, msgData, ['管理员', 'admin', 'Admin'], self.getAdmin, fromName, False, 0)
# res = self.doControl(res, msgData, ['admins', 'score', 'Admins'], self.showAdmin, fromName, False, 0)
# res = self.doControl(res, msgData, ['踢出', 'kick', 'del', '删除','rm'], self.rm, fromName, True,0)
# res = self.doControl(res, msgData, ['host', '房主'], self.host, fromName, True, 2)
# res = self.doControl(res, msgData, ['out', 'leave'], self.outRoom, fromName, False, 0)
#
# # 295796671
# res = self.doSetControl(res, msgData, ['wait', 'stay'], "notWait", False, echoStr="决定在这里住下来")
# res = self.doSetControl(res, msgData, ['开始迎客', 'start hello'], "ifWelcom", True, echoStr="开始迎客")
# res = self.doSetControl(res, msgData, ['停止迎客', 'stop hello'], "ifWelcom", False, echoStr="保持沉默")
# res = self.doSetControl(res, msgData, ['开始存活确认', 'start live'], "ifTime", True, echoStr="开始存活确认")
# res = self.doSetControl(res, msgData, ['停止存活确认', 'stop live'], "ifTime", False, echoStr="停止存活确认")
return res
#设置值类处理链
def doSetControl(self, ifDo, msgData, cmds, attrName, value, echoStr=""):
if(not ifDo): #False 则不处理 且继续返回False
return ifDo
res = True
for item in cmds:
if(msgData == item):
self.out("命中:" + str(item) )
res = False
self.let(attrName, value)
self.send("" + self.name + str(echoStr) + self.tail)
break
return res
# 处理链
def doControl(self, ifDo, msgData, cmds, callback, fromName, argType=False, adminLevel=0):
if(not ifDo): #False 则不处理 且继续返回False
return ifDo
res = True
if(argType):
for item in cmds:
itemLen = len(item)
index = msgData.find(item)
if(index == 0): #头命中
msgData = msgData[itemLen:9999].strip()
self.out("命中:" + str(item) + " 参数:" + str(msgData))
if(adminLevel > 0):
if(self.ifAdminName(fromName, adminLevel)):
callback(msgData)
res = False
else:
callback(msgData)
res = False
break
else:
for item in cmds:
if(msgData == item):
self.out("命中:" + str(item) )
callback()
res = False
break
return res
def addBad(self, fromName):
self.robot.turnUser(fromName, "1")
# [methodName arg1 arg2]
def filterFlag(self, msgData="", fromName=""):
res = True
msgData = msgData.strip()
flag = False
size = len(msgData)
msg = ""
keys = ["别说话", "你别说话", "闭嘴", "shutup"]
statusOn = ['笨蛋', '傻逼', 'sb', 'SB', 'Sb','sB', '傻b', '傻', 'bad']
statusOff = ['开心一点','开心点','我错了', '求你了', '后悔', '收回','我收回','对不起', '悔恨', '不要生气']
if(not flag):
for item in statusOff:
if(msgData == item):
self.robot.turnUser(fromName, "0")
self.status = self.status + self.statusOnDeta
if(self.status >= self.statusMax + self.statusOnDeta):
msg = self.name + "现在没有生气(╯▔皿▔)╯"
elif(self.status >= self.statusDefault):
msg = self.name + "心情好转了 不生气了﹏"
else:
msg = self.name + "气消了一点点，生气值-" + str(self.statusOnDeta) + self.tail
flag = True
break
if(self.robot.getUser(fromName).get("flag", "0") != "0"): #限制黑名单只接受道歉
self.out("黑名单只接受道歉 不想搭理" + fromName)
return False
if(not flag):
for item in keys:
if(msgData == item):
self.status = self.status - self.statusDownDeta
msg = "好的" + ",生气值陡升" + str(self.statusDownDeta) + ",当前" + str(100-self.status) + "% "
flag = True
break
if(not flag):
for item in statusOn:
if(msgData == item):
self.robot.turnUser(fromName, "1")
self.status = self.status - self.statusOffDeta
if(self.status <= self.statusMin - self.statusOffDeta):
msg = self.name + "已经气死了 没这号robot 😕"
elif(self.status <= self.statusMin):
msg = self.name + "已经气炸了 不想再说话了 ε=( o｀ω′)ノ "
else:
msg = self.name + "生气值暴涨" + str(self.statusOffDeta) + "，不想再搭理" + fromName + "了" + self.tail
flag = True
break
if(flag):#状态控制
self.out('filterFlag.' + str(flag) + "." + msgData)
res = False
self.send("" + msg)
return res
# methodName args eg: 'send aaaaa' 第一个空格分开函数名
def doMethod(self, cmd):
#music <on/off/turn/prev/next>
cmd = cmd.strip()
cmds = cmd.split(' ')
if(len(cmds) > 0 and cmds[0] == ""):
cmds.pop(0)
listArgs = cmds
size = len(listArgs)
res = False
if(size > 0):
if(hasattr(self, listArgs[0])):
method = getattr(self, listArgs[0])#获取的是个对象
if(callable(method)):
if(size == 2):
method(listArgs[1])
elif(size == 3):
method(listArgs[1], listArgs[2])
elif(size == 4):
method(listArgs[1], listArgs[2], listArgs[3])
elif(size == 5):
method(listArgs[1], listArgs[2], listArgs[3], listArgs[4])
else:
method()
res = True
else:
res = True
self.out(method)
return res
def let(self, attrName="", value=""):
self.out("变量赋值." + str(attrName) + "." + str(value))
if(hasattr(self, attrName)):
method = getattr(self, attrName)#获取的是个对象
if(callable(method)):
self.out("该属性为方法")
else:
setattr(self, attrName, value)
else:
self.out("该属性不存在")
def shutdown(self):
self.outRoom()
self.out("3秒后关闭程序")
# time.sleep(2)
tool.line()
# sys.exit()
# os.kill()
os._exit(0)
def restart(self):
# self.shutdown()
# self.login()
self.outRoom()
tool.line()
self.out("3秒后重启程序")
# time.sleep(3)
#重启程序
python = sys.executable
os.execl(python, python, * sys.argv)
def test(self):
self.login()
self.getRooms()
# self.goRoom("QGSNLntBvK")
# self.goRoomName("深海")
self.goARoom()
ThreadRun( "DoSend." + str(self.count), self.doHello ).start()
ThreadRun( "SayHello." + str(self.count), self.sayHello ).start()
ThreadRun( "GetHello." + str(self.count), self.getHello ).start()
# ThreadRun( "InputHello." + str(self.count), self.inputHello ).start()
# for i in range(len(self.roomIndex.keys())):
# self.goRoom( self.roomIndex.keys()[i] )
# self.music("turn")
# time.sleep(7)
# self.outRoom()
# time.sleep(3)
return
def test2(self):
self.login()
self.getRooms()
# self.goRoom("QGSNLntBvK")
# self.goRoomName("深海")
# self.goARoom()
# ThreadRun( "DoSend." + str(self.count), self.doHello ).start()
ThreadRun( "SayHello." + str(self.count), self.sayHello ).start()
ThreadRun( "GetHello." + str(self.count), self.getHello ).start()
# ThreadRun( "InputHello." + str(self.count), self.inputHello ).start()
return
class AutoSophia:
def __init__(self, name="0000000", count=0, makeRooms=[]):
self.robot = Robot()
self.http = Http()
self.name = name
self.count = count #编号
self.listMsgQue = [] #消息发送队列
self.timeDetaMsgSend = 1.1 #最小发送消息间隔s
self.makeRooms = makeRooms
self.roomIndex = {} #房间号 及其<用户>信息
self.roomMsg = {} #消息 记录
self.roomId = "" #当前房号
############### 心情模块
self.statusMin = 20
self.statusMax = 95
self.statusDefault = 80
self.status = 90 #说话欲望值 0-100
self.statusOnDeta = 15 #开心
self.statusOffDeta = 15 #难过
self.statusDownDeta = 40 #闭嘴
self.getMsgDetaTime = 1 #抓取消息间隔
self.lastMsgTime = int(
time.time() * 10000) * 1.0 / 10000 #上一次更新房间聊天记录时间
self.lastEchoTime = tool.getNowTime() #上次说话时间
self.maxDetaTime = 1000 * 60 * 3 # 最大沉默时间
self.lastMusicTime = tool.getNowTime()
self.maxMusicTime = 1000 * 60 * 4 #音乐间隔 暂不解析音乐文件时长控制
self.musicNow = {}
self.musicPlayType = -1
self.ifOnMusic = True
self.tail = " の... "
def out(self, obj):
print(
time.strftime("%Y%m%d %H:%M:%S", time.localtime()) + "." +
self.name + "." + str(obj))
return
def login(self):
# tool.line()
self.out("访问主页 获取 token session")
responce = self.http.doGet('http://drrr.com/')
re = responce.read()
soup = BeautifulSoup.BeautifulSoup(re)
# self.out soup.prettify()
nameList = soup.findAll('input', {'name': {'token'}})
if (len(nameList) > 0):
token = nameList[0]['data-value']
token = tool.encode(token)
self.out("抓取成功: ")
self.out("token\t " + token)
self.out("cookie\t " + tool.toString(self.http.getCookie()))
# tool.line()
self.out("模拟登录")
responce = self.http.doPost(
'http://drrr.com/', {
"name": self.name,
"login": "******",
"token": token,
"direct-join": "",
"language": "zh-CN",
"icon": "zaika-2x",
})
else:
self.out("error！ 没能抓取到token")
def createRoom(self, room=""):
# tool.line()
self.out("创建房间")
if (room == ""):
room = {}
room["name"] = self.name + "'s room"
responce = self.http.doPost(
"http://drrr.com/create_room/?", {
"name": room.get("name", ""),
"description": room.get("description", ""),
"limit": room.get("limit", 10),
"music": room.get("music", True),
"submit": room.get("music", "创建房间"),
})
def help(self):
self.out(dir(self))
def showUser(self, user, show=True):
userInfo = "U " + tool.fill(
user.get("device", ""), ' ', 15) + " " + tool.fill(
user.get("icon", ""), ' ', 15) + " " + user.get("name", "")
if (show):
self.out(userInfo)
return userInfo
def showRoom(self, roomId, show=True, i=0):
room = self.roomIndex.get(roomId, "")
if (room == ""):
self.getRooms()
room = self.roomIndex.get(roomId, "")
info = ""
if (room != ""):
info = ("##" + tool.fill(str(i), '#', 40) + "\n--G " + tool.fill(
room["id"], ' ', 15) + " " + tool.fill(
str(room["total"]) + "/" + str(room["limit"]), ' ', 15) +
" " + room["name"]) + "\n"
info = info + "music: " + str(
room.get("music", False)) + " language:" + room.get(
"language", ""
) + "\n" # " 静态房间: " + str(room.get("staticRoom", "")) + ""
# info = info + " hiddenRoom: " + str(room.get("staticRoom", "")) + " 游戏房间: " + str(room.get("gameRoom", "")) + " 成人房间: " + str(room.get("adultRoom", "")) + "\n"
info = info + "Host:" + room.get("host", {}).get("name", "") + "\n"
info = info + "Users: " + "\n"
for item in room.get("users", []):
info = info + "--" + self.showUser(item, False) + "\n"
if (show):
self.out(info)
return info
def showAllRoom(self):
if (self.roomIndex is None or self.roomIndex == ""
or self.roomIndex == {}):
self.getRooms()
tool.line()
self.out("展示所有房间信息")
i = 0
for key in self.roomIndex:
# room = self.roomIndex[key]
self.showRoom(key, True, i)
i = i + 1
tool.line()
# 获取当前房间人员列表
def getRoomUsers(self, roomId=""):
room = self.roomIndex.get(roomId, {})
users = room.get("users", [])
return users
# 获取用户所在房间
def getUserRoom(self,
userName="******",
userId="8f1b61e25098b0427f01d724716b70cb"):
i = 0
res = []
for key in self.roomIndex:
room = self.roomIndex[key]
users = room.get("users", [])
for user in users:
if (user.get("name", "") == userName):
# self.showRoom(key, True, i)
res.append(room)
if (user.get("id", "") == userId):
self.showRoom(key, True, i)
i = i + 1
if (len(res) <= 0):
self.out("用户不在线" + userName)
if (len(res) >= 2):
self.out("该用户多次出现？？？？？")
for item in res:
self.out(item)
tool.line()
return res
def goRoomName(self, roomName):
if (self.roomIndex is None or self.roomIndex == ""
or self.roomIndex == {}):
self.getRooms()
tool.line()
self.out("查找房间名字加入" + roomName)
i = 0
for key in self.roomIndex:
room = self.roomIndex[key]
name = room.get("name", "")
if (re.search(roomName, name) != None):
self.goRoom(key)
break
i = i + 1
tool.line()
def goRoom(self, roomId):
# tool.line()
self.out("加入房间:" + roomId)
# self.showRoom(roomId)
responce = self.http.doGet("http://drrr.com/room/?id=" + roomId)
self.roomId = roomId
# self.send("/me 大家好 我是暖手宝" + self.name + " 可以@ [点歌/turn/prev](*^_^*) @不一定会回 不@也不一定不会回(∩_∩) ")
return
def outRoom(self):
self.out("离开房间:" + self.roomId)
# self.send("/me " + self.name + "好无聊啊 "+self.name +"要出去溜达一会儿" + self.tail)
# self.send("/me "+self.name+"一定会回来的" + self.tail)
# self.send("/me 出去一下，马上回来" + self.tail)
# self.showRoom(self.roomId)
time.sleep(self.timeDetaMsgSend * len(self.listMsgQue) + 1) #等待一会儿消息发送
responce = self.http.doPost("http://drrr.com/room/?ajax=1", {
"leave": "leave",
})
self.roomId = ""
if (responce == "error"):
return False
return True
def getRooms(self, detail=False):
# tool.line()
self.out("获取房间列表")
responce = self.http.doGet("http://drrr.com/lounge?api=json")
jsonObj = tool.makeObj(json.loads(responce.read()))
rooms = jsonObj["rooms"]
if (len(rooms) > 0):
self.roomIndex.clear()
count = 0
userCount = 0
for i in range(len(rooms)):
room = rooms[i]
self.roomIndex[room["id"]] = room
count = count + 1
userCount = userCount + int(room.get("total", 0))
# self.out("#" + str(i) + "\t" + room["id"] + " " + str(room["total"]) + "/" + str(room["limit"]) + "\t " + room["name"])
self.out("共计房间" + tool.fill(str(count), ' ', 5) + " 用户" +
tool.fill(str(userCount), ' ', 5))
# self.out("解析完毕")
return rooms
# 定时消息发送队列
def doHello(self):
while (True):
if (self.roomId != ""):
self.out("开启消息发送队列 deta=" + str(self.timeDetaMsgSend) + "ms")
# self.listMsgQue = []
while (self.roomId != ""):
try:
detaTime = tool.getNowTime() - self.lastEchoTime
if (detaTime > self.timeDetaMsgSend): # 发送时差大于最小发送间隔
if (len(self.listMsgQue) > 0):
msg = self.listMsgQue.pop(0)
self.doSend(msg)
time.sleep(self.timeDetaMsgSend)
except Exception as e:
self.out("消息发送异常 消息队列:")
self.out(self.listMsgQue)
self.out(traceback.format_exc())
# self.out("当前房间roomId:" + self.roomId + " 未加入房间 暂时停止sayHello ")
time.sleep(3)
# 定时发送消息
def sayHello(self):
while (True):
if (self.roomId != ""):
self.out("开启定时发言，最大发言间隔" + str(self.maxDetaTime / 1000) + "s")
dt = 0
while (self.roomId != ""):
try:
# message = "Now Time is "+ time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
detaTime = tool.getNowTime() - self.lastEchoTime # ms
if (detaTime > self.maxDetaTime):
message = "/me " + time.strftime("%Y%m%d %H:%M:%S")
self.send(message)
self.out(str(i) + "\t" + message)
detaTime = tool.getNowTime() - self.lastMusicTime # ms
if (self.ifOnMusic and detaTime > self.maxMusicTime
and len(self.getRoomUsers(self.roomId)) >
1): #音乐开启 且 太久没放歌曲 且当前房间有至少两个人(包括自己robot)
self.playMusic()
if (dt % 300 == 0):
self.getRooms() #定时获取房间最新信息
time.sleep(10)
dt = dt + 10
dt = dt % 3600
except Exception as e:
self.out(traceback.format_exc())
# self.out("当前房间roomId:" + self.roomId + " 未加入房间 暂时停止sayHello ")
time.sleep(3)
# 定时抓取消息##########################
def getHello(self):
tt = self.getMsgDetaTime
while (True):
if (self.roomId != ""):
self.out("开启抓取发言，" + str(tt) + "s/次")
while (self.roomId != ""):
try:
obj = self.rece()
if (obj != ""):
self.makeHello(obj)
except Exception as e:
self.out(traceback.format_exc())
time.sleep(tt)
# self.out("当前房间roomId:" + self.roomId + " 未加入房间 暂时停止getHello ")
time.sleep(3)
# 抓取发言 json Obj
def rece(self):
# 获取最新时间的消息1530004210 157 s秒
res = ""
url = "http://drrr.com/json.php?update=" + str(self.lastMsgTime)
# self.out(url)
responce = self.http.doGet(url)
if (responce != "" and type(responce) != str):
jsonStr = responce.read()
if (jsonStr != ""):
res = tool.makeObj(json.loads(jsonStr))
else:
res = ""
else:
self.out("请求异常:" + str(responce))
return res
# 发送消息-添加队列
def send(self, message):
if (message != None and message != ""):
self.listMsgQue.append(message)
return
# 发送消息
def doSend(self, message):
if (message == ""):
return
self.out("Send." + message)
responce = self.http.doPost(
"http://drrr.com/room/?ajax=1",
{
"message": message, # [0:self.musicPlayType * 4],
"url": "",
})
# self.out("发送[" + message + "]" + re[0:66])
self.lastEchoTime = tool.getNowTime()
return
# 用户权限
def getUserAuth(selfself, name=""):
res = ""
return res
# 分享音乐
def playMusic(self, url="", name="", fromName=""):
self.musicPlayType = 0 #重置为随机播放
if (url[0:4] != "http"): #无地址url则是定向点播
if (name == ""): #无名字 则自动换
music = self.robot.turnMusic(self.musicPlayType)
else:
self.send("/me 正在搜索歌曲[" + name + "]" + self.tail)
music = self.robot.getMusic(name, fromName)
url = music.get("url", "")
name = music.get("name", "")
fromName = music.get("fromName", "")
if (fromName != ""):
msg = ""
rooms = self.getUserRoom(fromName)
if (len(rooms) > 0):
room = rooms[0]
if (room.get("id", "") == self.roomId): #在当前房间
pass
# msg = "/me 一首[" + name + "]送给" + fromName + "" + self.tail
else:
pass
# msg = "/me Share " + room.get("name")[0:4] + "/" + fromName + "'s[" + name + "]" + "" + self.tail
else: #不在线
pass
# msg = "/me Then play" + fromName + " ordered [" + name + "]" + "" + self.tail
self.send(msg)
self.out("分享歌曲url=" + url + " name=" + name + " fromName=" + fromName)
if (url == ""):
# self.send("/me 怼不起,没有找到类似的歌曲,瑟瑟发抖"+self.tail)
return
responce = self.http.doPost("http://drrr.com/room/?ajax=1", {
"music": "music",
"name": name,
"url": url,
})
self.musicNow = {"url": url, "name": name, "fromName": fromName}
self.lastMusicTime = tool.getNowTime()
return
def listMusic(self):
self.out(self.robot.listMusic)
# 切歌控制 on/off/turn/prev/next/remove
def music(self, cmd="on"):
self.out("music:" + cmd)
if (cmd == "on"):
self.ifOnMusic = True
self.send("/me 已经打开音乐点播" + self.tail)
elif (cmd == "off"):
self.ifOnMusic = False
self.send("/me 已经关闭音乐点播" + self.tail)
elif (cmd == "turn"):
self.ifOnMusic = True
self.musicPlayType = 0
self.playMusic()
elif (cmd == "prev"):
self.ifOnMusic = True
self.musicPlayType = -1
self.playMusic()
elif (cmd == "next"):
self.ifOnMusic = True
self.musicPlayType = 1
self.playMusic()
elif (cmd == "remove"):
self.robot.removeMusic(self.musicNow.get("url", ""))
self.send("/me 降低音乐" + self.musicNow.get("name") + "播放频率 " +
self.tail)
return
# 手动控制
def inputHello(self):
self.out("开启输入监控！")
self.help()
while (True):
try:
cmd = raw_input("")
if (cmd != ""):
if (not self.doMethod(cmd.split(" "))):
self.out("手动发送:" + cmd)
self.send(cmd)
time.sleep(1)
except Exception as e:
self.out(traceback.format_exc())
return
# 抓取到消息的auto回复
def makeHello(self, obj):
res = ""
try:
# tool.line()
# self.out("抓取到消息obj")
# self.out(obj)
newTime = obj.get("update", self.lastMsgTime)
newTime = int(newTime)
# print(newTime)
self.lastMsgTime = newTime
talks = obj.get('talks', "")
users = obj.get('users', "")
if (users != ""):
room = self.roomIndex.get(self.roomId, "")
if (room != ""):
self.roomIndex[self.roomId]['users'] = users
else:
self.roomIndex[self.roomId] = obj
if (talks != ""):
onceDocount = 0
for item in talks:
# self.out(item)
msgTime = item.get("time", tool.getNowTime())
msgId = item.get('id', " ")
msgType = item.get('type', 'message')
msgData = ""
msgFromName = item.get('from', {}).get('name', "")
if (msgFromName == ""):
msgFromName = item.get('user', {}).get('name', "")
if (msgType == 'me'):
msgData = item.get('content', "")
elif (msgType == 'message'):
msgData = item.get('message', "")
elif (msgType == 'join'):
# msgFromName = item.get('user', {}).get('name', "")
msgData = '欢迎' + msgFromName + self.tail
elif (msgType == 'leave'):
msgData = ' ' + msgFromName + ' 默默的离开了 ' + self.tail
elif (msgType == 'music'):
music = item.get('music', {})
name = music.get('name', '')
url = music.get('url', '')
# msgData = '悄悄的的把[' + name + ']给记在小本子上 ' + self.tail
######################################################## 不处理
if (self.roomMsg.get(msgId, "") !=
""): #已经处理过 或者是自己发送的 或者取出发送者失败
# self.out("旧消息 " + msgId + " type:" + msgType + " data:" + msgData)
break
if (msgType == "me" or msgType == "message"): #只记录聊天消息
self.robot.addMsg(msgId, msgFromName, msgData, msgTime)
if (msgFromName == self.name or msgFromName == ""):
break
#############################################################
if (msgType == 'music'):
music = {
"name": name,
"url": url,
"fromName": msgFromName
}
res = self.robot.addMusic(music) #添加用户分享记录
if (res == 1): #更新则不提示
msgData = ""
self.musicNow = music
self.lastMusicTime = tool.getNowTime()
self.roomMsg[msgId] = item #标记未已经处理 历史消息
if (self.status > self.statusMax):
self.status = self.statusMax
elif (self.status < self.statusMin):
self.status = self.statusMin
detaTime = tool.getNowTime() - self.lastEchoTime # ms 60s
olRan = tool.getRandom(
0,
self.maxDetaTime) / 1000 #0-180 过于久没有发过消息了 权重高则可能自回复
weight = (self.maxDetaTime -
detaTime) / 1000 #多久没说话了 最大多长时间必须说话
ran = int(1.0 * olRan * (1 + 1.0 *
(self.status - 60) / 100))
self.out("新消息 " + msgId + " 发言权" +
tool.fill(str(weight) + "", ' ', 6) + " 随机数" +
tool.fill(str(olRan) + "->" + str(ran), ' ', 6) +
" from:" + tool.fill(msgFromName, ' ', 12) +
" type:" + tool.fill(msgType, ' ', 6) + " data:" +
msgData)
flag = 0 #不回复
if (msgType == 'message' or msgType == 'me'): #普通聊天消息
if (re.search('@' + self.name + " ", msgData) !=
None): #有@自己 且权重不太低
msgData = re.sub('@' + self.name + " ", "",
msgData) #摘除@自己
ran = tool.getRandom(0, 100)
if (ran < self.status):
flag = 1
else:
self.out("@me 随机数=" + str(ran) + " 小于 说话欲望=" +
str(self.status) + " ")
flag = 2
msg = "生气程度:" + str(
100 - self.status) + "%,不想搭理" + self.tail
elif (ran > weight and re.search('@', msgData)
== None): # 没有@ 且 权重高 主动搭话概率
flag = 1
else: #事件
flag = 2
res = ""
if (self.filterFlag(msgData, msgFromName)): #最高级 权限是否黑名单过滤
if (flag == 1):
if (self.robot.getUser(msgFromName).get(
"flag", "0") != "0"):
self.out("不想搭理" + msgFromName)
else:
if (self.filterCmd(
msgData,
msgFromName)): #若过滤器未处理 则继续交由下面处理
robotRes = self.robot.do(
msgData, self.name)
code = str(robotRes.get("code", ""))
if (code[0:1] != '4'):
res = '@' + str(
msgFromName) + " " + robotRes.get(
"text", "")
else:
self.out("robot接口调用失败 code=" + code)
elif (flag == 2):
res = msg
if (res != "" and flag != 0
and onceDocount < 6): # 最多一次抓取发送3个
res = '/me ' + res
onceDocount = onceDocount + 1
self.send(res)
except Exception as e:
self.out("Exception:" + str(e))
# tool.line()
return res
# /do help 指令控制行为
def filterCmd(self, msgData="", fromName=""):
res = True
msgData = msgData.strip()
flag = False
size = len(msgData)
pr = ['放音乐', '播放音乐', '放歌', '开启放歌']
if (not flag):
for item in pr:
if (msgData == item):
msgData = ""
self.music("on")
flag = True
break
pr = ['不要放音乐', '停止放歌', '停止音乐', '别放歌', '关闭放歌', '关闭音乐', '别放了']
if (not flag):
for item in pr:
if (msgData == item):
msgData = ""
self.music("off")
flag = True
break
pr = ['prev', '上一曲', '上一首', '换回去']
if (not flag):
for item in pr:
if (msgData == item):
msgData = ""
self.musicPlayType = -1
flag = True
break
nnn = [
'下一曲', '下一首', '切歌', '换', '换歌', '不好听', '难听', '难听死了', '换换换', '换一首',
'next', 'turn'
]
if (not flag):
for item in nnn:
if (msgData == item):
msgData = ""
flag = True
break
ppp = [
'点歌', 'music', '歌曲', '点播', '下面播放', '想听', '播放', '放', 'play', 'mp3'
]
if (not flag):
for item in ppp:
itemLen = len(item)
index = msgData.find(item)
if (index == 0): #头命中
msgData = msgData[itemLen:9999].strip()
flag = True
break
elif (index > 0 and index == size - itemLen): # 尾命中
msgData = msgData[0:size - itemLen].strip()
flag = True
break
ooo = [
('一首', '献给大家'),
('一首', '送给大家'),
('点', '这首歌'),
]
if (not flag):
for before, after in ooo:
index = msgData.find(before)
if (index == 0):
index1 = msgData.find(after)
if (index1 + len(after) == size):
flag = True
msgData = msgData[len(before):index1].strip()
break
if (flag): #抽离点歌 名字
self.out('filterCmd.' + str(flag) + "." + msgData)
res = False
self.playMusic(url="", name=msgData, fromName=fromName)
elif (re.search('/do ', msgData) != None):
res = False
cmd = msgData[4:9999]
cmd = cmd.strip()
cmds = cmd.split(' ')
if (len(cmds) > 0 and cmds[0] == ""):
cmds.pop(0)
if (not self.doMethod(cmds)):
self.send("/me ########## @" + self.name +
" /do music <on/off/turn/prev/next> ########")
return res
# [methodName arg1 arg2]
def filterFlag(self, msgData="", fromName=""):
res = True
msgData = msgData.strip()
flag = False
size = len(msgData)
msg = ""
keys = ["别说话", "你别说话", "闭嘴", "shutup"]
statusOn = ['笨蛋', '傻逼', 'sb', 'SB', 'Sb', 'sB', '傻b', '傻']
statusOff = [
'开心一点', '开心点', '我错了', '求你了', '后悔', '收回', '我收回', '对不起', '悔恨', '不要生气'
]
if (not flag):
for item in statusOff:
if (msgData == item):
self.robot.turnUser(fromName, "0")
self.status = self.status + self.statusOnDeta
if (self.status >= self.statusMax + self.statusOnDeta):
msg = self.name + "现在没有生气(╯▔皿▔)╯"
elif (self.status >= self.statusDefault):
msg = self.name + "心情好转了 不生气了﹏"
else:
msg = self.name + "气消了一点点，生气值-" + str(
self.statusOnDeta) + self.tail
flag = True
break
if (self.robot.getUser(fromName).get("flag", "0") != "0"): #限制黑名单只接受道歉
res = True
self.out("黑名单只接受道歉 不想搭理" + fromName)
return res
if (not flag):
for item in keys:
if (msgData == item):
self.status = self.status - self.statusDownDeta
msg = "好的" + ",生气值陡升" + str(
self.statusDownDeta) + ",当前" + str(100 -
self.status) + "% "
flag = True
break
if (not flag):
for item in statusOn:
if (msgData == item):
self.robot.turnUser(fromName, "1")
self.status = self.status - self.statusOffDeta
if (self.status <= self.statusMin - self.statusOffDeta):
msg = self.name + "已经气死了 没这号robot 😕"
elif (self.status <= self.statusMin):
msg = self.name + "已经气炸了 不想再说话了 ε=( o｀ω′)ノ "
else:
msg = self.name + "生气值暴涨" + str(
self.statusOffDeta
) + "，不想再搭理" + fromName + "了" + self.tail
flag = True
break
if (flag): #状态控制
self.out('filterFlag.' + str(flag) + "." + msgData)
res = False
self.send("/me " + msg)
return res
def doMethod(self, listArgs):
size = len(listArgs)
res = False
if (size > 0):
if (hasattr(self, listArgs[0])):
method = getattr(self, listArgs[0]) #获取的是个对象
if (callable(method)):
if (size == 2):
method(listArgs[1])
elif (size == 3):
method(listArgs[1], listArgs[2])
elif (size == 4):
method(listArgs[1], listArgs[2], listArgs[3])
elif (size == 5):
method(listArgs[1], listArgs[2], listArgs[3],
listArgs[4])
else:
method()
res = True
else:
self.out(method)
return res
def test(self):
self.login()
self.getRooms()
# self.goRoom("YfdWkQ1lEs")
self.goRoomName("上帝")
ThreadRun("DoSend." + str(self.count), self.doHello).start()
ThreadRun("SayHello." + str(self.count), self.sayHello).start()
ThreadRun("GetHello." + str(self.count), self.getHello).start()
ThreadRun("InputHello." + str(self.count), self.inputHello).start()
tool.wait()
return
def test2(self):
self.login()
self.getRooms()
self.createRoom()
# self.goRoom("YfdWkQ1lEs")
# self.goRoomName("上帝")
ThreadRun("DoSend." + str(self.count), self.doHello).start()
ThreadRun("SayHello." + str(self.count), self.sayHello).start()
ThreadRun("GetHello." + str(self.count), self.getHello).start()
ThreadRun("InputHello." + str(self.count), self.inputHello).start()
tool.wait()
class iCloudStorage:
def __init__(self, url, account=None, passwd=None, encoding='utf-8'):
self.url = url.rstrip("/")
self.account = account
self.passwd = passwd
self.encoding = encoding
self.token = None
self.uttpUtil = Http()
#认证
def auth(self):
auth_url = self.url + '/auth'
if self.account == None:
_params = None
return False
else:
_params = {'account': self.account, 'password': self.passwd}
reponse = self.uttpUtil.doPost(auth_url,
params=_params,
accept='application/json')
jsonUtil = JsonParser(code=self.encoding)
result = jsonUtil.decode(reponse)
if 'result' in result:
if result['result'] != "OK":
raise CustomAuthException('认证失败')
else:
raise CustomAuthException('认证失败')
self.token = result['content']['token']
return True
#通过共享名称获取文件
def downloadShare(self, shareFile, fileName, params=None):
dowload_url = self.url + '/files/get/s/' + shareFile
bFlag = self.uttpUtil.download(dowload_url, fileName, params)
return bFlag
#通过文件ID获取文件
def download(self, fileId, fileName, params=None):
dowload_url = self.url + '/files/get/file/' + fileId + "?token=" + self.token
bFlag = self.uttpUtil.download(dowload_url, fileName, params)
return bFlag
#通过文件ID获取文件属性
def getFileAttribute(self, fileId, params=None):
_url = self.url + '/files/get/fileMetadata/' + fileId + "?token=" + self.token
reponse = self.uttpUtil.doGet(_url, params, accept='application/json')
jsonUtil = JsonParser(code=self.encoding)
result = jsonUtil.decode(reponse)
if 'result' in result:
if result['result'] != "OK":
raise CustomAuthException('获取文件失败')
else:
raise CustomAuthException('获取文件失败')
return result['content']
#通过共享名称获取文件属性
def getShareFileAttribute(self, shareFile, params=None):
_url = self.url + '/files/get/s/fileMetadata/' + shareFile
reponse = self.uttpUtil.doGet(_url, params, accept='application/json')
jsonUtil = JsonParser(code=self.encoding)
result = jsonUtil.decode(reponse)
if 'result' in result:
if result['result'] != "OK":
raise CustomAuthException('获取文件失败')
else:
raise CustomAuthException('获取文件失败')
result['content']['imageid'] = shareFile
return result['content']
class Auto:
def __init__(self, name="Test"):
self.name = name
self.http = Http()
self.db = Database()
self.db.execute('''
create table if not exists auto(
url text,
about text,
time text,
v1 text,
v2 text,
v3 text,
v4 text,
v5 text
)
''')
return
# 日志输出
def out(self, obj):
print(
time.strftime("%Y%m%d %H:%M:%S", time.localtime()) + "." +
self.__module__ + self.name + "." + str(obj))
# tools.out(obj)
return
# 实时控制帮助
def help(self):
self.out(dir(self))
return
# doMethod([methodName arg1 arg2]) -> methodName(arg1,arg2)
def doMethod(self, listArgs):
size = len(listArgs)
res = None
if (size > 0):
if (hasattr(self, listArgs[0])):
method = getattr(self, listArgs[0])
if (callable(method)):
if (size == 2):
res = method(listArgs[1])
elif (size == 3):
res = method(listArgs[1], listArgs[2])
elif (size == 4):
res = method(listArgs[1], listArgs[2], listArgs[3])
elif (size == 5):
res = method(listArgs[1], listArgs[2], listArgs[3],
listArgs[4])
else:
res = method()
else:
self.out(method)
return res
# 手动命令监控
def inputHello(self):
self.out("开启输入监控！")
self.help()
while (True):
try:
cmd = raw_input("")
if (cmd != ""):
if (not self.doMethod(cmd.split(" "))):
self.doCmd(cmd)
time.sleep(1)
except Exception as e:
self.out(repr(e))
return
# 测试用
def test(self):
self.login()
# 输入监控线程
ThreadRun("InputHello." + str(self.name), self.inputHello).start()
tool.wait()
return
# 非函数调用 属性变量查看 其他的指令控制
def doCmd(self, cmd):
self.out("其他指令." + str(cmd))
return
# 认证登录
def login(self):
self.out("访问主页 获取 token session")
responce = self.http.doGet('http://drrr.com/')
re = responce.read()
soup = BeautifulSoup.BeautifulSoup(re)
nameList = soup.findAll('input', {'name': {'token'}})
if (len(nameList) > 0):
token = nameList[0]['data-value']
token = tool.encode(token)
self.out("抓取成功: ")
self.out("token\t " + token)
self.out("cookie\t " + tool.toString(self.http.getCookie()))
self.out("模拟登录")
responce = self.http.doPost(
'http://drrr.com/', {
"name": name,
"login": "******",
"token": token,
"direct-join": "",
"language": "zh-CN",
"icon": icon,
})
if (responce != "error"):
return True
else:
return False
else:
self.out("error！ 没能抓取到token")
return False
class Robot:
"""机器人智能语义应答"""
id = ""
name = ""
def __init__(self):
self.id = "test id"
self.name = "test name"
self.http = Http()
self.auto163 = Auto163("Music")
self.apiKey = "bfbf6432b655493b9e861b470bca9921"
self.userId = "WalkerDust"
self.db = Database()
self.db.execute('''
create table if not exists music(
url text primary key,
name text,
duration text,
fromName text,
count text
)
''')
self.db.execute('''
create table if not exists user(
name text primary key,
id text,
icon text,
flag text
)
''')
self.db.execute('''
create table if not exists msg(
id text primary key,
userName text,
data text,
time text
)
''')
self.initMusic()
self.palyHistoryMusic = []
return
def out(self, obj):
print(self.__module__ + "." + str(obj))
return
# 音乐模块
def initMusic(self):
li = ""
count = self.db.getCount("select * from music ")
if (count <= 0): #毫无数据 则 加入默认数据
li = []
with open('music.txt', 'r') as f:
data = f.readlines()
for item in data:
name = item.strip()
url = "http://39.107.26.100:8088/file/" + name
fromName = ""
self.db.execute('insert into music values(?,?,?,?,?)', url,
name, "0", fromName, "5")
return
def nextMusic(self, name=""):
self.nextNames.append(name)
# 内部点播 若有名字则按照名字本地搜索和云搜索 否则 按照type切歌
def turnMusic(self, playType):
music = {}
if (playType == -1 and len(self.palyHistoryMusic) > 1): #上一曲
music = self.palyHistoryMusic.pop() #弹出顶 取末
music = self.palyHistoryMusic[-1]
else:
cc = 0
while (music.get("url", "") == ""):
size = self.db.getCount("select * from music ")
if (size <= 0):
break
num = 5
page = int(1.0 * size / num)
page = tool.getRandom(0, page)
(size,
listRes) = self.db.executeQueryPage("select * from music",
page, num)
getSize = len(listRes)
count = tool.getRandom(0, getSize)
music = listRes[count]
tool.line()
self.out("随机找到歌曲页 " + "size:" + str(size) + " page:" +
str(page) + " num:" + str(num) + " listResSize:" +
str(getSize))
for item in listRes:
self.out("url:" + item.get("url") + " name:" +
item.get("name") + " fromName:" +
item.get("fromName") + " time:" +
str(tool.calcTime(item.get("duration", 0))))
self.out("选中了" + str(count))
tool.line()
url = music.get("url", "")
if (self.http.existAudio(url)):
break
else:
self.out("访问音乐文件失败" + url)
self.removeMusic(url)
music = {}
if (cc > 20):
break
cc = cc + 1
return music
def getMusic(self, musicName="", fromName=""):
music = {}
if (musicName != ""):
# res = self.db.executeQueryOne("select * from music where name=? ", musicName)
# if(res.get("url", "") != ""):
# music = res
# else:
res = self.auto163.getMusic(musicName, fromName) # [music,music]
rres = []
for item in res:
if (self.addMusic(item)): #有效歌曲才参与接下来的选择
rres.append(item)
res = rres
if (len(res) > 0):
#按照 前面权重依次递减 1 2 3 4 5 -> 16 8 4 2 1
ii = tool.getRandomWeight(0, len(res))
self.out("0, " + str(len(res)) + " -> " + str(ii))
music = res[ii]
if (music.get("url", "") != ""):
self.addHistory(music)
return music
def addHistory(self, music):
self.palyHistoryMusic.append(music)
if (len(self.palyHistoryMusic) > 20):
self.palyHistoryMusic.pop(0)
# 外部点播音乐记录
def addMusic(self, music):
url = music.get("url", "")
name = music.get("name", "")
fromName = music.get("fromName", "")
duration = music.get("duration", 0)
oldMusic = self.db.executeQueryOne(
"select * from music where url = ? ", url)
if (oldMusic.get("url", "") == ""):
# self.out("添加音乐")
# self.out(music)
self.db.execute('insert into music values(?,?,?,?,?)', url, name,
duration, fromName, "1")
else: #更新该音乐数据
# self.out("更新音乐")
count = int(oldMusic.get("count", 0))
count = str(count + 1)
music["count"] = count
# self.out(music)
self.db.execute(
'update music set name=?, fromName=?, count=?,duration=? where url=?',
name, fromName, count, duration, url)
if (not self.http.existAudio(url)):
self.out("访问音乐文件失败" + url)
self.removeMusic(url)
return False
return True
def removeMusic(self, url=""):
index = 0
self.out("移除音乐" + url)
self.db.execute('delete from music where url = ? ', url)
return
# 人员信息管理 权限 状态
# name text primary key,
# id text,
# icon text,
# flag text
def addUser(self, user={}):
name = user.get("name", "")
id = user.get("id", "")
icon = user.get("icon", "")
flag = user.get("flag", "0")
if (self.db.getCount("select * from user where name=?", name) > 0):
self.db.execute("update user set id=?,icon=?,flag=? where name=?",
id, icon, flag, name)
else:
self.db.execute(
"insert into user(name,id,icon,flag) values(?,?,?,?)", name,
id, icon, flag)
def turnUser(self, name, flag):
if (self.db.getCount("select * from user where name=?", name) > 0):
self.db.execute("update user set flag=? where name=?", flag, name)
else:
self.db.execute("insert into user(name,flag) values(?,?)", name,
flag)
def getUser(self, name=""):
res = self.db.executeQueryOne("select * from user where name=?", name)
if (res == None):
res = {}
return res
# 消息监控
def addMsg(self, id, userName, data, msgTime):
if (self.db.getCount("select * from msg where id=?", id) <= 0):
self.db.execute('insert into msg values(?,?,?,?)', id, userName,
data, msgTime)
return
# 智能应答
def do(self, msg, userId="CC"):
res = ""
if (msg == ""):
return "消息不能为空"
response = self.http.doPost(
'http://www.tuling123.com/openapi/api',
{
"key": self.apiKey,
"info": msg,
"userid": userId,
# "userInfo":{
# "apiKey":self.apiKey,
# "userId":self.userId,
# },
# "reqType":0,
# "perception":{
# "inputText":{
# "text":msg,
# }
# },
})
jsonStr = response.read()
if (jsonStr != ""):
res = tool.makeObj(json.loads(jsonStr))
code = res.get("code", "")
self.out("Robot. " + str(msg) + " -> " + jsonStr)
else:
self.out("Robot. " + str(msg) + " -> error !!!!!!!!! ")
return res
def doParse(self, obj):
res = ""
text = obj.get("text", "")
res = res + text
url = obj.get("url", "")
list = obj.get("list", "")
if (url != ""):
res = res + " \n" + url
i = 0
if (list != "" and len(list) > 0):
for item in list:
# res = res + " \n"
# for key,value in item.items():
# res = res + " \n" + str(value)
ttt = item.get("url", "")
if (ttt != ""):
res = res + " \n" + str(ttt)
ttt = item.get("detailurl", "")
if (ttt != ""):
res = res + " \n" + str(ttt)
if (len(res) > 200):
break
# {
# "name": "鱼香肉丝",
# "icon": "",
# "info": "瘦肉、黑木耳、胡萝卜、靑椒、豆瓣酱，葱姜蒜、白糖，香醋，料酒",
# "detailurl": "http://m.xiachufang.com/recipe/100352761/?ref=tuling"
# }
return res
def test(self):
while (True):
cmd = raw_input("")
if (cmd != ""):
res = self.do(cmd)
self.out(res)
time.sleep(1)
return
