From 06ef1410b136b12d69995f2e154cf86326712175 Mon Sep 17 00:00:00 2001 From: Russell Date: Wed, 21 Jan 2015 19:24:18 +0000 Subject: [PATCH] Added tildebot --- Code/irc/tildebot.py | 145 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 145 insertions(+) create mode 100755 Code/irc/tildebot.py diff --git a/Code/irc/tildebot.py b/Code/irc/tildebot.py new file mode 100755 index 0000000..13e87ca --- /dev/null +++ b/Code/irc/tildebot.py @@ -0,0 +1,145 @@ +#!/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 formatter +import get_users +import mentions +import pretty_date +import inflect + +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() + +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_prize(user): + if(random.randint(1,10) > 2): + prizes = [1] * 8 + [2] * 4 + [3] * 2 + [5] * 1 + prize = random.choice(prizes) + return [prize, user + " is " + ("super " if prize > 4 else "really " if prize > 2 else "") + "cool and gets " + p.number_to_words(prize) + " tildes!"] + else: + return [0, user + " is a meanie and gets no tildes!"] + +def give_tilde(channel, user, time): + found = False + with open("tildescores.txt", "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])): + ircsock.send("PRIVMSG " + channel + " :You have asked for a tilde too recently. Try again later.\n") + else: + prize = get_prize(user) + 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(user) + 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): + with open("tildescores.txt", "r") as scorefile: + for idx,score in enumerate(scorefile): + person = score.strip("\n").split("&^%") + if(person[0] == user): + ircsock.send("PRIVMSG " + channel + " :" + user + " has " + p.number_to_words(person[1]) + " tildes!\n") + return + #person has not played yet + ircsock.send("PRIVMSG " + channel + " :" + user + " has no tildes yet!\n") + +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) + +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) + ircmsg = ircmsg.strip('\n\r') + + if ircmsg.find("PING :") != -1: + ping() + + formatted = formatter.format_message(ircmsg) + + if "" == formatted: + continue + + # print formatted + + split = formatted.split("\t") + time = split[0] + user = split[1] + command = split[2] + channel = split[3] + messageText = split[4] + + if ircmsg.find(":!tildescore") != -1: + show_tildescore(channel, user) + elif ircmsg.find(":!tilde") != -1: + give_tilde(channel, user, time) + + if ircmsg.find(":!rollcall") != -1: + rollcall(channel) + + if ircmsg.find("PING :") != -1: + ping() + + sys.stdout.flush() + +ircsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) +connect(options.server, options.channel, options.nick) +listen()