tweet_sentiment.py

import sys
import json

def hw():
    print 'Hello, world!'

def lines(fp):
    print str(len(fp.readlines()))

def main():
    sent_file = open(sys.argv[1])
    tweet_file = open(sys.argv[2])
    hw()
    lines(sent_file)
    lines(tweet_file)


def make_afinn_dict(afinn_file):
    with open(afinn_file) as f:
        # lines -- list of strings
        lines = f.readlines()
    linecells = [convert_line(line) for line in lines]
    return dict(linecells)


def convert_line(line):
    word, score_str = line.rstrip().split('\t')
    score = int(score_str)
    return [word, score]


def make_tweet_json(input_file):
    with open(input_file) as f:
        lines = f.readlines()
    return [make_tweet(line) for line in lines]
    

def make_tweet(line):
    return json.loads(line)

def calc_score(afinn_dict, text):
    words = text.split()
    words_scores = [afinn_dict[word] if afinn_dict.has_key(word) else 0 for word in words]
    return sum(words_scores)

if __name__ == '__main__':
    afinn_file = sys.argv[1]
    input_file = sys.argv[2]
    afinn_dict = make_afinn_dict(afinn_file)
    tweets = make_tweet_json(input_file)
    scores = [calc_score(afinn_dict, tweet['text']) if tweet.has_key('text') else 0 for tweet in tweets]
    for score in scores:
        print score