First tilde projects commit

Code/irc/topicbot.py

@@ -0,0 +1,187 @@
+#!/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='topicbot',
+                  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 get_topic(channel, user, time):
+  #topic scores are saved as <USER>&^%<GETS SCORE>&^%<SETS SCORE>
+  with open("topicscores.txt", "r") as scorefile:
+    scores = scorefile.readlines()
+  userscore = 1
+  found = False
+  with open("topicscores.txt", "w") as scorefile:
+    for idx,score in enumerate(scores):
+      data = score.strip("\n").split("&^%")
+      if(data[0] == user):
+        found = True
+        userscore = int(data[1])+1
+        scores[idx] = data[0] + "&^%" + str(userscore) + "&^%" + data[2] + "\n"
+    scorefile.writelines(scores)
+    if(not found):
+      scorefile.write(user + "&^%1&^%0")
+
+
+  with open("topics_" + channel + ".txt", "r") as topics:
+    topic = topics.readlines()[-1].strip("\n").split("&^%", 3)
+    ircsock.send("PRIVMSG "+ channel +" :I've told you " + p.number_to_words(userscore) + " times! It's \"" + topic[2] + "\" (Set by " + topic[1] + " " + pretty_date.pretty_date(int(topic[0])) + ")\n")
+
+def count_topic(channel, user, time, msg):
+  with open("topics_" + channel + ".txt", "a") as topics:
+      topics.write(time + "&^%" + user + "&^%" + msg + "\n")
+  with open("topicscores.txt", "r") as scorefile:
+    scores = scorefile.readlines()
+  userscore = 1
+  found = False
+  with open("topicscores.txt", "w") as scorefile:
+    for idx,score in enumerate(scores):
+      data = score.strip("\n").split("&^%")
+      if(data[0] == user):
+        found = True
+        userscore = int(data[2])+1
+        scores[idx] = data[0] + "&^%" + data[1] + "&^%" + str(userscore) + "\n"
+    scorefile.writelines(scores)
+    if(not found):
+      scorefile.write(user + "&^%0&^%1")
+  ircsock.send("PRIVMSG "+ channel +" :" + user + " has changed the topic " + p.number_to_words(userscore) + " times!\n")
+
+def set_topic(channel, user, time, msg):
+  ircsock.send("TOPIC "+ channel +" :" + msg + "\n")
+  count_topic(channel, user, time, msg)
+
+def random_topic(channel, user, time):
+    with open("randomtopics.txt") as rtopics:
+      msg = random.choice(rtopics.readlines()).strip("\n")
+      set_topic(channel, user, time, msg)
+
+def rollcall(channel):
+  ircsock.send("PRIVMSG "+ channel +" :topicbot reporting! I respond to !topic !settopic !randomtopic !thistory\n")
+
+def topic_score():
+    ircsock.send("PRIVMSG "+ channel +" :Not implemented yet")
+
+def topic_scores():
+    ircsock.send("PRIVMSG "+ channel +" :Not implemented yet")
+
+def topic_history(channel, user, count):
+  try:
+    iCount = int(count.split()[1])
+  except ValueError:
+    iCount = 3
+  if(iCount > 10):
+    iCount = 10
+  if(iCount < 1):
+    iCount = 3
+  with open("topics_" + channel + ".txt", "r") as topicsfile:
+    #topics = topicsfile.readlines()[-iCount:].reverse()
+    ircsock.send("PRIVMSG "+ channel +" :Ok, here were the last " + p.number_to_words(iCount) + " topics\n")
+    for idx,topic in enumerate(reversed(topicsfile.readlines()[-iCount:])):
+      topic = topic.strip("\n").split("&^%", 3)
+      ircsock.send("PRIVMSG "+ channel +" :" + str(idx+1) + ": \"" + topic[2] + "\" (Set by " + topic[1] + " " + pretty_date.pretty_date(int(topic[0])) + ")\n")
+
+
+def connect(server, channel, botnick):
+  ircsock.connect((server, 6667))
+  ircsock.send("USER "+ botnick +" "+ botnick +" "+ botnick +" :This bot is a result of a tutoral covered on http://shellium.org/wiki.\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(command == "TOPIC" and user != options.nick):
+      count_topic(channel,user, time, messageText)
+
+    if ircmsg.find(":!topic") != -1:
+      get_topic(channel, user, time)
+
+    if ircmsg.find(":!settopic") != -1:
+      set_topic(channel, user, time, messageText[10:])
+
+    if ircmsg.find(":!tscores") != -1:
+      topic_scores()
+    elif ircmsg.find(":!tscores") != -1:
+      topic_score()
+
+    if ircmsg.find(":!randomtopic") != -1:
+      random_topic(channel, user, time)
+
+    if ircmsg.find(":!thistory") != -1:
+      topic_history(channel, user, messageText)
+
+    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()