mirror of
https://github.com/RussellChamp/tilde-projects.git
synced 2026-10-28 00:22:34 +00:00
Modified some bots. Added madlib bot.
This commit is contained in:
1 parent
e9fc646207
commit
1671c4f3cb
19 files changed
+482
-25
No files matched your search
@@ -0,0 +1,2 @@
|
||||
I'm the {{a number}} {{a body part}}, {{another number}} {{another body part}} {{verb ending in -ing}} {{a color}} people eater!
|
||||
The End
|
||||
Executable
+110
@@ -0,0 +1,110 @@
|
||||
#!/usr/bin/python
|
||||
|
||||
import glob
|
||||
import random
|
||||
import re
|
||||
|
||||
file_pattern = "/home/*/madlibs/*.madlib"
|
||||
file_regex = re.compile(r'^/home/(.*?)/.*/([^/]*)\.madlib$')
|
||||
word_regex = re.compile(r'{{(.*?)}}')
|
||||
|
||||
# Take a file path and return a tuple with the title and original path
|
||||
def munge_story(file_name):
|
||||
match = re.match(file_regex, file_name)
|
||||
count = count_words_file(file_name)
|
||||
return ("'" + match.group(2).replace('_', ' ') + "' by ~" + match.group(1), match.group(0), count)
|
||||
|
||||
# Retrive a list of all madlib story files
|
||||
def find_stories(limit=999, shuffle=False):
|
||||
files = glob.glob(file_pattern)
|
||||
if shuffle == True:
|
||||
files = random.sample(files, max(1, min(limit, len(files))))
|
||||
else:
|
||||
files = files[:limit]
|
||||
return map(munge_story, files)
|
||||
|
||||
# Count the number of replacable words in the story
|
||||
def count_words(story):
|
||||
return len(word_regex.findall(story))
|
||||
|
||||
# Count the number of replacable words in the story when given a file path
|
||||
def count_words_file(storyPath):
|
||||
with open(storyPath, "r") as storyFile:
|
||||
story = storyFile.read()
|
||||
return count_words(story)
|
||||
|
||||
# Get the next replaceable word in the story. Includes full token and just the word
|
||||
# If you specify rand as True, it will get a random word instead of the next one
|
||||
def find_next_word(story, rand=False):
|
||||
matches = list(word_regex.finditer(story))
|
||||
if len(matches) == 0:
|
||||
return None
|
||||
match = matches[0]
|
||||
if rand is True:
|
||||
match = random.choice(matches)
|
||||
|
||||
return (match.group(0), match.group(1))
|
||||
|
||||
# Replace a word and return the entire story body
|
||||
def replace_word(story, query, word):
|
||||
return story.replace(query, word, 1)
|
||||
#return re.sub(query, word, story, 1)
|
||||
|
||||
# A helper function that will split the story up into chat-printable lengths
|
||||
def yield_lines(line, max_size):
|
||||
words = []
|
||||
words_length = 0
|
||||
# Add words to the words line untill we run out of space
|
||||
for word in line.split():
|
||||
# If the next word itself is longer thatn max_size, we'll have to chop it up
|
||||
while len(word) > max_size:
|
||||
splitsize = max_size - words_length - 1
|
||||
words.append(word[:splitsize] + "-")
|
||||
yield ' '.join(words)
|
||||
words = []
|
||||
words_length = 0
|
||||
word = word[splitsize:]
|
||||
if words_length + len(word) + 1 > max_size:
|
||||
yield ' '.join(words)
|
||||
words = []
|
||||
words_length = 0
|
||||
words.append(word)
|
||||
words_length += len(word) + 1 # For the space
|
||||
yield ' '.join(words) # For any words left over
|
||||
|
||||
# Open a story file and ask for the parts
|
||||
def make_story(storyFile):
|
||||
with open(storyFile, "r") as storyFile:
|
||||
story = storyFile.read()
|
||||
match = word_regex.search(story)
|
||||
while match is not None:
|
||||
word = raw_input('Give me {}: '.format(match.group(1)))
|
||||
print "Replacing '{}' with '{}'".format(match.group(0), word)
|
||||
print story
|
||||
story = re.sub(match.group(0), word, story, 1)
|
||||
match = word_regex.search(story)
|
||||
print story
|
||||
|
||||
def start():
|
||||
stories = find_stories(20, True)
|
||||
if len(stories) == 0:
|
||||
print "Sorry, no stories found. :/"
|
||||
return
|
||||
|
||||
input = -1
|
||||
while input < 0 or input >= len(stories):
|
||||
for idx, story in enumerate(stories):
|
||||
print "[{}] {}".format(idx, story[0])
|
||||
input = raw_input("Which story would you like?: ")
|
||||
try:
|
||||
# Try to convert the string input into a number
|
||||
input = int(input)
|
||||
except:
|
||||
# If they put something stupid in, treat it as a -1 and ask for
|
||||
# input again
|
||||
input = -1
|
||||
# Call make_story with the file path of the selected story
|
||||
make_story(stories[input][1])
|
||||
|
||||
if __name__ == "__main__":
|
||||
start()
|
||||
Binary file not shown.
Executable
+232
@@ -0,0 +1,232 @@
|
||||
#!/usr/bin/python
|
||||
# http://wiki.shellium.org/w/Writing_an_IRC_bot_in_Python
|
||||
|
||||
# Import some necessary libraries.
|
||||
import socket
|
||||
import os
|
||||
import sys
|
||||
from optparse import OptionParser
|
||||
import fileinput
|
||||
import random
|
||||
import time
|
||||
import re
|
||||
import operator
|
||||
|
||||
import msgformatter
|
||||
import madlib
|
||||
|
||||
class State:
|
||||
idle = 0 # When the bot is started and after it has finished a story
|
||||
thinking = 1 # I intentionally speed throttle this bot so it feels a little more natural
|
||||
story_selection = 2 # When the bot is waiting for a user input for story selection
|
||||
word_input = 3 # When the bot is waiting for user input after prompted for a word
|
||||
|
||||
# Globals
|
||||
MAX_LINE = 80 # The maximum number of characters in a line
|
||||
MAX_STORIES = 5 # The maximum number of stories a user can pick from
|
||||
THROTTLE_FACTOR = 1 # A factor by which all sleep event durations will be multiplied
|
||||
# Allow madlbibbot to run multiple simultaneous stories
|
||||
state = {} # The current state of the bot
|
||||
stories = {} # The list of stories available to users
|
||||
story = {} # The madlib story currently being worked on
|
||||
nextword = {} # The word that the bot is currently expecting data for
|
||||
|
||||
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='#madlibs',
|
||||
help="the channel to join", metavar="CHANNEL")
|
||||
parser.add_option("-n", "--nick", dest="nick", default='madlibbot',
|
||||
help="the nick to use", metavar="NICK")
|
||||
|
||||
(options, args) = parser.parse_args()
|
||||
|
||||
def resetGlobals(channel=""):
|
||||
global state
|
||||
global stories
|
||||
global story
|
||||
global nextword
|
||||
|
||||
if channel == "":
|
||||
state = {}
|
||||
stories = {}
|
||||
story = {}
|
||||
nextword = {}
|
||||
else:
|
||||
state[channel] = State.idle
|
||||
stories[channel] = []
|
||||
story[channel] = ""
|
||||
nextword[channel] = {}
|
||||
|
||||
def get_stories(channel):
|
||||
global state
|
||||
global stories
|
||||
state[channel] = State.thinking
|
||||
stories[channel] = madlib.find_stories(MAX_STORIES, True)
|
||||
if len(stories[channel]) == 0:
|
||||
sendmsg(channel, "Uh oh! There are no stories!")
|
||||
state[channel] = State.idle
|
||||
else:
|
||||
sendmsg(channel, "Here are a couple good stories:")
|
||||
time.sleep(1 * THROTTLE_FACTOR)
|
||||
for idx, story in enumerate(stories[channel]):
|
||||
sendmsg(channel, "[{}] {} ({} words)".format(idx, story[0], story[2]))
|
||||
time.sleep(0.5 * THROTTLE_FACTOR)
|
||||
sendmsg(channel, "Please select a story by index:")
|
||||
state[channel] = State.story_selection
|
||||
|
||||
# Handle user input when the bot is directly addressed
|
||||
def handle_bot_msg(channel, msg):
|
||||
global state
|
||||
global stories
|
||||
|
||||
if channel not in state:
|
||||
state[channel] = State.idle
|
||||
|
||||
saved_state = state[channel]
|
||||
state[channel] = State.thinking
|
||||
time.sleep(1 * THROTTLE_FACTOR)
|
||||
|
||||
# First check if we should quit the current game
|
||||
if saved_state != State.idle and msg == "!quit":
|
||||
quit_game(channel)
|
||||
elif saved_state == State.idle and msg == "startgame":
|
||||
get_stories(channel)
|
||||
elif saved_state == State.story_selection:
|
||||
handle_story_selection(channel, msg)
|
||||
elif saved_state == State.word_input:
|
||||
handle_story_step(channel, msg)
|
||||
else:
|
||||
state[channel] = State.idle
|
||||
|
||||
# Handle how to quit the game
|
||||
def quit_game(channel):
|
||||
resetGlobals(channel)
|
||||
sendmsg(channel, "Ok, quitting the current game.")
|
||||
|
||||
# Handle user input when we are in story selection mode
|
||||
def handle_story_selection(channel, msg):
|
||||
global stories
|
||||
global state
|
||||
try:
|
||||
imsg = int(msg)
|
||||
if imsg < 0 or imsg > len(stories[channel]):
|
||||
sendmsg(channel, "Selection out of bounds. Try again!")
|
||||
return
|
||||
time.sleep(1 * THROTTLE_FACTOR)
|
||||
sendmsg(channel, "Give me a second to load up {}".format(stories[channel][imsg][0]))
|
||||
|
||||
with open(stories[channel][imsg][1], "r") as storyFile:
|
||||
story[channel] = storyFile.read()
|
||||
stories[channel] = {} # Clear out the saved selectable stories in memory
|
||||
story_start(channel)
|
||||
except ValueError:
|
||||
sendmsg(channel, "Invalid selection. Try again!")
|
||||
state[channel] = State.story_selection
|
||||
|
||||
# Handle when a story is being started
|
||||
def story_start(channel):
|
||||
global story
|
||||
global state
|
||||
global nextword
|
||||
|
||||
state[channel] = State.thinking
|
||||
sendmsg(channel, "Alright! Let's get started!")
|
||||
nextword[channel] = madlib.find_next_word(story[channel], True)
|
||||
time.sleep(0.5 * THROTTLE_FACTOR)
|
||||
sendmsg(channel, "Give me {}:".format(nextword[channel][1]))
|
||||
state[channel] = State.word_input
|
||||
|
||||
# Handle user input when we have asked the user for input and are expecting a
|
||||
# response
|
||||
def handle_story_step(channel, msg):
|
||||
global state
|
||||
global story
|
||||
global nextword
|
||||
|
||||
state[channel] = State.thinking
|
||||
word = nextword[channel] #madlib.find_next_word(story[channel])
|
||||
if word is not None:
|
||||
story[channel] = madlib.replace_word(story[channel], nextword[channel][0], msg)
|
||||
time.sleep(1 * THROTTLE_FACTOR)
|
||||
|
||||
nextword[channel] = madlib.find_next_word(story[channel], True)
|
||||
if nextword[channel] is None:
|
||||
finish_story(channel)
|
||||
return
|
||||
#else
|
||||
count = madlib.count_words(story[channel])
|
||||
sendmsg(channel, "Thanks! Now give me {} ({} words left)".format(nextword[channel][1], count))
|
||||
state[channel] = State.word_input
|
||||
|
||||
# Finish the story
|
||||
def finish_story(channel):
|
||||
global state
|
||||
global story
|
||||
|
||||
sendmsg(channel, "Ok, here's the story...");
|
||||
sendmsg(channel, '=' * MAX_LINE)
|
||||
for line in story[channel].splitlines():
|
||||
for part in madlib.yield_lines(line, MAX_LINE):
|
||||
time.sleep(0.4 * THROTTLE_FACTOR)
|
||||
sendmsg(channel, part)
|
||||
padlen = (MAX_LINE - 9)/2
|
||||
mod = (MAX_LINE - 9) % 2
|
||||
sendmsg(channel, '=' * padlen + ' THE END ' + '=' * (padlen + mod))
|
||||
|
||||
story[channel] = ""
|
||||
state[channel] = State.idle
|
||||
|
||||
# System things
|
||||
def ping():
|
||||
ircsock.send("PONG :pingis\n")
|
||||
|
||||
def sendmsg(chan , msg):
|
||||
ircsock.send("PRIVMSG {} :{}\n".format(chan, msg))
|
||||
|
||||
def joinchan(chan):
|
||||
ircsock.send("JOIN {}\n".format(chan))
|
||||
|
||||
def rollcall(channel, botnick):
|
||||
sendmsg(channel, "Do you like MadLibs? Start a collaborative story by saying '{}: startgame'".format(botnick))
|
||||
|
||||
def connect(server, channel, botnick):
|
||||
ircsock.connect((server, 6667))
|
||||
ircsock.send("USER {} {} {} :krowbar\n".format(botnick, botnick, botnick)) # user authentication
|
||||
ircsock.send("NICK {}\n".format(botnick))
|
||||
joinchan(channel)
|
||||
|
||||
def listen(botnick):
|
||||
botmsgre = re.compile('^{}\:?\s*(.*)$'.format(botnick)) # re to strip the bot's name from a message
|
||||
while 1:
|
||||
ircmsg = ircsock.recv(2048)
|
||||
ircmsg = ircmsg.strip('\n\r')
|
||||
|
||||
if ircmsg.find("PING :") != -1:
|
||||
ping()
|
||||
|
||||
formatted = msgformatter.format_message(ircmsg)
|
||||
if "" == formatted:
|
||||
continue
|
||||
# print formatted
|
||||
|
||||
split = formatted.split("\t")
|
||||
msgtime = split[0]
|
||||
user = split[1]
|
||||
command = split[2]
|
||||
channel = split[3]
|
||||
message = split[4]
|
||||
|
||||
if message.startswith("!rollcall") == True or message.startswith("!help") == True:
|
||||
rollcall(channel, botnick)
|
||||
elif message.startswith(botnick) == True:
|
||||
botmsg = botmsgre.match(message).group(1)
|
||||
handle_bot_msg(channel, botmsg)
|
||||
|
||||
sys.stdout.flush()
|
||||
time.sleep(1 * THROTTLE_FACTOR)
|
||||
|
||||
ircsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
connect(options.server, options.channel, options.nick)
|
||||
listen(options.nick)
|
||||
@@ -0,0 +1 @@
|
||||
nohup: failed to run command './madlibbot/madlibbot.py': No such file or directory
|
||||
@@ -0,0 +1,16 @@
|
||||
import time
|
||||
import re
|
||||
|
||||
def format_message(message):
|
||||
pattern = r'^:.*\!~(.*)@.* (.*) (.*) :(.*)'
|
||||
now = int(time.time())
|
||||
matches = re.match(pattern, message)
|
||||
if not matches:
|
||||
return ''
|
||||
|
||||
nick = matches.group(1).strip()
|
||||
command = matches.group(2).strip()
|
||||
channel = matches.group(3).strip()
|
||||
message = matches.group(4).strip()
|
||||
|
||||
return "%s\t%s\t%s\t%s\t%s" % (now, nick, command, channel, message)
|
||||
@@ -0,0 +1,2 @@
|
||||
The {{an adjective}}, {{a color}} {{an animal}} jumped over the {{an adjective}} {{a color}} {{an animal}}.
|
||||
The End
|
||||
Reference in new issue
Block a user