mirror of
https://github.com/jlu5/PyLink.git
synced 2024-12-30 22:52:34 +01:00
126 lines
3.7 KiB
Python
126 lines
3.7 KiB
Python
"""
|
|
games.py: Create a bot that provides game functionality (dice, 8ball, etc).
|
|
"""
|
|
import random
|
|
import urllib.request
|
|
import urllib.error
|
|
from xml.etree import ElementTree
|
|
|
|
from pylinkirc import utils
|
|
from pylinkirc.log import log
|
|
|
|
gameclient = utils.registerService("Games", manipulatable=True)
|
|
reply = gameclient.reply # TODO find a better syntax for ServiceBot.reply()
|
|
|
|
# commands
|
|
def dice(irc, source, args):
|
|
"""<num>d<sides>
|
|
|
|
Rolls a die with <sides> sides <num> times.
|
|
"""
|
|
if not args:
|
|
reply(irc, "No string given.")
|
|
return
|
|
|
|
try:
|
|
# Split num and sides and convert them to int.
|
|
num, sides = map(int, args[0].split('d', 1))
|
|
except ValueError:
|
|
# Invalid syntax. Show the command help.
|
|
gameclient.help(irc, source, ['dice'])
|
|
return
|
|
|
|
assert 1 < sides <= 100, "Invalid side count (must be 2-100)."
|
|
assert 1 <= num <= 100, "Cannot roll more than 100 dice at once."
|
|
|
|
results = []
|
|
for _ in range(num):
|
|
results.append(random.randint(1, sides))
|
|
|
|
# Convert results to strings, join them, format, and reply.
|
|
s = 'You rolled %s: %s (total: %s)' % (args[0], ' '.join([str(x) for x in results]), sum(results))
|
|
reply(irc, s)
|
|
|
|
gameclient.add_cmd(dice, 'd')
|
|
gameclient.add_cmd(dice)
|
|
|
|
eightball_responses = ["It is certain.",
|
|
"It is decidedly so.",
|
|
"Without a doubt.",
|
|
"Yes, definitely.",
|
|
"You may rely on it.",
|
|
"As I see it, yes.",
|
|
"Most likely.",
|
|
"Outlook good.",
|
|
"Yes.",
|
|
"Signs point to yes.",
|
|
"Reply hazy, try again.",
|
|
"Ask again later.",
|
|
"Better not tell you now.",
|
|
"Cannot predict now.",
|
|
"Concentrate and ask again.",
|
|
"Don't count on it.",
|
|
"My reply is no.",
|
|
"My sources say no.",
|
|
"Outlook not so good.",
|
|
"Very doubtful."]
|
|
def eightball(irc, source, args):
|
|
"""[<question>]
|
|
|
|
Asks the Magic 8-ball a question.
|
|
"""
|
|
reply(irc, random.choice(eightball_responses))
|
|
gameclient.add_cmd(eightball)
|
|
gameclient.add_cmd(eightball, '8ball')
|
|
gameclient.add_cmd(eightball, '8b')
|
|
|
|
def fml(irc, source, args):
|
|
"""[<id>]
|
|
|
|
Displays an entry from fmylife.com. If <id>
|
|
is not given, fetch a random entry from the API."""
|
|
try:
|
|
query = args[0]
|
|
except IndexError:
|
|
# Get a random FML from the API.
|
|
query = 'random'
|
|
|
|
# TODO: configurable language?
|
|
url = ('http://api.betacie.com/view/%s/nocomment'
|
|
'?key=4be9c43fc03fe&language=en' % query)
|
|
try:
|
|
data = urllib.request.urlopen(url).read()
|
|
except urllib.error as e:
|
|
reply(irc, 'Error: %s' % e)
|
|
return
|
|
|
|
tree = ElementTree.fromstring(data.decode('utf-8'))
|
|
tree = tree.find('items/item')
|
|
|
|
try:
|
|
category = tree.find('category').text
|
|
text = tree.find('text').text
|
|
fmlid = tree.attrib['id']
|
|
url = tree.find('short_url').text
|
|
except AttributeError as e:
|
|
log.debug("games.FML: Error fetching FML %s from URL %s: %s",
|
|
query, url, e)
|
|
reply(irc, "Error: That FML does not exist or there was an error "
|
|
"fetching data from the API.")
|
|
return
|
|
|
|
if not fmlid:
|
|
reply(irc, "Error: That FML does not exist.")
|
|
return
|
|
|
|
# TODO: customizable formatting
|
|
votes = "\x02[Agreed: %s / Deserved: %s]\x02" % \
|
|
(tree.find('agree').text, tree.find('deserved').text)
|
|
s = '\x02#%s [%s]\x02: %s - %s \x02<\x0311%s\x03>\x02' % \
|
|
(fmlid, category, text, votes, url)
|
|
reply(irc, s)
|
|
gameclient.add_cmd(fml)
|
|
|
|
def die(irc):
|
|
utils.unregisterService('games')
|