tilde-projects/Code/irc/wangbot.py

#!/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 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="numberwang_bot",
    help="the nick to use",
    metavar="NICK",
)

(options, args) = parser.parse_args()

p = inflect.engine()
LIMIT_GUESSING = True
MIN_ROUNDS = 5
MAX_ROUNDS = 12
SCORE_FILE = "numberwangscores.txt"
SHOW_TOP_NUM = 5
GOOD_CHAN = "#bots"

roundsLeft = 0
bonusRound = 0
guesses = 0
lastGuesser = ""
currentScores = {}


def resetGlobals():
    global roundsLeft
    global bonusRound
    global guesses
    global lastGuesser
    global currentScores
    roundsLeft = 0
    bonusRound = 0
    guesses = 0
    lastGuesser = ""
    currentScores.clear()


def ping(pong):
    ircsock.send("PONG {}\n".format(pong))


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 start_numberwang(channel, user):
    if channel != "#bots":
        ircsock.send(
            "PRIVMSG "
            + channel
            + " :Numberwang has been disabled for "
            + channel
            + " due to spamminess. Please join "
            + GOOD_CHAN
            + " to start a game.\n"
        )
        return

    print(user + " started a game")
    resetGlobals()
    ircsock.send("PRIVMSG " + channel + " :It's time for Numberwang!\n")
    time.sleep(1)
    ircsock.send("PRIVMSG " + channel + " :Here's how to play:\n")

    ircsock.send("PRIVMSG " + channel + " :1. There are 10 rounds\n")
    ircsock.send(
        "PRIVMSG "
        + channel
        + " :2. Each round lasts 10 seconds. You're up against the clock!\n"
    )
    ircsock.send(
        "PRIVMSG "
        + channel
        + " :3. Play your numbers, as long as they're between 0 and 99.\n"
    )
    ircsock.send("PRIVMSG " + channel + " :4. That's Numberwang!\n")
    time.sleep(2)
    ircsock.send("PRIVMSG " + channel + " :Let's get started!\n")
    global roundsLeft
    global bonusRound
    roundsLeft = random.randint(MIN_ROUNDS, MAX_ROUNDS)
    bonusRound = random.randint(2, roundsLeft - 1)
    print(
        "There will be "
        + str(roundsLeft)
        + " rounds with the bonus on round "
        + str(roundsLeft - bonusRound + 1)
    )


def print_scores(channel):
    scoreStrs = []
    first = True
    for name in currentScores:
        scoreStrs.append(
            name
            + " is "
            + ("also " if not first and random.randint(1, 3) == 3 else "")
            + "on "
            + str(currentScores[name])
        )
        first = False
    ircsock.send("PRIVMSG " + channel + " :" + p.join(scoreStrs) + "!\n")


def guess_numberwang(channel, user, messageText):
    global guesses
    global lastGuesser
    global currentScores
    global roundsLeft
    print(user + " guessed '" + messageText + "'")
    guess = re.sub(
        "[^0-9]", "", messageText.split()[0]
    )  # must have a number in the first 'word'
    if guess:
        if LIMIT_GUESSING and user == lastGuesser:
            ircsock.send(
                "PRIVMSG "
                + channel
                + " :"
                + user
                + ", you just guessed! Give another player a try!\n"
            )
        else:
            guesses += 1
            lastGuesser = user
            ###CORRECT GUESS###
            if (
                random.randint(0, 10) > 10 - guesses
            ):  # the more guesses, the higher the probability
                guesses = 0
                lastGuesser = ""
                ircsock.send(
                    "PRIVMSG " + channel + " :" + user + ": THAT'S NUMBERWANG!\n"
                )
                points = random.randint(2, 10) * (
                    random.randint(2, 4) if roundsLeft == bonusRound else 1
                )
                if user in currentScores.keys():
                    currentScores[user] += points
                else:
                    currentScores[user] = points
                roundsLeft -= 1
                time.sleep(2)
                if roundsLeft == 0:
                    ircsock.send(
                        "PRIVMSG "
                        + channel
                        + " :Numberwang is now over. Thank you for playing!\n"
                    )
                    ircsock.send("PRIVMSG " + channel + " :Final scores:\n")
                    print_scores(channel)
                    save_scores()
                else:
                    print_scores(channel)
                    newRoundStr = ""
                    if roundsLeft == 1:
                        newRoundStr += "The last round is Wangernumb!"
                    elif roundsLeft == bonusRound:
                        newRoundStr += "**Bonus Round!**"
                    else:
                        newRoundStr += "New Round!"
                    if random.randint(1, 10) > 8:
                        newRoundStr += " Let's rotate the board!"
                    ircsock.send(
                        "PRIVMSG " + channel + " :" + newRoundStr + " Start guessing!\n"
                    )

            ###INCORRECT GUESS###
            else:
                ircsock.send(
                    "PRIVMSG "
                    + channel
                    + " :"
                    + random.choice(["Sorry", "I'm sorry", "No", "Nope"])
                    + ", "
                    + user
                    + ", "
                    + random.choice(
                        [
                            "that's not",
                            "that is not",
                            "that isn't",
                            "that is not",
                            "that won't make",
                            "that will not make",
                        ]
                    )
                    + " Numberwang!\n"
                )


