python mongodb 讀寫CSV檔案

標籤：

# -*- coding: utf-8 -*-
import os
import csv
import pymongo
from pymongo import MongoClient

#建立串連
client = MongoClient(‘10.20.4.79‘, 27017)
#client = MongoClient(‘10.20.66.106‘, 27017)
db_name = ‘ta‘ #資料庫名
db = client[db_name]

 

#讀取CVS檔案並插入到mongoDB資料庫 的tvsplst集合中
def InsertDataFromCVS(csvFile):
　　strFile = unicode(csvFile,‘utf8‘)
　　if os.path.isfile(strFile):
　　　　readFile = file(strFile, ‘rb‘)
　　　　reader = csv.reader(readFile)
　　　　i = 0
　　　　listKey =[]
　　　　sqlDict = {}
　　　　for line in reader:
　　　　　　i = i + 1
　　　　　　sqlDict.clear()
　　　　　　if i == 1:
　　　　　　　　for it in line:
　　　　　　　　　　if it == ‘PlstID‘:
　　　　　　　　　　　　listKey.append(‘_id‘)
　　　　　　　　　　else:
　　　　　　　　　　　　listKey.append(it)
　　　　　　else :
　　　　　　　　j = 0
　　　　　　　　for it in line:
　　　　　　　　　　sqlDict[listKey[j]] = it.decode(‘gbk‘)
　　　　　　　　　　j = j + 1

　　　　　　　　try:
　　　　　　　　　　db.tvsplst.insert(sqlDict)
　　　　　　　　except:
　　　　　　　　　　print i,sqlDict
　　　　readFile.close()

#測試代碼
strcsvFile=‘D:\\TVSPlst.csv‘
InsertDataFromCVS(strcsvFile)

 

#從mongoDB資料庫中讀取tvsplst資料寫到CSV檔案裡面
def WriteTvsPlsttoCsv(csvFile):
　　strFile = unicode(csvFile,‘utf8‘)
　　headList =[‘PlstID‘, ‘MediaID‘, ‘PlstDate‘, ‘Hard_STime‘]
　　writer = csv.writer(file(strFile, ‘wb‘))
　　writer.writerow(headList)
　　headList[0]=‘_id‘

　　for u in db.tvsplst.find():
　　　　vList = []
　　　　for k in headList:
　　　　　　vList.append(u[k].encode(‘utf8‘))
　　　　　　writer.writerow(vList)

#測試代碼
csvfile = ‘D:\\test.csv‘
WriteTvsPlsttoCsv(csvfile)

python mongodb 讀寫CSV檔案