mirror of
https://github.com/RussellChamp/tilde-projects.git
synced 2026-10-28 08:32:32 +00:00
Since this means there's multiple possible answers, I made it so that the answer can now be a function, and that gets run with the guess. This may open the door to other puzzle ideas that have been ignored due to having more than one correct answer.
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()
|