DragonOsman

Sentiments tweets

Mar 5th, 2017
276
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.98 KB | None | 0 0
  1. #!/usr/bin/env python3
  2.  
  3. import os
  4. import sys
  5. import nltk
  6.  
  7. from analyzer import Analyzer
  8. from termcolor import colored
  9. from helpers import get_user_timeline
  10.  
  11. def main():
  12.     if len(sys.argv) != 2:
  13.         sys.exit("Usage: ./tweets twitterhandle")
  14.        
  15.     positives = os.path.join(sys.path[0], "positive-words.txt")
  16.     negatives = os.path.join(sys.path[0], "negative-words.txt")
  17.    
  18.     text = sys.argv[1].strip("@")
  19.     tweets = helpers.get_user_timeline(text, 50)
  20.     if tweets == None:
  21.         sys.exit("Error: problem getting tweets")
  22.    
  23.     analyzer = Analyzer(positives, negatives)
  24.     for tweet in tweets:
  25.         final_score = analyzer.analyze(tweet)
  26.         print(final_score, end="")
  27.         if final_score > 0.0:
  28.             print(colored(final_score, "green"))
  29.         elif final_score < 0.0:
  30.             print(colored(final_score, "red"))
  31.         else:
  32.             print(colored(final_score, "yellow"))
  33.            
  34. if __name__ == "__main__":
  35.     main()
Add Comment
Please, Sign In to add comment