#!/usr/bin/env python

from sentiwordnet import SentiWordNetCorpusReader

######################################################################

def senti_synset_viewer():
    swn = SentiWordNetCorpusReader('SentiWordNet_3.0.0_20100705.txt')
    for senti_synset in swn.all_senti_synsets():
        print senti_synset.synset.name, senti_synset.pos_score, senti_synset.neg_score

######################################################################

def sentiwordnet_to_csv():
    csvwriter = csv.writer(file('sentiwordnet-lemma-stats.csv', 'w'))
    csvwriter.writerow(["Word", "Tag", "PosScore", "NegScore"])
    for senti_synset in swn.all_senti_synsets():
        synset = senti_synset.synset
        tag = synset.pos
        if tag == 's':
            tag = 'a'
        pos_score = senti_synset.pos_score
        neg_score = senti_synset.neg_score        
        for lemma in senti_synset.synset.lemmas:
            row = [lemma.name, tag, pos_score, neg_score]
            csvwriter.writerow(row)
    print "Created CSV."