theData = getAllFields(theSymbol) data.append(theData) except: print >> sys.stderr, "Failed trying to get data for %s" % (theSymbol,) return data def processSymbols(symbolList=sys.argv[1:]): #pool = multiprocessing.Pool(processes=1) #return filter(None, pool.map(getAllFields, symbolList)) return filter(None, map(getAllFields, symbolList)) #from pyGTrends import pyGTrends #connector = pyGTrends('mohanraotekale','tekalemohan') def fetchHotness(companyName, numPeriods=5): connector.download_report((companyName)) lines = connector.csv().split('\n') skipCount = -len(lines)/numPeriods periods = {} count = 0 for i in range(len(lines)-1, 1, skipCount): periods['t-minus-%03d' % (count,)] = lines[i].split(",")[1].strip() count += 1 return periods if __name__ == '__main__': data = processSymbols(sys.argv[1:]) csvutil.writeCSVContentsFromDictionaries(data, sorted_field_list) #getData(t)
import csvutil, sys, optparse def parseArgs(args): parser = optparse.OptionParser() parser.add_option("-d", "--delimiter", dest="delimiter", default='\t') parser.add_option("-f", "--formula", dest="formula", default='\t') parser.add_option("-t", "--formula-column-title", dest="columnTitle", default='\t') (options, args) = parser.parse_args() return options, args if __name__ == '__main__': options, args = parseArgs(sys.argv[1:]) headerFields, rows = csvutil.getHeaderAndContentsAsDictionaries(delimiter=options.delimiter) newHeaderFields = headerFields[:] newHeaderFields.append(options.columnTitle) for i in range(1, len(rows) + 1): rows[i-1][options.columnTitle] = options.formula.replace('_NNNN', str(i+1)) #print rows[i-1] csvutil.writeCSVContentsFromDictionaries(rows, fieldnames=newHeaderFields, delimiter=options.delimiter)