def __init__(self, api=None):
"""
Function: init
--------------
Initialize global parameter
"""
super(listener, self).__init__()
self.number_tweets = 0 # Initialize number of tweets to 0
self.max_tweets = 10 # Maximum number of tweet. Over this number the program stops
self.med_words_list =[] # List of median of all tweet collected
self.med_words = [] # Ledian of unique words in one tweet
self.out = output_print()
self.median_number_list = []
def median_number(self, file_name):
"""
Function: median_number
-----------------------
file_name: input file are tweets and are used to calculate median
return a list of medians. It counts the unique words per tweet
"""
file_object = open(file_name, "r")
number_word_list = []
for line in file_object:
unique_words_per_tweet = sorted(set(line.rstrip().split(" ")))
number_word_list.append(len(Counter((unique_words_per_tweet))))
self.median_number_list.append(numpy.median(numpy.array(number_word_list)))
return self.median_number_list
input_file = sys.argv[1]
output_file = sys.argv[2]
out = output_print()
med = median_unique()
med_list = med.median_number(input_file)
out.print_med_to_screen(med_list)
out.print_med_to_file(med_list, output_file)
