# coding:utf-8 import os import csv import pymongo import time ''' mongodb 删除数据库 use test; db.dropDatabase(); mongodb删除表 db.mytable.drop(); 清空表 db.mytable.remove({}) ''' class c_stock: def __init__(self, code, date, open, high, low, close,volume, adj_close): self.code = code self.date = date self.open = open self.high = high self.low = low self.close = close self.volume = volume self.adj_close = adj_close #mongodb_link = 'mongodb://127.0.0.1:27017' #mongoClient = MongoClient(mongodb_link) conn = pymongo.MongoClient("localhost", 27017) conn.db_stock.authenticate("d","zz") db = conn.db_stock stocks = db.stocks def write_dict(type): db.stocks.insert(type.__dict__) ''' with open('000002.csv') as csvfile: reader = csv.DictReader(csvfile) for row in reader: one = c_stock(code='000002', date=row['date'], open=float(row['open']), high=float(row['high']), low=float(row['low']), close=float(row['close']), volume=int(row['volume']), adj_close=float(row['adj_close']), ) write_dict(one) ''' #db.stocks.drop() #db.stocks.remove({}) reader = csv.reader(file('000002.csv','rb')) for row in reader: if reader.line_num == 1: continue one = c_stock(code='000002', date=row[0], open=float(row[1]), high=float(row[2]), low=float(row[3]), close=float(row[4]), volume=int(row[5]), adj_close=float(row[6])) write_dict(one) print db.stocks.count() #time.sleep(3) #for i in db.stocks.find(): # print i
python mongodb 设置密码前一篇ok,csv文件存入mongodb
最新推荐文章于 2023-07-11 17:39:58 发布