cleaned comments, added help and info funcs

This commit is contained in:
Shauna 2013-12-21 22:31:39 -05:00
parent 8328ef9b3a
commit 12de7a4db5

39
bot.py
View File

@ -2,6 +2,7 @@
import socket import socket
import time import time
import Queue import Queue
import random
from threading import Thread from threading import Thread
# Some basic variables used to configure the bot # Some basic variables used to configure the bot
@ -23,10 +24,10 @@ joinchan(channel)
# Creates separate thread for reading messages from the server # Creates separate thread for reading messages from the server
def getIRC(): def getIRC():
while True: while True:
ircmsg = ircsock.recv(2048) # receive data from the server <-------- this needs to be fixed! Threading? Buffer file? ircmsg = ircsock.recv(2048) # receive data from the server
ircmsg = ircmsg.strip('\n\r') # removing any unnecessary linebreaks. ircmsg = ircmsg.strip('\n\r') # removing any unnecessary linebreaks.
q.put(ircmsg) q.put(ircmsg) # Put in queue for main loop to read
print(ircmsg) # Here we print what's coming from the server print(ircmsg)
q = Queue.LifoQueue() q = Queue.LifoQueue()
t = Thread(target=getIRC) t = Thread(target=getIRC)
@ -40,15 +41,11 @@ class newcomer(object): # Newcomer class created when someone joins the room
self.nick = nick self.nick = nick
self.born = time.time() self.born = time.time()
self.status = 0 self.status = 0
print "newcomer object named " + self.nick + " created"
def changeStatus(self,status): def changeStatus(self,status):
self.status = status self.status = status # Right now there's just status 0 (not replied to) and 1 (replied to)
print "status of newcomer object changed to " + str(self.status)
# Status options: 0 (nothing has happened since joining), 1 (someone has talked but not to newcomer), 2 (someone has talked to newcomer)
def aroundFor(self): def aroundFor(self):
# print "timecheck: " + str(self.nick) + " has been around for " + str(time.time() - self.born)
return time.time() - self.born return time.time() - self.born
@ -56,11 +53,15 @@ class newcomer(object): # Newcomer class created when someone joins the room
def ping(): # Responds to server Pings. def ping(): # Responds to server Pings.
ircsock.send("PONG :pingis\n") ircsock.send("PONG :pingis\n")
def hello(speaker): # This function responds to a user that inputs "Hello Mybot" def hello(speaker,greeting): # This function responds to a user that inputs "Hello Mybot"
ircsock.send("PRIVMSG "+ channel +" :Hello! "+ speaker + "\n") ircsock.send("PRIVMSG " + channel +" :" + greeting + " " + speaker + "\n")
def welcome(speaker): def help(speaker): # This function explains what the bot is when queried.
ircsock.send("PRIVMSG "+ channel +" :Welcome "+ speaker + "!\n") ircsock.send("PRIVMSG " + channel +" :I'm a bot! I'm from here: https://github.com/shaunagm/oh-irc-bot. You can change my behavior by submitting a pull request or by talking to shauna. \n")
def welcome(newcomer): # This welcomes a specific person.
ircsock.send("PRIVMSG "+ channel +" :Welcome "+ newcomer + "!\n")
ircsock.send("PRIVMSG "+ channel +" :(pssst shauna there's someone here)\n")
#### Main function #### Main function
@ -80,16 +81,16 @@ while 1:
if ircmsg.find("PRIVMSG "+ channel) != -1: # If someone has spoken into the channel if ircmsg.find("PRIVMSG "+ channel) != -1: # If someone has spoken into the channel
for i in newList: for i in newList:
print "speaker: " + speaker
print "i.nick: " + i.nick
if speaker != i.nick: # Don't turn off response if the person speaking is the person who joined. if speaker != i.nick: # Don't turn off response if the person speaking is the person who joined.
i.changeStatus(1) # set status to "someone has spoken in channel" for all waiting newcomers i.changeStatus(1) # set status to "someone has spoken in channel" for all waiting newcomers
print "yay now never respond"
# if someone has spoken directly to a newcomer, set status to 'someone has replied to newcomer' if ircmsg.find(botnick) != -1 and ircmsg.find("PRIVMSG #") != -1: # If someone talks to (or refers to) the bot
helloArray = ['Hello','hello','Hi','hi','Hey','hey','Yo','yo ','Sup','sup']
if ircmsg.find(":Hello "+ botnick) != -1: # Response to 'Hello botnick' helpArray = ['Help','help','Info','info','faq','FAQ','explain yourself','EXPLAIN YOURSELF']
hello(speaker) if any(x in ircmsg for x in helloArray):
hello(speaker,random.choice(helloArray))
if any(y in ircmsg for y in helpArray):
help(speaker)
if ircmsg.find("PING :") != -1: # if the server pings us then we've got to respond! if ircmsg.find("PING :") != -1: # if the server pings us then we've got to respond!
ping() ping()