# coding=utf-8
'''
Created on 2016年4月5日
使用mongodb对数组的条件查询快速得到类目结构
@author: ChenHao
'''
from util_common import Constant
from pymongo.mongo_client import MongoClient
from analysis import detailAnalysis_kind
import time

'''
运算得到mouser的分类
'''
# TimeStart = time.time()
# # 先获得现在的kind
# # cli = MongoClient("mongodb://localhost:27017/")
# cli = MongoClient(Constant.MONGODB_URL)
# db = cli.spider
# rs = db.kind_temp_1.find()
# kindlist = list()
# for r in rs:
#     kindlist.append(r)
# 
# print (kindlist)
# print (len(kindlist))
# cli.close()
# 
# detail_analysis_kind = detailAnalysis_kind.DetailAnalysisKind(kindlist)
# cli = MongoClient(Constant.MONGODB_URL)
# db = cli.spider
# rs = db.component_original.find({}, {"kinds": True})
# print (rs.count())
# for r in rs:
#     detail_analysis_kind._add_into_out_list(r["kinds"])
# detail_analysis_kind.._set_isLeaf()
# db.kind_temp_2.insert_many(detail_analysis_kind.out_list)
# print (detail_analysis_kind.out_list)
# print (len(detail_analysis_kind.out_list))
# cli.close()
# 
# TimeEnd = time.time()
# print ("耗时", TimeEnd - TimeStart)

'''
生成csv文件给测试部对照
'''

cli = MongoClient(Constant.MONGODB_URL)
db = cli.spider

# fout = open("kind_mouser.csv", "a")
# rs = db.kind_temp_2.find()
# for r in rs:
#     line = str(r["id"]) + "," + str(r["nameCn"]).replace(",", "，") + "," + str(r["parentid"]) + "," + str(r["isLeaf"]) + "\n"
#     fout.write(line)
# fout.close()


fout = open("kind_uu.csv", "a")
rs = db.kind_uu.find()
for r in rs:
    line = str(r["id"]) + "," + str(r["nameCn"]).replace(",", "，") + "," + str(r["parentid"]) + "," + str(r["isLeaf"]) + "\n"
    fout.write(line)
fout.close()

cli.close()