# -*- coding:utf-8 -*- 
# Author: loco_python
import json
import csv
import sys
reload(sys)
sys.setdefaultencoding('utf-8')
from pymongo import MongoClient


def mongodb_to_json():
    client = MongoClient('mongodb://localhost:27017/')
    db = client.database_name   # database_name (連線資料庫的名字)
    collection = db.collection_name # collection_name（連線表的名字）
 

    with open('aaaaa.json', 'a') as f:
        data = list()
        for item in collection.find():
            data.append(item)
        f.write(json.dumps(data))

def json_to_csv():
    # json檔案
    json_file = open('aaaaa.json', 'r')
    # csv檔案
    csv_file = open('aaaaa.csv', 'w')
    # csv讀寫器
 

    csv_writer = csv.writer(csv_file)
    # 表頭和正文
    data = json.load(json_file)
    # 表頭
    sheet_title = data[0].keys()
    content_list = [dic.values() for dic in data]
    # 5.通過讀寫器 寫入csv檔案
    csv_writer.writerow(sheet_title)
    csv_writer.writerows(content_list)
    # 6.關閉檔案
    json_file.close()
    csv_file.close()


if
 
 __name__ == '__main__':
    mongodb_to_json()
    json_to_csv()
    print 'OK'