from pytrends.request import TrendReq


# Login to Google. Only need to run this once, the rest of requests will use the same session.
pytrend = TrendReq()

# Create payload and capture API tokens. Only needed for interest_over_time(), interest_by_region() & related_queries()
keyword = 'war on terror'
pytrend.build_payload(kw_list=[keyword])

# Interest Over Time
print('Interest Over Time')
interest_over_time_df = pytrend.interest_over_time()
print(interest_over_time_df.head())

# Interest by Region
print('Interest by Region')
interest_by_region_df = pytrend.interest_by_region()
print(interest_by_region_df.head())

# Related Queries, returns a dictionary of dataframes
print('Related Queries')
related_queries_dict = pytrend.related_queries()
# print(related_queries_dict)


rising_df = related_queries_dict[keyword]['rising']
top_df = related_queries_dict[keyword]['top']
related_queries = {
            'rising': rising_df.to_dict(orient='record'),
            'top': top_df.to_dict(orient='record')
        }
print(related_queries)



# Get Google Hot Trends data
# trending_searches_df = pytrend.trending_searches()
# print(trending_searches_df.head())

# Get Google Top Charts
# top_charts_df = pytrend.top_charts(cid='actors', date=201611)
# print(top_charts_df.head())

# Get Google Keyword Suggestions
# print('Google Keyword Suggestions')
# suggestions_dict = pytrend.suggestions(keyword='war on terror')
# print(suggestions_dict)