Code: Alles auswählen
# -*- coding: utf-8 -*-
import time
import codecs
import random
import glob
from pytrends.request import TrendReq
google_username = "@gmail.com"
google_password = ""
f = open('', 'r')
stocks_no_name = []
for line in f.readlines():
data = line.split('\t')
stock_no = data[0].strip()
stock_name = data[1].strip()
stocks_no_name.append([stock_no, stock_name])
f.close()
files=glob.glob('*.csv')
downloaded_files = [fd.title().lower()[0:4] for fd in files]
stocks_no_name_new = []
for stock_no_name in stocks_no_name:
if not stock_no_name[0] in downloaded_files:
stocks_no_name_new.append(stock_no_name)
stocks_no_name = stocks_no_name_new
print len(stocks_no_name)
# connect to Google
pytrend = TrendReq(google_username, google_password, custom_useragent='My Pytrends Script')
while stocks_no_name:
stock_index = random.randint(0,len(stocks_no_name)-1)
stock_no_name = stocks_no_name[stock_index]
stock_no = stock_no_name[0]
stock_name = stock_no_name[1]
print stock_no, stock_name
try:
one_stock_data = []
trend_payload = {'q': stock_name, 'date': ''2013-12-29 2016-12-31', 'geo': 'TW','tz': 'Etc/GMT+8'}
# trend
trend = pytrend.trend(trend_payload)
time.sleep(random.randint(120, 360))
table = trend['table']
rows = table['rows']
for i in range(len(rows)):
row_data = []
for j in range(len(rows[0]['c'])):
row_data.append(rows[i]['c'][j]['v'])
one_stock_data.append(row_data)
# output one_stock_data to a file
filename = unicode(stock_no, errors='ignore') + '.csv'
outfile = codecs.open(filename, "wb", "utf-8")
for i in range(len(one_stock_data)):
one_stock_data_str = str(one_stock_data[i][0]) + ", " + str(one_stock_data[i][1])
if i != len(one_stock_data) - 1:
one_stock_data_str = one_stock_data_str + "\r\n"
outfile.write(one_stock_data_str)
outfile.close()
stocks_no_name.pop(stock_index)
except:
time.sleep(random.randint(120, 360))
continue
Code: Alles auswählen
from pytrends.request import TrendReq
import time
from random import randint
google_username = ""
google_password = ""
#Connect to google
pytrends = TrendReq(google_username, google_password, custom_useragent='My Pytrends Script')
# trend
trend_payload = {kw_list=['pizza'],timeframe='2013-12-29 2016-12-31', geo='TW'}
trend = pytrends.build_payload(trend_payload)
#Request a report based on keyword
print(pizza)
#Save a csv
path = 'c:\Users\user\Desktop'
connector.save_csv(path, pizza)
time.sleep(randint(5,10))