mirror of
https://github.com/jlu5/PyLink.git
synced 2024-12-25 04:02:45 +01:00
ed34c43b6c
AFAIK, random already defaults the seed to the system time, and random.seed() without any arguments thus sets it to something that is already set.
131 lines
3.8 KiB
Python
131 lines
3.8 KiB
Python
"""
|
|
games.py: Create a bot that provides game functionality (dice, 8ball, etc).
|
|
"""
|
|
|
|
import sys
|
|
import os
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
import random
|
|
import urllib.request
|
|
import urllib.error
|
|
from xml.etree import ElementTree
|
|
|
|
import utils
|
|
from log import log
|
|
import world
|
|
|
|
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')
|