mongo.py 文件源码

def loadMcCsv(self, fileName, dbName, symbol):
        """?Multicharts???csv??????????Mongo????"""
        import csv

        start = time()
        print u'????CSV??%s???????%s?%s?' %(fileName, dbName, symbol)

        # ??????????
        host, port = loadMongoSetting()

        client = pymongo.MongoClient(host, port)    
        collection = client[dbName][symbol]
        collection.ensure_index([('datetime', pymongo.ASCENDING)], unique=True)   

        # ???????????
        reader = csv.DictReader(file(fileName, 'r'))
        for d in reader:
            bar = CtaBarData()
            bar.vtSymbol = symbol
            bar.symbol = symbol
            bar.open = float(d['Open'])
            bar.high = float(d['High'])
            bar.low = float(d['Low'])
            bar.close = float(d['Close'])
            bar.date = datetime.strptime(d['Date'], '%Y/%m/%d').strftime('%Y%m%d')
            bar.time = d['Time']
            bar.datetime = datetime.strptime(bar.date + ' ' + bar.time, '%Y%m%d %H:%M:%S')
            bar.volume = d['TotalVolume']

            flt = {'datetime': bar.datetime}
            collection.update_one(flt, {'$set':bar.__dict__}, upsert=True)  
            print bar.date, bar.time

        print u'????????%s' % (time()-start)