#
# Usage: python load_gtrends.py [keyword1] [keyword2] ...
#
import lxml
import os
import sys
from pytrends.pyGTrends import pyGTrends
from pytrends.pyGTrends import parse_data

def load_gtrends(keylist):
    google_username = ""
    google_password = ""
    path = "./dat/"
    if not os.path.exists(path):
        os.mkdir(path)

    # connect to Google
    custom_useragent = {'User-Agent': 'getGTrend'}
    connector = pyGTrends(google_username, google_password, custom_useragent)

    # make request
    for key in keylist:
        payload = {'q': key}
        connector.request_report(payload)
        # download file
        connector.save_csv(path, key)
        #data = parse_data(connector.data)
        #print(data)

if __name__ == "__main__":
    arg = sys.argv
    num = len(arg)
    key = []
    for i in range(1, num):
        key.append(arg[i])
    load_gtrends(key)