225 lines
9.1 KiB
Python
Executable File
225 lines
9.1 KiB
Python
Executable File
#!/usr/bin/python
|
|
# http://wiki.shellium.org/w/Writing_an_IRC_bot_in_Python
|
|
|
|
# Import some necessary libraries.
|
|
import socket
|
|
import os
|
|
import sys
|
|
import time
|
|
from optparse import OptionParser
|
|
import fileinput
|
|
import random
|
|
|
|
import formatter
|
|
import get_users
|
|
import names
|
|
import mentions
|
|
import pretty_date
|
|
import inflect
|
|
import puzzle
|
|
|
|
parser = OptionParser()
|
|
|
|
parser.add_option("-s", "--server", dest="server", default='127.0.0.1',
|
|
help="the server to connect to", metavar="SERVER")
|
|
parser.add_option("-c", "--channel", dest="channel", default='#bot_test',
|
|
help="the channel to join", metavar="CHANNEL")
|
|
parser.add_option("-n", "--nick", dest="nick", default='tildebot',
|
|
help="the nick to use", metavar="NICK")
|
|
|
|
(options, args) = parser.parse_args()
|
|
|
|
p = inflect.engine()
|
|
challenges = {}
|
|
SCORE_FILE = "tildescores.txt"
|
|
JACKPOT_FILE = "tildejackpot.txt"
|
|
JACKPOT_MIN = 3
|
|
DEBUG = False
|
|
|
|
def ping():
|
|
ircsock.send("PONG :pingis\n")
|
|
|
|
def sendmsg(chan , msg):
|
|
ircsock.send("PRIVMSG "+ chan +" :"+ msg +"\n")
|
|
|
|
def joinchan(chan):
|
|
ircsock.send("JOIN "+ chan +"\n")
|
|
|
|
def hello():
|
|
ircsock.send("PRIVMSG "+ channel +" :Hello!\n")
|
|
|
|
def too_recent(time1, time2):
|
|
if(int(time1) - int(time2) < 60*60):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def get_positive():
|
|
return random.choice(['Yes', 'Yep', 'Yeppers', 'Correct','You got it', 'Yeah', 'Right on', 'Uh-huh', 'Positive', 'Totally right', 'Close enough', 'That\'s it'])
|
|
|
|
def get_negative():
|
|
return random.choice(['No', 'Nope', 'Sorry', 'Wrong', 'Nuh-uh', 'Negatory', 'Incorrect', 'Not today', 'Try again', 'Maybe later', 'Probably not', 'Answer hazy', 'Not quite',\
|
|
'Not even close'])
|
|
|
|
def get_superlative(score):
|
|
if score > 4:
|
|
return random.choice(["super cool", "totally rad", "extraordinary", "dynomite", "#topdrawer", "a #TopLad", "the cat's meow", "a tilde town hero", "my favorite person", "incredibly lucky",\
|
|
"unbelievable", "a tilde town hunk", "could bring all the boys to the yard", "worth twice their weight in gold", "the hero we need", "no ordinary townie"]);
|
|
elif score > 2:
|
|
return random.choice(["really cool", "pretty neat", "rather nice", "a dynamic doggo", "radical", "intense", "pretty lucky", "knows the territory", "has what it takes", "has mad skillz",\
|
|
"going the distance", "a hard worker", "my sunshine", "ready to rumble"]);
|
|
else:
|
|
return random.choice(["cool", "nice", "acceptible", "good enough", "a promising pupper", "better than a horse", "swell", "a little lucky", "just credible", "my friend", "probably not a robot",\
|
|
"valuable to the team"]);
|
|
|
|
def get_bad_thing():
|
|
return random.choice(["is a meanie", "mugs me right off", "is worse than a horse", "smells like a ghost", "probably didn't bathe today", "didn't guess hard enough", "isn't lucky",\
|
|
"smells of elderberries", "should reconsider their life choices", "did't believe in the heart of the tilde", "came to the wrong chat channel", "should have stopped while they were ahead",\
|
|
"requires annotations from an authoratative source", "could have been a contender", "spreads vicious rumors", "drank my milkshake", "is probably cheating", "is trying too hard"]);
|
|
|
|
def get_prize(name, isHuman, bonus=0):
|
|
prizes = [1] * 8 + [2] * 4 + [3] * 2 + [5] * isHuman #no 5pt prize for non-humans
|
|
prize = random.choice(prizes) + bonus
|
|
if(random.randint(1,10) > 6 - 4 * isHuman): #80% of the time it's a normal prize (40% for not humans)
|
|
return [prize, name + ": " + (get_positive() if isHuman else get_negative()) + "! You are " + get_superlative(prize) + " and get " + p.number_to_words(prize) + " tildes!"]
|
|
else: #20% of the time its a jackpot situation
|
|
with open(JACKPOT_FILE, "r+") as jackpotfile:
|
|
jackpot = int(jackpotfile.readline().strip("\n"))
|
|
jackpotfile.seek(0)
|
|
jackpotfile.truncate()
|
|
if(random.randint(1,10) > 1 or not isHuman): #90% of the time it's a non-prize. non-humans never win jackpot
|
|
new_jackpot = jackpot+max(1,prize)
|
|
jackpotfile.write(str(new_jackpot)) #increase the jackpot by the prize size
|
|
return [0, name + " " + get_bad_thing() + " and gets no tildes! (Jackpot is now " + str(new_jackpot) + " tildes)"]
|
|
else: #hit jackpot!
|
|
jackpotfile.write(str(JACKPOT_MIN))
|
|
return [jackpot, name + " hit the jackpot and won **" + p.number_to_words(jackpot) + " tildes!**"]
|
|
|
|
def show_jackpot(channel):
|
|
with open(JACKPOT_FILE, "r") as jackpotfile:
|
|
jackpot = int(jackpotfile.readline().strip("\n"))
|
|
ircsock.send("PRiVMSG " + channel + " :The jackpot is currently " + p.number_to_words(jackpot) + " tildes!\n")
|
|
|
|
def give_tilde(channel, user, name, time, human, bonus=0):
|
|
found = False
|
|
with open(SCORE_FILE, "r+") as scorefile:
|
|
scores = scorefile.readlines()
|
|
scorefile.seek(0)
|
|
scorefile.truncate()
|
|
for score in scores:
|
|
person = score.strip("\n").split("&^%")
|
|
if(person[0] == user):
|
|
found = True
|
|
if(too_recent(time, person[2]) and not DEBUG):
|
|
ircsock.send("PRIVMSG " + channel + " :" + name + " asked for a tilde too recently and " + get_bad_thing() + ". Try again later.\n")
|
|
else:
|
|
prize = get_prize(name, human, bonus)
|
|
score = person[0] + "&^%" + str(int(person[1]) + prize[0]) + "&^%" + time + "\n"
|
|
ircsock.send("PRIVMSG " + channel + " :" + prize[1] + "\n")
|
|
scorefile.write(score)
|
|
if(not found):
|
|
prize = get_prize(name, True, bonus)
|
|
ircsock.send("PRIVMSG " + channel + " :Welcome to the tilde game! Here's " + p.number_to_words(prize[0]+1) + " free tilde(s) to start you off.\n")
|
|
scorefile.write(user + "&^%" + str(prize[0]+1) + "&^%" + time + "\n")
|
|
|
|
def show_tildescore(channel, user, name):
|
|
with open(SCORE_FILE, "r") as scorefile:
|
|
for idx,score in enumerate(scorefile):
|
|
person = score.strip("\n").split("&^%")
|
|
if(person[0] == user):
|
|
ircsock.send("PRIVMSG " + channel + " :" + name + " has " + p.number_to_words(person[1]) + " tildes!\n")
|
|
return
|
|
#person has not played yet
|
|
ircsock.send("PRIVMSG " + channel + " :" + name + " has no tildes yet!\n")
|
|
|
|
def challenge(channel, user, name, time):
|
|
if(channel != "#bots" and not DEBUG):
|
|
ircsock.send("PRIVMSG " + channel + " :" + name + " is a meanie and gets no tildes. **Tildebot now only gives out tildes in the #bots channel.**\n")
|
|
return
|
|
global challenges;
|
|
challenge = puzzle.make_puzzle();
|
|
challenges[user] = challenge[1:]; #challenges[USER] = [ANSWER, BONUS]
|
|
ircsock.send("PRIVMSG " + channel + " :" + name + ": " + challenge[0] + "\n")
|
|
|
|
def challenge_response(channel, user, name, time, msg):
|
|
global challenges
|
|
print(msg);
|
|
if(challenges.has_key(user)):
|
|
answer, bonus = challenges[user]
|
|
if((callable(answer) and answer(msg.lower())) or (msg.lower() == str(answer).lower() or msg == p.number_to_words(answer))):
|
|
give_tilde(channel, user, name, time, True, bonus);
|
|
else:
|
|
give_tilde(channel, user, name, time, False, 0);
|
|
del challenges[user]; #delete the user from challenges either way
|
|
|
|
|
|
|
|
def rollcall(channel):
|
|
ircsock.send("PRIVMSG "+ channel +" :tildebot reporting! I respond to !tilde !tildescore\n")
|
|
|
|
def connect(server, channel, botnick):
|
|
ircsock.connect((server, 6667))
|
|
ircsock.send("USER "+ botnick +" "+ botnick +" "+ botnick +" :krowbar\n") # user authentication
|
|
ircsock.send("NICK "+ botnick +"\n")
|
|
|
|
joinchan(channel)
|
|
if(not DEBUG):
|
|
joinchan("#bots")
|
|
|
|
def get_user_from_message(msg):
|
|
try:
|
|
i1 = msg.index(':') + 1
|
|
i2 = msg.index('!')
|
|
return msg[i1:i2]
|
|
except ValueError:
|
|
return ""
|
|
|
|
def listen():
|
|
while 1:
|
|
|
|
ircmsg = ircsock.recv(2048)
|
|
for msg in ircmsg.split('\n'):
|
|
msg = msg.strip('\n\r')
|
|
|
|
if msg.find("PING :") != -1:
|
|
ping()
|
|
|
|
formatted = formatter.format_message(msg)
|
|
|
|
if "" == formatted:
|
|
continue
|
|
|
|
# print formatted
|
|
|
|
split = formatted.split("\t")
|
|
iTime = split[0]
|
|
user = split[1]
|
|
name = names.get_name(user)
|
|
command = split[2]
|
|
channel = split[3]
|
|
messageText = split[4]
|
|
|
|
if msg.find(":!tildescore") != -1:
|
|
show_tildescore(channel, user, name)
|
|
elif msg.find(":!tilde") != -1 and not challenges.has_key(user):
|
|
challenge(channel, user, name, iTime)
|
|
elif challenges.has_key(user) and (channel == "#bots" or DEBUG):
|
|
challenge_response(channel, user, name, iTime, messageText)
|
|
#give_tilde(channel, user, name, iTime)
|
|
|
|
if msg.find(":!jackpot") != -1:
|
|
show_jackpot(channel)
|
|
|
|
if msg.find(":!rollcall") != -1:
|
|
rollcall(channel)
|
|
|
|
if msg.find("PING :") != -1:
|
|
ping()
|
|
|
|
sys.stdout.flush()
|
|
time.sleep(1)
|
|
|
|
ircsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
connect(options.server, options.channel, options.nick)
|
|
listen()
|