import matplotlib.pyplot as plt
import matplotlib.dates as dates
from pytrends.request import TrendReq

plt.rcdefaults()
plt.rcParams['figure.figsize'] = 6, 4
grid_alpha = 0.15
line_width = 1.5

username = "google username"
password = "google password"
pytrend = TrendReq(username, password, hl='en-US', tz=360, custom_useragent=None)

keyword=['fidget spinner+fidget spinners']
start_date = "2017-01-15"
end_date = "2017-06-05"
window_size = 7

#Building dataframe
pytrend.build_payload(kw_list=keyword, cat=0, timeframe='{} {}'.format(start_date, end_date))
df = pytrend.interest_over_time()
df = df.rolling(window=window_size,center=False).mean().dropna()
df[keyword] = 100*df[keyword]/df[keyword].max()

#Plotting dataframe
fig, ax = plt.subplots()
print('start_date:{} end_date:{} window_size:{}'.format(start_date, end_date, window_size))
ax.plot(df, linewidth=line_width)

ax.grid(linestyle='-', linewidth=1, alpha=grid_alpha)
ax.xaxis_date()
ax.xaxis.set_major_formatter(dates.DateFormatter('%b %d'))
fig.autofmt_xdate()
fig.tight_layout()

#Saving visualisation
fig.savefig('fidget_spinner_search_popularity_7d_rolling_mean.svg')