Added some more scripts

Code/python/chatcloud2.py

@@ -0,0 +1,68 @@
+#!/usr/bin/python
+import fileinput
+import json
+import time
+import calendar
+import re
+import shutil
+import argparse
+import logging, sys
+
+logging.basicConfig(stream=sys.stderr, level=logging.DEBUG)
+
+parser = argparse.ArgumentParser(description='Generate word cloud data based off of irc chat logs')
+parser.add_argument('-logfile', help='irc log file to read from', default='/home/jumblesale/Code/irc/log')
+parser.add_argument('-outfile', help='output file to write to', default='')
+
+parser.add_argument('-timeend', type=int, help='end time of the word cloud (in epoch time)', default=calendar.timegm(time.gmtime()))
+parser.add_argument('-timestart', type=int, help='start time of the word cloud (in epoch time)', default=-1)
+
+parser.add_argument('-bannedUsersFile', help='file containing list of banned users', default='/home/krowbar/Code/python/bannedUsers')
+parser.add_argument('-bannedWordsFile', help='file containing list of banned words', default='/home/krowbar/Code/python/bannedWords')
+
+parser.add_argument('-minLength', type=int, help='minimum size of words to include in the cloud', default=3)
+parser.add_argument('-minOccurrence', type=int, help='the minimum occurence of a word to include it in the cloud', default=3)
+
+args = parser.parse_args()
+
+wordData = {} # keyed by "word" that contains a count
+#we only care about recent chats, let's say for the past sixteen hours
+
+args.timestart = args.timestart if args.timestart != -1 else args.timeend - (16 * 60 * 60)
+#timeCutoff = calendar.timegm(time.strptime("1 Oct 16", "%d %b %y"))
+logging.info("Generating word cloud based off words from " + str(args.timestart) + " to " + str(args.timeend))
+
+bannedWords = open(args.bannedWordsFile).read().splitlines()
+bannedUsers = open(args.bannedUsersFile).read().splitlines()
+
+with open(args.logfile, "r") as log:
+    for line in log:
+        try:
+            time, user, message = line.split("\t", 3)
+            time = int(time)
+        except ValueError:
+            continue #There are some bad lines in the log file that we'll ignore if we can't parse
+        if user in bannedUsers:
+            continue #We don't care what they say
+        if time >= args.timestart and time <= args.timeend:
+            #print "Processing line from " + user + " at " + str(time)
+            for word in re.sub('[\'\"\`\/\\;:,.?!*&^\-()<>\{\}|_\[\]0-9]', ' ', message).lower().split():
+                #changing symbols into spaces instead of stripping them avoids compounded words
+                if len(word) < args.minLength or word in bannedWords:
+                    #print "Rejecting " + word
+                    continue
+                #if the word already exists in the list
+                if word in wordData:
+                    wordData[word] += 1
+                else: #if they are new
+                    wordData[word] = 1
+                    #print "Added word: " + word
+wordData = {i:wordData[i] for i in wordData if wordData[i] >= args.minOccurrence }
+if len(wordData) == 0:
+    wordData = {"NOTHING": 1, "INTERESTING": 1, "TODAY": 1}
+if args.outfile == '':
+    print json.dumps(wordData)
+else:
+    with open(args.outfile + ".tmp", "w") as tmpFile:
+        tmpFile.write(json.dumps(wordData))
+    shutil.move(args.outfile + ".tmp", args.outfile)