SentimentTest.py

from __future__ import print_function
import sys, json, requests
from textblob import TextBlob
from colorama import *
init()

def areyouhappy(username=None, depth=None):
    if username is None:
        username = "presidentobama"
    if depth is None:
        depth = 10
    else: depth = int(depth)

    url = r"http://www.reddit.com/user/%s/comments.json?limit=100" % username
    totalsent=0
    postcount=0
    comment_max = 500
    label = "*"*22
    pos=[]
    neg=[]

    r = requests.get(url)
    data = json.loads(r.text)
    posts = data['data']['children']
    for idx, c in enumerate(posts):
        c = c['data']
        try:
            # TODO: Check 'body_html' for <pre> tags to skip code...
            if len(c['body']) < comment_max:
                sentence = TextBlob(c['body'])
                if sentence.sentiment[0] > 0:
                    pos.append((sentence.string, sentence.sentiment[0]))
                else:
                    neg.append((sentence.string, sentence.sentiment[0]))
                totalsent += sentence.sentiment[0]
                postcount += 1
        except:
            pass

    psort = sorted(pos, key=lambda x: x[-1])
    pos_sort = sorted(psort, reverse=True)

    nsort = sorted(neg, key=lambda x: x[-1])
    neg_sort = sorted(nsort) 

    print(Fore.GREEN + label,"Top Positive Comments:", label, sep="\n")
    for idx, x in enumerate(pos_sort):
        if idx < depth:
            print(Fore.RESET + u'{0}: "{1}"'.format(idx+1, pos_sort[idx][0]).encode('utf-8', 'ignore'), end="\n\n")
    print(Fore.RED + label,"Most Negative Comments:", label, sep="\n")
    for idx, x in enumerate(neg_sort):
        if idx < depth:
            print(Fore.RESET + u'{0}: "{1}"'.format(idx+1, neg_sort[idx][0]).encode('utf-8', 'ignore'), end="\n\n")

    overall = totalsent/postcount
    if overall > 0:
        print("\n\n\nYou're happy overall though, your overall positivity is {0}".format(overall))
    else:
        print("\n\n\nYou're a candidate for the playahatas ball - your overall negativity is {0}".format(overall))

try:
    areyouhappy(username=sys.argv[1])
except:
    areyouhappy()