def __init__(self, channel, nickname, server, port):
Bot.__init__(self)
self.idle = {}
self.resumeState = {}
self.adapter = TestBot(self, channel, nickname, server, port)
self.adapter.start()
class GustafoBot(Bot):
CHAT = 0
TIMEOUT = 30.0
def __init__(self, channel, nickname, server, port):
Bot.__init__(self)
self.idle = {}
self.resumeState = {}
self.adapter = TestBot(self, channel, nickname, server, port)
self.adapter.start()
def forget(self):
for timer in self.idle.values():
timer.cancel()
self.idle = {}
self.resumeState = {}
State.forget()
db = Database()
db.drop_tables()
db.close_conn()
self.on_join()
def die(self):
State.die()
for timer in self.idle.values():
timer.cancel()
def send_message(self, nick, msg):
to_send = nick + ": " + msg
self.adapter.send_message(to_send)
def on_user_join(self, nick, timestamp):
print "##### JOIN #####"
knowers = self.adapter.get_users()
knowers.remove(self.adapter.nickname)
timestamp = time.strftime("%X", time.localtime())
msg = "joined at " + timestamp
context = {'author': nick,
'recipient': "",
'msg': msg,
'knowers': knowers}
State.forceState(FindGossip, context)
def on_user_exit(self, nick, timestamp):
print "##### EXIT #####"
knowers = self.adapter.get_users()
knowers.remove(self.adapter.nickname)
timestamp = time.strftime("%X", time.localtime())
msg = "left at " + timestamp
context = {'author': nick,
'recipient': "",
'msg': msg,
'knowers': knowers}
State.forceState(FindGossip, context)
def on_join(self):
self.idle[GustafoBot.CHAT] = Timer(GustafoBot.TIMEOUT, self.on_chat_inactive)
self.idle[GustafoBot.CHAT].start()
def on_chat_inactive(self):
State.users = self.adapter.get_users()
users = self.adapter.get_users()
users.remove(self.adapter.nickname)
if "foaad" in users:
user = "******"
elif len(users) > 0:
random.shuffle(users)
user = users[0]
else:
self.idle[GustafoBot.CHAT] = Timer(GustafoBot.TIMEOUT, self.on_chat_inactive)
res = State.forceState(InitialOutreach, {'_nick': user})
if res is not None:
self.send_message(user, res)
self.idle[user] = Timer(GustafoBot.TIMEOUT, self.on_user_inactive, [user])
self.idle[user].start()
def on_user_inactive(self, nick):
State.users = self.adapter.get_users()
if State.userState[nick] is not SolicitResponse:
self.resumeState[nick] = State.userState[nick]
res = State.forceState(SolicitResponse, {'_nick': nick})
#res = State.forceState(SolicitUser,{'_nick': nick})
self.idle[nick] = Timer(GustafoBot.TIMEOUT, self.on_user_inactive, [nick])
self.idle[nick].start()
else:
res = State.forceState(GiveUpState, {'_nick': nick})
del(State.userState[nick])
del(self.idle[nick])
if len(State.userState) == 0:
self.idle[GustafoBot.CHAT] = Timer(10.0, self.on_chat_inactive)
self.idle[GustafoBot.CHAT].start()
if res is not None:
self.send_message(nick, res)
def on_message(self, user, timestamp, msg):
self.idle[GustafoBot.CHAT].cancel()
if self.idle.get(user, None) is not None:
self.idle[user].cancel()
if State.userState[user] is SolicitResponse:
State.userState[user] = self.resumeState[user]
it = time.time()
print self.adapter.get_users()
State.users = self.adapter.get_users()
res = State.query(user, msg)
rt = time.time()
if rt - it < 3.0:
print "Sleep:", 3.0 - (rt - it)
time.sleep(3.0 - (rt - it))
if res is not None:
self.send_message(user, res)
print user
self.idle[user] = Timer(GustafoBot.TIMEOUT, self.on_user_inactive, [user])
self.idle[user].start()
def on_chat(self, t, f, msg):
knowers = self.adapter.get_users()
knowers.remove(self.adapter.nickname)
context = {'author': f,
'recipient': t,
'msg': msg,
'knowers': knowers}
State.forceState(FindGossip, context)