Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #!/usr/bin/env python3
- import os
- import sys
- import nltk
- from analyzer import Analyzer
- from termcolor import colored
- from helpers import get_user_timeline
- def main():
- if len(sys.argv) != 2:
- sys.exit("Usage: ./tweets twitterhandle")
- positives = os.path.join(sys.path[0], "positive-words.txt")
- negatives = os.path.join(sys.path[0], "negative-words.txt")
- text = sys.argv[1].strip("@")
- tweets = helpers.get_user_timeline(text, 50)
- if tweets == None:
- sys.exit("Error: problem getting tweets")
- analyzer = Analyzer(positives, negatives)
- for tweet in tweets:
- final_score = analyzer.analyze(tweet)
- print(final_score, end="")
- if final_score > 0.0:
- print(colored(final_score, "green"))
- elif final_score < 0.0:
- print(colored(final_score, "red"))
- else:
- print(colored(final_score, "yellow"))
- if __name__ == "__main__":
- main()
Add Comment
Please, Sign In to add comment