def stop_numberwang(channel, user):
    print(user + " stopped a game")
    resetGlobals()
    ircsock.send(
        "PRIVMSG "
        + channel
        + " :Numberwang has been stopped. No points have been awarded. "
        + user
        + " is such a party pooper!\n"
    )


def save_scores():
    with open(SCORE_FILE, "r+") as scorefile:
        scores = scorefile.readlines()
        scorefile.seek(0)
        scorefile.truncate()
        for line in scores:
            for name in currentScores:
                score = line.strip("\n").split("&^%")
                if score[0] == name:
                    line = (
                        score[0]
                        + "&^%"
                        + str(int(score[1]) + currentScores[name])
                        + "\n"
                    )
                    del currentScores[name]
                    break
            scorefile.write(line)

        for name in currentScores:  # new wangers
            line = name + "&^%" + str(currentScores[name]) + "\n"
            scorefile.write(line)


def show_highscores(channel):
    with open(SCORE_FILE, "r") as scorefile:
        scores = []
        for line in scorefile.readlines():
            sline = line.strip("\n").split("&^%")
            scores.append((int(sline[1]), sline[0]))
        scores = sorted(scores, reverse=True)[:SHOW_TOP_NUM]

        ircsock.send("PRIVMSG " + channel + " :   ====TOP WANGERS====\n")
        for score in scores:
            ircsock.send(
                "PRIVMSG "
                + channel
                + " :== ~"
                + score[1]
                + " ("
                + str(score[0])
                + " points!) ==\n"
            )


def show_user_score(channel, user):
    with open(SCORE_FILE, "r") as scorefile:
        for line in scorefile.readlines():
            score = line.strip("\n").split("&^%")
            if user == score[0]:
                ircsock.send(
                    "PRIVMSG "
                    + channel
                    + " :"
                    + user
                    + ": Your global numberwang score is "
                    + str(score[1])
                    + "!\n"
                )
                return
        # if we don't find a score line
        ircsock.send(
            "PRIVMSG "
            + channel
            + " :"
            + user
            + ": You haven't scored any points yet!\n"
        )


def rollcall(channel):
    ircsock.send(
        "PRIVMSG "
        + channel
        + " :Is it time for Numberwang? It might be! Start a new game with !numberwang or stop a current game with !wangernumb Get your score with !myscore and the list of top wangers with !topwangers\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)
    joinchan(GOOD_CHAN)


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[:4] == "PING":
            ping(ircmsg.split(" ")[1])

        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(":!numberwang") != -1 and roundsLeft == 0:
            start_numberwang(channel, user)

        if channel == GOOD_CHAN:
            if ircmsg.find(":!wangernumb") != -1 and roundsLeft > 0:
                stop_numberwang(channel, user)
            if roundsLeft > 0:
                guess_numberwang(channel, user, messageText)

        if ircmsg.find(":!topwangers") != -1:
            show_highscores(channel)
        if ircmsg.find(":!myscore") != -1:
            show_user_score(channel, user)

        if ircmsg.find(":!rollcall") != -1:
            rollcall(channel)

        if ircmsg[:4] == "PING":
            ping(ircmsg.split(" ")[1])

        sys.stdout.flush()
        time.sleep(1)


ircsock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
connect(options.server, options.channel, options.nick)
listen()