mongoDBにデータを追加する

Tweet

CSVファイルを1行読み込んではmongoDBに1行insertするサンプルプログラム。
一括でインポートする方法もあるはずだが、ひとまず基礎練習ということで書いてみた。

# coding:UTF-8
import csv
from pymongo import MongoClient

if __name__ == '__main__':
    client = MongoClient('mongodb://localhost:27017/')
    db = client["company"] #DBの名前を指定

    with open("./00_zenkoku_all_20210531.csv", 'r', encoding="utf-8") as f:
        reader = csv.reader(f)
        for row in reader:
            obj = {
                "corporateNumber": row[1],
                "process": row[2],
                "correct": row[3],
                "updateDate": row[4],
                "changeDate": row[5],
                "name": row[6],
                "nameImageId": row[7],
                "kind": row[8],
                "prefectureName": row[9],
                "cityName": row[10],
                "streetNumber": row[11],
                "addressImageId": row[12],
                "prefectureCode": row[13],
                "cityCode": row[14],
                "postCode": row[15],
                "addressOutside": row[16],
                "addressOutsideImageId": row[17],
                "closeDate": row[18],
                "closeCause": row[19],
                "successorCorporateNumber": row[20],
                "changeCause": row[21],
                "assignmentDate": row[22],
                "latest": row[23],
                "enName": row[24],
                "enPrefectureName": row[25],
                "enCityName": row[26],
                "enAddressOutside": row[27],
                "furigana": row[28],
                "hihyoji": row[29],
            }
            db.MasterCompanies.insert_one(obj) #dbの後ろはコレクション名
            print(obj["cityName"], obj["name"])
        db.close()