import sys
import unicodecsv as csv
import unicodedata
from googletrans import Translator
import time

if sys.version_info[0] < 3:
    import got
else:
    import got3 as got

def main():

    def printTweet(t):
        print("Text: %s\n" % t.text.encode('utf-8'))

    tweetCriteria = got.manager.TweetCriteria().setUsername("MLP_officiel").setSince("2017-03-23").setUntil("2017-04-23").setMaxTweets(1000)
    tweets = got.manager.TweetManager.getTweets(tweetCriteria)

    translator = Translator()

    with open('tweets.csv', 'wb') as csvfile:
        tweetwriter = csv.writer(csvfile, encoding='utf-8', delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)
        for tweet in tweets:
            text_fran = unicode(tweet.text).encode("utf-8")
            trans_object = translator.translate(text_fran)

            # Ovo je bitno da nas google ne banuje
            time.sleep(1)

            text = unicode(trans_object.text).encode("utf-8")
            tweetwriter.writerow([tweet.date, text])


if __name__ == '__main__':
    main()