Esempio n. 1
0
            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)

Esempio n. 2
0
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)