import xml.etree.ElementTree as ET from subdir import service, util from bson.objectid import ObjectId import uuid ###############helper################## def getDingercj_(root, bh, bt, qdbm, debm): result = [["ID", "人材机编码", "名称", "规格型号", "单位", "单价", "产地", "供应厂商", "人材机类别", "甲供标志", "含量", "合价", "暂估价标志", "主要材料标志", "主材标志", "设备标志" ]] #result.append(["ming cheng", "jin e", "zan gu jia", "anquan wenming fei", "guifei"]) item = None for child in root: if child.tag == "Dxgcxx": for child1 in child: Dwgcbh = child1.attrib["Dwgcbh"] if Dwgcbh == bh: item = child1 break ##result.append([child.attrib["Dxgcmc"], child.attrib["Je"], child.attrib["Zgj"], child.attrib["Aqwmf"], child.attrib["Gf"]]) Fywj = None for child in item: if child.tag == 'Qdxm': Fywj= child break Rcjhz = None for child in item: if child.tag == 'Rcjhz': Rcjhz = child break kv = {} for entry in Rcjhz: kv[entry.attrib["RcjId"]]=entry container = None for child in Fywj: ##child is mx if child.tag == "QdBt": if child.attrib["Mc"] == bt: container = child break if not container: container = Fywj zimu = None for child in container: if child.attrib["Qdbm"] == qdbm: zimu = child break box = None if not zimu: return result for child in zimu: if child.tag == "Qdxdezj": box = child break hlmx_parent = None for child in box: if child.attrib["Debm"] == debm: for grandchild in child: if grandchild.tag == "Qdxdercjhl": hlmx_parent = grandchild for child in hlmx_parent: result.append([child.attrib["RcjId"], kv[child.attrib["RcjId"]].attrib["RcjBm"], kv[child.attrib["RcjId"]].attrib["Mc"], kv[child.attrib["RcjId"]].attrib["Ggxh"], kv[child.attrib["RcjId"]].attrib["Dw"], kv[child.attrib["RcjId"]].attrib["Dj"], kv[child.attrib["RcjId"]].attrib["Cd"], kv[child.attrib["RcjId"]].attrib["Gycs"], kv[child.attrib["RcjId"]].attrib["Rcjlb"], kv[child.attrib["RcjId"]].attrib["Jgbz"], child.attrib["Rcjhl"], child.attrib["Rcjhj"], "", "", "", "" ]) return result def getQingdanrcj_(root, bh, bt, bm): result = [["ID", "人材机编码", "名称", "规格型号", "单位", "单价", "产地", "供应厂商", "人材机类别", "甲供标志", "含量", "合价", "暂估价标志", "主要材料标志", "主材标志", "设备标志" ]] #result.append(["ming cheng", "jin e", "zan gu jia", "anquan wenming fei", "guifei"]) item = None for child in root: if child.tag == "Dxgcxx": for child1 in child: Dwgcbh = child1.attrib["Dwgcbh"] if Dwgcbh == bh: item = child1 break ##result.append([child.attrib["Dxgcmc"], child.attrib["Je"], child.attrib["Zgj"], child.attrib["Aqwmf"], child.attrib["Gf"]]) Fywj = None for child in item: if child.tag == 'Qdxm': Fywj= child break Rcjhz = None for child in item: if child.tag == 'Rcjhz': Rcjhz = child break kv = {} for entry in Rcjhz: kv[entry.attrib["RcjId"]]=entry container = None for child in Fywj: ##child is mx if child.tag == "QdBt": if child.attrib["Mc"] == bt: container = child break if not container: container = Fywj zimu = None for child in container: if child.attrib["Qdbm"] == bm: zimu = child break box = None if not zimu: return result for child in zimu: if child.tag == "Qdxrcjhl": box = child break for child in box: result.append([child.attrib["RcjId"], kv[child.attrib["RcjId"]].attrib["RcjBm"], kv[child.attrib["RcjId"]].attrib["Mc"], kv[child.attrib["RcjId"]].attrib["Ggxh"], kv[child.attrib["RcjId"]].attrib["Dw"], kv[child.attrib["RcjId"]].attrib["Dj"], kv[child.attrib["RcjId"]].attrib["Cd"], kv[child.attrib["RcjId"]].attrib["Gycs"], kv[child.attrib["RcjId"]].attrib["Rcjlb"], kv[child.attrib["RcjId"]].attrib["Jgbz"], child.attrib["Rcjhl"], child.attrib["Rcjhj"], child.attrib["Zgjbz"], child.attrib["Zyclbz"], child.attrib["Zcbz"], child.attrib["Sbbz"] ]) return result def getDjcsDingercj_(root, bh, bt, qdbm, debm): result = [["ID", "人材机编码", "名称", "规格型号", "单位", "单价", "产地", "供应厂商", "人材机类别", "甲供标志", "含量", "合价", "暂估价标志", "主要材料标志", "主材标志", "设备标志" ]] #result.append(["ming cheng", "jin e", "zan gu jia", "anquan wenming fei", "guifei"]) item = None for child in root: if child.tag == "Dxgcxx": for child1 in child: Dwgcbh = child1.attrib["Dwgcbh"] if Dwgcbh == bh: item = child1 break ##result.append([child.attrib["Dxgcmc"], child.attrib["Je"], child.attrib["Zgj"], child.attrib["Aqwmf"], child.attrib["Gf"]]) Fywj = None for child in item: if child.tag == 'Csxm': Fywj= child break Rcjhz = None for child in item: if child.tag == 'Rcjhz': Rcjhz = child break kv = {} for entry in Rcjhz: kv[entry.attrib["RcjId"]]=entry container = None for child in Fywj: if child.tag == "DjCs": container = child break zimu = None for child in container: if child.attrib["Bm"] == qdbm: zimu = child break box = None if not zimu: return result for child in zimu: if child.tag == "Csxdezj": box = child break hlmx_parent = None for child in box: if child.attrib["Debm"] == debm: for grandchild in child: if grandchild.tag == "Csxdercjhl": hlmx_parent = grandchild for child in hlmx_parent: result.append([child.attrib["RcjId"], kv[child.attrib["RcjId"]].attrib["RcjBm"], kv[child.attrib["RcjId"]].attrib["Mc"], kv[child.attrib["RcjId"]].attrib["Ggxh"], kv[child.attrib["RcjId"]].attrib["Dw"], kv[child.attrib["RcjId"]].attrib["Dj"], kv[child.attrib["RcjId"]].attrib["Cd"], kv[child.attrib["RcjId"]].attrib["Gycs"], kv[child.attrib["RcjId"]].attrib["Rcjlb"], kv[child.attrib["RcjId"]].attrib["Jgbz"], child.attrib["Rcjhl"], child.attrib["Rcjhj"], "", "", "", "" ]) return result def getDjcsQingdanrcj_(root, bh, bt, bm): result = [["ID", "人材机编码", "名称", "规格型号", "单位", "单价", "产地", "供应厂商", "人材机类别", "甲供标志", "含量", "合价", "暂估价标志", "主要材料标志", "主材标志", "设备标志" ]] #result.append(["ming cheng", "jin e", "zan gu jia", "anquan wenming fei", "guifei"]) item = None for child in root: if child.tag == "Dxgcxx": for child1 in child: Dwgcbh = child1.attrib["Dwgcbh"] if Dwgcbh == bh: item = child1 break ##result.append([child.attrib["Dxgcmc"], child.attrib["Je"], child.attrib["Zgj"], child.attrib["Aqwmf"], child.attrib["Gf"]]) Fywj = None for child in item: if child.tag == 'Csxm': Fywj= child break hl_parent = None for child in Fywj: if child.tag == "DjCs": hl_parent = child Rcjhz = None for child in item: if child.tag == 'Rcjhz': Rcjhz = child break kv = {} for entry in Rcjhz: kv[entry.attrib["RcjId"]]=entry zimu = None for child in hl_parent: if child.attrib["Bm"] == bm: zimu = child break box = None if not zimu: return result for child in zimu: if child.tag == "Csxrcjhl": box = child break for child in box: result.append([child.attrib["RcjId"], kv[child.attrib["RcjId"]].attrib["RcjBm"], kv[child.attrib["RcjId"]].attrib["Mc"], kv[child.attrib["RcjId"]].attrib["Ggxh"], kv[child.attrib["RcjId"]].attrib["Dw"], kv[child.attrib["RcjId"]].attrib["Dj"], kv[child.attrib["RcjId"]].attrib["Cd"], kv[child.attrib["RcjId"]].attrib["Gycs"], kv[child.attrib["RcjId"]].attrib["Rcjlb"], kv[child.attrib["RcjId"]].attrib["Jgbz"], child.attrib["Rcjhl"], child.attrib["Rcjhj"], child.attrib["Zgjbz"], child.attrib["Zyclbz"], child.attrib["Zcbz"], child.attrib["Sbbz"] ]) return result ######################end of helper ######################## async def resolve(manager, websocket, data, client): tree = ET.parse("data/" + data) root = tree.getroot() print(root) Dwgcbh = [] jingjibiao = {} jingjibiao["BiaoDuanNO"] = root.attrib["BiaoDuanNO"] jingjibiao["Jsfs"] = root.attrib["Jsfs"] jingjibiao["Version"] = root.attrib["Version"] jingjibiao["Xmmc"] = root.attrib["Xmmc"] jingjibiao["Dxgcxx"] = [] for child in root: if child.tag == "ZhaoBiaoXx": jingjibiao["ZhaoBiaoXx"] = {} jingjibiao["ZhaoBiaoXx"]["BzTime"] = child.attrib["BzTime"] jingjibiao["ZhaoBiaoXx"]["Bzr"] = child.attrib["Bzr"] jingjibiao["ZhaoBiaoXx"]["FhTime"] = child.attrib["FhTime"] jingjibiao["ZhaoBiaoXx"]["Fhr"] = child.attrib["Fhr"] jingjibiao["ZhaoBiaoXx"]["Zbr"] = child.attrib["Zbr"] jingjibiao["ZhaoBiaoXx"]["ZbrDb"] = child.attrib["ZbrDb"] jingjibiao["ZhaoBiaoXx"]["Zxr"] = child.attrib["Zxr"] jingjibiao["ZhaoBiaoXx"]["ZxrDb"] = child.attrib["ZxrDb"] elif child.tag == "TouBiaoXx": jingjibiao["TouBiaoXx"] = {} jingjibiao["TouBiaoXx"]["Zbr"] = child.attrib["Zbr"] jingjibiao["TouBiaoXx"]["Tbr"] = child.attrib["Tbr"] jingjibiao["TouBiaoXx"]["TbrDb"] = child.attrib["TbrDb"] jingjibiao["TouBiaoXx"]["Bzr"] = child.attrib["Bzr"] jingjibiao["TouBiaoXx"]["BzTime"] = child.attrib["BzTime"] jingjibiao["TouBiaoXx"]["Tbzj"] = child.attrib["Tbzj"] jingjibiao["TouBiaoXx"]["Zgj"] = child.attrib["Zgj"] jingjibiao["TouBiaoXx"]["Aqwmf"] = child.attrib["Aqwmf"] jingjibiao["TouBiaoXx"]["Gf"] = child.attrib["Gf"] elif child.tag == "Dxgcxx": Dxgcxx = {} Dxgcxx["Aqwmf"] = child.attrib["Aqwmf"] Dxgcxx["Dxgcbh"] = child.attrib["Dxgcbh"] Dxgcxx["Dxgcmc"] = child.attrib["Dxgcmc"] Dxgcxx["Gf"] = child.attrib["Gf"] Dxgcxx["Je"] = child.attrib["Je"] Dxgcxx["Zgj"] = child.attrib["Zgj"] Dxgcxx["Dwgc"] = [] for grandchild in child: if grandchild.tag == "Dwgcxx": Dwgcxx = {} Dwgcxx["DogNum"] = grandchild.attrib["DogNum"] Dwgcxx["Dwgcbh"] = grandchild.attrib["Dwgcbh"] Dwgcbh.append({"bh": grandchild.attrib["Dwgcbh"], "num": grandchild.attrib["DogNum"]}) Dwgcxx["Dwgcmc"] = grandchild.attrib["Dwgcmc"] Dwgcxx["MachineKey"] = grandchild.attrib["MachineKey"] Dwgcxx["SoftName"] = grandchild.attrib["SoftName"] Dwgcxx["SoftNum"] = grandchild.attrib["SoftNum"] Dwgcxx["Zylb"] = grandchild.attrib["Zylb"] Dxgcxx["Dwgc"].append(Dwgcxx) jingjibiao["Dxgcxx"].append(Dxgcxx) db = client["baojia"] collection = db["jingjibiao"] biao_id = (await collection.insert_one(jingjibiao)).inserted_id biao_id = str(biao_id) collection = db["Dwgc"] for entry in Dwgcbh: bjhz = service.getBjhz(root, entry["bh"])###array which contains __children gfsj = service.getGfsj(root, entry["bh"])##array which contains __children qtxm = service.getQtxm(root, entry["bh"])##array which contains __children zlje = service.getZlje(root, entry["bh"]) jrg = service.getJrg(root, entry["bh"]) zcbfwf = service.getZcbfwf(root, entry["bh"]) fbrgycl = service.getFbrgycl(root, entry["bh"]) rcjhz = service.getRcjhz(root, entry["bh"]) zjcs = service.getZjcs(root, entry["bh"]) qdbt = service.getQdxm(root, entry["bh"]) zygczgj = service.getZygczgj(root, entry["bh"]) Dwgc = {} Dwgc["bjhz"] = bjhz Dwgc["gfsj"] = gfsj Dwgc["qtxm"] = qtxm Dwgc["zlje"] = zlje Dwgc["jrg"] = jrg Dwgc['zygczgj'] = zygczgj Dwgc["zcbfwf"] = zcbfwf Dwgc["fbrgycl"] = fbrgycl Dwgc["rcjhz"] = rcjhz Dwgc["zjcs"] = zjcs Dwgc["qdbt"] = qdbt Dwgc["DogNum"] = entry["num"] Dwgc["Dwgcbh"] = entry["bh"] Dwgc["biao_id"] = biao_id await collection.insert_one(Dwgc) collection = db["Djcs"] for entry in Dwgcbh: djcs = service.getDjcs(root, entry["bh"]) for cs in djcs: cs["DogNum"] = entry["num"] cs["Dwgcbh"] = entry["bh"] cs["biao_id"] = entry["biao_id"] rcj = getDjcsQingdanrcj_(root, entry["bh"], "", cs['清单编码']) cs["rcj"] = rcj if "__children" in cs: children = cs["__children"] for child in children: dercj = getDjcsDingercj_(root, entry["bh"], "", cs["清单编码"], child["清单编码"]) child["dercj"] = dercj child['fuzhuEnable'] = False await collection.insert_one(cs) collection = db["qdxm"] for entry in Dwgcbh: bts = service.getQdxm(root, entry["bh"]) for bt in bts: qdmx = service.getQdmx(root, entry["bh"], bt) for qd in qdmx: qd["DogNum"] = entry["num"] qd["Dwgcbh"] = entry["bh"] qd["biao_id"] = biao_id qd["bt"] = bt rcj = getQingdanrcj_(root, entry["bh"], bt, qd["清单编码"]) qd["rcj"] = rcj if "__children" in qd: children = qd["__children"] for child in children: dercj = getDingercj_(root, entry["bh"], bt, qd["清单编码"], child["清单编码"]) child["dercj"] = dercj child['fuzhuEnable'] = False await collection.insert_one(qd) collection = db["qufei"] document = jingjibiao dxgc = document['Dxgcxx'] result = [] for entry in dxgc: newone = {"名称" : entry['Dxgcmc'], "工程类型":"", "工程类别": "", "管理费（%）": "", "利润（%）": "","备注":"", "key" : str(uuid.uuid4()) } children = [] for dwgc in entry['Dwgc']: children.append({"名称" : dwgc['Dwgcmc'], "工程类型":"", "工程类别": "", "管理费（%）": "", "利润（%）": "","备注":"", "key" : dwgc['Dwgcbh']}) newone["children"] = children result.append(newone) await collection.insert_one({"biao_id": biao_id, "qufei": [{ "名称" : document['Xmmc'], "工程类型":"", "工程类别": "", "管理费（%）": "", "利润（%）": "","备注":"", "key" : str(uuid.uuid4()), "children" : result }]}) await manager.send_personal_message(f"You wrote: {data}", websocket) async def list_files(client): db = client["baojia"] collection = db["jingjibiao"] result = [] async for post in collection.find(): result.append([str(post['_id']), post["Xmmc"]]) return result def getDwgc(id, Zylb): result = [] #result.append(["ming cheng", "jin e", "zan gu jia", "anquan wenming fei", "guifei"]) ##result.append([child.attrib["Dxgcmc"], child.attrib["Je"], child.attrib["Zgj"], child.attrib["Aqwmf"], child.attrib["Gf"]]) id = id + "Zylb" + Zylb result.append({"id": id + "bao jia hui zong", "label": "报价汇总表"}) result.append({"id": id + "gui fei shui jin", "label": "规费税金"}) result.append({"id": id + "qing dan xiang mu", "label": "清单项目"}) result.append({"id": id + "cuo shi xiang mu", "label": "措施项目"}) result.append({"id": id +"qi ta xiang mu", "label": "其他项目"}) result.append({"id": id +"zan lie jin e", "label": "暂列金额"}) result.append({"id": id +"cai liao zan gu jia", "label": "材料暂估价"}) result.append({"id": id + "zhuan ye gong cheng zan gu jia", "label": "专业工程暂估价"}) result.append({"id": id + "ji ri gong", "label": "计日工"}) result.append({"id": id + "zong cheng bao fu wu fei", "label": "总承包服务费"}) result.append({"id": id + "zong jia xiang mu jin du kuan zhi fu fen jie", "label": "总价项目进度款支付分解"}) result.append({"id": id + "fa bao ren gong ying cai liao", "label": "发包人供应材料"}) result.append({"id": id + "cheng bao ren gong ying cai liao", "label": "承包人供应材料"}) result.append({"id": id + "ren cai ji hui zong", "label": "人材机汇总"}) return result async def getOutline(client, id): db = client["baojia"] collection = db["jingjibiao"] document = await collection.find_one({'_id': ObjectId(id)}) result = [] if 'TouBiaoXx' in document: result.append({"id" : "TouBiaoXx", "label" : "投标信息" }) if 'Dxgcxx' in document: Dxgcxx = document["Dxgcxx"] for dxgc in Dxgcxx: result2 = [] Dwgc = dxgc["Dwgc"] for grandchild in Dwgc: result2.append({"id" : grandchild['Dwgcbh'], "Zylb":grandchild['Zylb'], "label": grandchild["Dwgcmc"], "children" : getDwgc(grandchild['Dwgcbh'], grandchild['Zylb'])}) result.append({"id" : dxgc['Dxgcbh'], "label" : dxgc['Dxgcmc'], "children" : result2}) return result async def getDetail(client, id): db = client["baojia"] collection = db["jingjibiao"] document = await collection.find_one({'_id': ObjectId(id)}) result = [] result.append(["名称", "金额", "暂估价", "安全文明施工费", "规费"]) if 'TouBiaoXx' in document: toubiaoxx = document["TouBiaoXx"] result.append([toubiaoxx["Zbr"], toubiaoxx["Tbzj"], toubiaoxx["Zgj"], toubiaoxx["Aqwmf"], toubiaoxx["Gf"]]) if 'Dxgcxx' in document: Dxgcxx = document["Dxgcxx"] for dxgc in Dxgcxx: result.append([dxgc["Dxgcmc"], dxgc["Je"], dxgc["Zgj"], dxgc["Aqwmf"], dxgc["Gf"]]) return result async def getBjhz(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['bjhz'] async def getGfsj(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['gfsj'] async def getQtxm(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['qtxm'] async def getZygczgj(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['zygczgj'] async def getZlje(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['zlje'] async def getJrg(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['jrg'] async def getZcbfwf(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({"Dwgcbh": id, "biao_id": objectid}) return document['zcbfwf'] async def getFbrgycl(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['fbrgycl'] async def getRcjhz(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['rcjhz'] async def getZjcs(client, objectid, id): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({ "Dwgcbh": id, "biao_id": objectid}) return document['zjcs'] async def getDjcs(client, objectid, id): db = client["baojia"] collection = db["Djcs"] result = [] async for post in collection.find({"Dwgcbh": id, "biao_id": objectid}): ##post["ID"] = post["_id"] del post["_id"] ##print( post["rcj"]) ## del post["DogNum"] result.append(post) return result async def getDjcsQingdanrcj(client, name, bh, bt, bm): db = client["baojia"] collection = db["Djcs"] document = await collection.find_one({ "Dwgcbh": bh, "清单编码": bm, "biao_id": name}) result = document["rcj"] return result async def getDjcsDingercj(client, name, bh, bt, qdbm, debm): db = client["baojia"] collection = db["Djcs"] document = await collection.find_one({"Dwgcbh": bh, "清单编码": qdbm, "biao_id": name}) children = document["__children"] result = [] for child in children: if child["清单编码"] == debm: result = child["dercj"] return result async def getQdxm(client, name, bh): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({'biao_id': name, "Dwgcbh": bh}) if document: return document['qdbt'] else: return [] async def getQdmx(client, name, bh, bt): db = client["baojia"] collection = db["qdxm"] result = [] async for post in collection.find({'biao_id': name, "Dwgcbh": bh, "bt": bt}): ##post["ID"] = post["_id"] del post["_id"] #del post["rcj"] #del post["DogNum"] result.append(post) return result async def getQingdanrcj(client, name, bh, bt, bm): db = client["baojia"] collection = db["qdxm"] document = await collection.find_one({ "Dwgcbh": bh, "清单编码": bm, "biao_id": name, "bt": bt}) result = document["rcj"] return result async def getDingercj(client, name, bh, bt, qdbm, debm): db = client["baojia"] collection = db["qdxm"] document = await collection.find_one({"Dwgcbh": bh, "清单编码": qdbm, "biao_id": name, "bt": bt}) result = [] if "__children" in document: children = document["__children"] for child in children: if child["清单编码"] == debm: result = child["dercj"] return result async def getQufei(client, id): db = client["baojia"] collection = db["qufei"] document = await collection.find_one({'biao_id': id}) return document['qufei'] async def searchde(client, zhuanye, text): db = client["dinge"] collection = db["de-collection"] result = [] async for post in collection.find({'$text': {"$search": "\"" + text + "\""}, "zhuanye": zhuanye}): result.append(post["DEBH"] + " " + post['GCLMC'] ) if len(result) > 10: break return result ##################################editor########################## async def updatezjcs(client, id, bh, mc, fl): db = client["baojia"] collection = db["Dwgc"] document = await collection.find_one({'biao_id': id, 'Dwgcbh': bh}, {'rcjhz': 0}) zjcs = document['zjcs'] for entry in zjcs: if entry['名称'] == mc: entry['费率'] = fl entry['金额'] = str(float(entry['计算基数']) * float(fl) / float(100)) await collection.update_one({'biao_id': id, "Dwgcbh": bh}, {'$set': {'zjcs': zjcs}}) async def save(client, data): db = client["baojia"] collection = db["qdxm"] biao_id = None bh = None for entry in data: entry["__children"] = entry["_children"] del entry["_children"] biao_id = entry['biao_id'] bh = entry['Dwgcbh'] await collection.replace_one({"biao_id": entry["biao_id"], "Dwgcbh": entry["Dwgcbh"], "bt": entry["bt"], "清单编码": entry["清单编码"]}, entry) if not biao_id: return {"result" : "ok"} rcjhz = [] async for post in collection.find({'biao_id': biao_id, "Dwgcbh": bh}, {'__children': 0}): qdrcj = post['rcj'][1:] for entry in qdrcj: entry[10] = float(entry[10]) * float(post['数量']) rcjhz = rcjhz + qdrcj collection = db["Djcs"] async for post in collection.find({'biao_id': biao_id, "Dwgcbh": bh}, {'__children': 0}): qdrcj = post['rcj'][1:] for entry in qdrcj: entry[10] = float(entry[10]) * float(post['数量']) rcjhz = rcjhz + qdrcj newHuizong = util.huizongrcj(rcjhz) collection = db["Dwgc"] await collection.update_one({'biao_id': biao_id, 'Dwgcbh': bh}, {'$set': {'rcjhz' : newHuizong}}) await zongjiaDwgc(client, biao_id, bh) return {"result": "ok"} async def savedjcs(client, data): db = client["baojia"] collection = db["Djcs"] biao_id = None bh = None for entry in data: entry["__children"] = entry["_children"] del entry["_children"] biao_id = entry['biao_id'] bh = entry['Dwgcbh'] await collection.replace_one({"biao_id": entry["biao_id"], "Dwgcbh": entry["Dwgcbh"], "清单编码": entry["清单编码"]}, entry) if not biao_id: return {"result" : "ok"} rcjhz = [] collection = db["qdxm"] async for post in collection.find({'biao_id': biao_id, "Dwgcbh": bh}, {'__children': 0}): qdrcj = post['rcj'][1:] for entry in qdrcj: entry[10] = float(entry[10]) * float(post['数量']) rcjhz = rcjhz + qdrcj collection = db["Djcs"] async for post in collection.find({'biao_id': biao_id, "Dwgcbh": bh}, {'__children': 0}): qdrcj = post['rcj'][1:] for entry in qdrcj: entry[10] = float(entry[10]) * float(post['数量']) rcjhz = rcjhz + qdrcj newHuizong = util.huizongrcj(rcjhz) collection = db["Dwgc"] await collection.update_one({'biao_id': biao_id, 'Dwgcbh': bh}, {'$set': {'rcjhz' : newHuizong}}) await zongjiaDwgc(client, biao_id, bh) return {"result": "ok"} async def applyFL(client, id, data): db = client["baojia"] collection = db["qufei"] await collection.replace_one({"biao_id": id}, {"biao_id": id, "qufei": data}) collection = db['qdxm'] xm = data[0] for dxgc in xm["children"]: for dwgc in dxgc["children"]: bh = dwgc['key'] glf = float(dwgc['管理费（%）']) lr = float(dwgc['利润（%）']) bz = dwgc['备注'] ##print(bh+glf+lr+bz) async for post in collection.find({'biao_id': id, "Dwgcbh": bh}): rgfs = float(post['人工费']) clfs = float(post['材料费']) jxfs = float(post['机械费']) sl = float(post['数量']) if rgfs < 0.0001 and clfs < 0.0001 and jxfs < 0.0001: continue ##danxiang = False for child in post['__children']:##一条定额 bm = child['清单编码'] if bm.startswith("D"): ##danxiang = True continue rgf = float(child['人工费']) clf = float(child['材料费']) jxf = float(child['机械费']) zhdj = float(child['综合单价']) sl_ = float(child['数量']) if len(bz) == 0: ##常规计算 child['管理费'] = (rgf + jxf) * glf / float(100) child['利润'] = (rgf + jxf) * lr / float(100) child['综合单价'] = rgf + clf + jxf + child['管理费'] + child['利润'] child['合价'] = sl * sl_ * child['综合单价'] else: child['管理费'] = (rgf) * glf / float(100) child['利润'] = (rgf) * lr / float(100) child['综合单价'] = rgf + clf + jxf + child['管理费'] + child['利润'] child['合价'] = sl * sl_ * child['综合单价'] if True: post['管理费'] = 0 post['利润'] = 0 for entry in post['__children']: post['管理费'] = post['管理费'] + float(entry['管理费']) * float(entry['数量']) post['利润'] = post['利润'] + float(entry['利润']) * float(entry['数量']) post['管理费'] = str(post['管理费']) post['利润'] = str(post['利润']) post['综合单价'] = str(float(post['人工费']) + float(post['材料费']) + float(post['机械费']) + float(post['管理费']) + float(post['利润'])) post['合价'] = str(float(post['综合单价']) * float(post['数量'])) await collection.replace_one({'_id': post['_id']}, post ) collection = db['Djcs'] for dxgc in xm["children"]: for dwgc in dxgc["children"]: bh = dwgc['key'] glf = float(dwgc['管理费（%）']) lr = float(dwgc['利润（%）']) bz = dwgc['备注'] ##print(bh+glf+lr+bz) async for post in collection.find({'biao_id': id, "Dwgcbh": bh}): rgfs = float(post['人工费']) clfs = float(post['材料费']) jxfs = float(post['机械费']) sl = float(post['数量']) if rgfs < 0.0001 and clfs < 0.0001 and jxfs < 0.0001: continue ##danxiang = False##一般有单项定额的，清单就一条定额，就是这个单项定额 for child in post['__children']: bm = child['清单编码'] if bm.startswith("D"): ##danxiang = True continue rgf = float(child['人工费']) clf = float(child['材料费']) jxf = float(child['机械费']) zhdj = float(child['综合单价']) sl_ = float(child['数量']) if len(bz) == 0: ##常规计算 child['管理费'] = (rgf + jxf) * glf / float(100) child['利润'] = (rgf + jxf) * lr / float(100) child['综合单价'] = rgf + clf + jxf + child['管理费'] + child['利润'] child['合价'] = sl * sl_ * child['综合单价'] else: child['管理费'] = (rgf) * glf / float(100) child['利润'] = (rgf) * lr / float(100) child['综合单价'] = rgf + clf + jxf + child['管理费'] + child['利润'] child['合价'] = sl * sl_ * child['综合单价'] if True: post['管理费'] = 0 post['利润'] = 0 for entry in post['__children']: post['管理费'] = post['管理费'] + float(entry['管理费']) * float(entry['数量']) post['利润'] = post['利润'] + float(entry['利润']) * float(entry['数量']) post['管理费'] = str(post['管理费']) post['利润'] = str(post['利润']) post['综合单价'] = str(float(post['人工费']) + float(post['材料费']) + float(post['机械费']) + float(post['管理费']) + float(post['利润'])) post['合价'] = str(float(post['综合单价']) * float(post['数量'])) await collection.replace_one({'_id': post['_id']}, post ) for dxgc in xm["children"]: for dwgc in dxgc["children"]: bh = dwgc['key'] await zongjiaDwgc(client, id, bh) return {"result": "ok"} async def tiaojia(client, biao_id, bh, bm, mingcheng, danwei, jiage, glf, lr, bz): db = client["baojia"] collection = db["qdxm"] async for post in collection.find({'biao_id': biao_id, "Dwgcbh": bh}): if "__children" not in post: ##print(post) continue children = post["__children"] qdsl = post['数量'] hit = False for entry in children:###对一条定额，调价 hitde = False dercj = entry["dercj"] dergf = 0 declf = 0 dejxf = 0 for i in range(1, len(dercj)): rcj = dercj[i] if rcj[1] == bm and rcj[2] == mingcheng and rcj[4] == danwei: hit = True hitde = True rcj[5] = jiage hanliang = rcj[10] rcj[11] = str(float(jiage) * float(hanliang)) if int(rcj[8]) == 1: dergf = dergf + float(rcj[11]) elif int(rcj[8]) == 2: declf = declf + float(rcj[11]) elif int(rcj[8]) == 3: dejxf = dejxf + float(rcj[11]) if hitde: entry['人工费'] = str(dergf) entry['辅材费'] = str(declf) entry['材料费'] = str(declf) entry['机械费'] = str(dejxf) if bm.startswith("D"): entry['管理费'] = '0' entry['利润'] = '0' elif bz == 0: entry['管理费'] = str((dergf + dejxf) * float(glf) / float(100)) entry['利润'] = str((dergf + dejxf) * float(lr) / float(100)) else: entry['管理费'] = str((dergf) * float(glf) / float(100)) entry['利润'] = str((dergf) * float(lr) / float(100)) entry['综合单价'] = str(float(entry['人工费']) + float(entry['材料费']) + float(entry['机械费']) + float(entry['管理费']) + float(entry['利润'])) entry['合价'] = str(float(entry['综合单价']) * float(entry['数量']) * float(qdsl)) if hit: ##调整清单 qdrcj = post['rcj'] qdrgf = 0 qdclf = 0 qdjxf = 0 for i in range(1, len(qdrcj)): entry = qdrcj[i] if entry[1] == bm and entry[2] == mingcheng and entry[4] == danwei: entry[5] = jiage hanliang = entry[10] entry[11] = str(float(jiage) * float(hanliang)) if int(entry[8]) == 1: qdrgf = qdrgf + float(entry[11]) elif int(entry[8]) == 2: qdclf = qdclf + float(entry[11]) elif int(entry[8]) == 3: qdjxf = qdjxf + float(entry[11]) post['人工费'] = str(qdrgf) post['辅材费'] = str(qdclf) post['材料费'] = str(qdclf) post['机械费'] = str(qdjxf) post['管理费'] = 0 post['利润'] = 0 for entry in children: post['管理费'] = post['管理费'] + float(entry['管理费']) * float(entry['数量']) post['利润'] = post['利润'] + float(entry['利润']) * float(entry['数量']) post['管理费'] = str(post['管理费']) post['利润'] = str(post['利润']) post['综合单价'] = str(float(post['人工费']) + float(post['材料费']) + float(post['机械费']) + float(post['管理费']) + float(post['利润'])) post['合价'] = str(float(post['综合单价']) * float(qdsl)) print(post) await collection.replace_one({'_id': post['_id']}, post ) collection = db["Djcs"] async for post in collection.find({'biao_id': biao_id, "Dwgcbh": bh}): children = post["__children"] qdsl = post['数量'] hit = False for entry in children:###对一条定额，调价 hitde = False dercj = entry["dercj"] dergf = 0 declf = 0 dejxf = 0 for i in range(1, len(dercj)): rcj = dercj[i] if rcj[1] == bm and rcj[2] == mingcheng and rcj[4] == danwei: hit = True hitde = True rcj[5] = jiage hanliang = rcj[10] rcj[11] = str(float(jiage) * float(hanliang)) if int(rcj[8]) == 1: dergf = dergf + float(rcj[11]) elif int(rcj[8]) == 2: declf = declf + float(rcj[11]) elif int(rcj[8]) == 3: dejxf = dejxf + float(rcj[11]) if hitde: entry['人工费'] = str(dergf) entry['辅材费'] = str(declf) entry['材料费'] = str(declf) entry['机械费'] = str(dejxf) if bm.startswith("D"): entry['管理费'] = '0' entry['利润'] = '0' elif bz == 0: entry['管理费'] = str((dergf + dejxf) * float(glf) / float(100)) entry['利润'] = str((dergf + dejxf) * float(lr) / float(100)) else: entry['管理费'] = str((dergf) * float(glf) / float(100)) entry['利润'] = str((dergf) * float(lr) / float(100)) entry['综合单价'] = str(float(entry['人工费']) + float(entry['材料费']) + float(entry['机械费']) + float(entry['管理费']) + float(entry['利润'])) entry['合价'] = str(float(entry['综合单价']) * float(entry['数量']) * float(qdsl)) if hit: ##调整清单 qdrcj = post['rcj'] qdrgf = 0 qdclf = 0 qdjxf = 0 for i in range(1, len(qdrcj)): entry = qdrcj[i] if entry[1] == bm and entry[2] == mingcheng and entry[4] == danwei: entry[5] = jiage hanliang = entry[10] entry[11] = str(float(jiage) * float(hanliang)) if int(entry[8]) == 1: qdrgf = qdrgf + float(entry[11]) elif int(entry[8]) == 2: qdclf = qdclf + float(entry[11]) elif int(entry[8]) == 3: qdjxf = qdjxf + float(entry[11]) post['人工费'] = str(qdrgf) post['辅材费'] = str(qdclf) post['材料费'] = str(qdclf) post['机械费'] = str(qdjxf) post['管理费'] = 0 post['利润'] = 0 for entry in children: post['管理费'] = post['管理费'] + float(entry['管理费']) * float(entry['数量']) post['利润'] = post['利润'] + float(entry['利润']) * float(entry['数量']) post['管理费'] = str(post['管理费']) post['利润'] = str(post['利润']) post['综合单价'] = str(float(post['人工费']) + float(post['材料费']) + float(post['机械费']) + float(post['管理费']) + float(post['利润'])) post['合价'] = str(float(post['综合单价']) * float(qdsl)) await collection.replace_one({'_id': post['_id']}, post ) collection = db["Dwgc"] document = await collection.find_one({'biao_id': biao_id, "Dwgcbh": bh}) hz = document['rcjhz'] for entry in hz: if entry[1] == bm and entry[2] == mingcheng and entry[4] == danwei: entry[6] = jiage entry[7] = float(jiage) * float(entry[5]) await collection.update_one({'biao_id': biao_id, "Dwgcbh": bh}, {'$set': {'rcjhz': hz}}) await zongjiaDwgc(client, biao_id, bh) async def zongjiaDwgc(client, biao_id, bh): db = client["baojia"] collection = db["qdxm"] sum = 0 async for post in collection.find({'biao_id': biao_id, "Dwgcbh": bh}, {'__children': 0}): sum = sum + float(post['合价']) collection = db["Djcs"] async for post in collection.find({'biao_id': biao_id, "Dwgcbh": bh}, {'__children': 0}): sum = sum + float(post['合价']) collection = db['Dwgc'] document = await collection.find_one({'biao_id': biao_id, "Dwgcbh": bh}, {'rcjhz': 0}) zjcs = document['zjcs'] def process(n): n['计算基数'] = str(sum) n['金额'] = float(sum) * float(n['费率']) / float(100) if "__children" in n: for entry in n['__children']: entry['计算基数'] = str(sum) entry['金额'] = float(sum) * float(entry['费率']) / float(100) return n result = [process(item) for item in zjcs] for entry in result: if '安全文明施工' in entry['名称']: sum_ = 0 for child in entry['__children']: sum_ = sum_ + float(child['金额']) entry['金额'] = str(sum_) entry['计算基数'] = str(sum_) zjcs_sum = 0 for entry in result: zjcs_sum = zjcs_sum + float(entry['金额']) qtxm = document['qtxm'] qtxm_sum = 0 for entry in qtxm: qtxm_sum = qtxm_sum + float(entry['金额']) gfsj = document['gfsj'] gf_sum = 0 sj_sum = 0 for item in gfsj: if item['名称'] == '规费': child = item['__children'] for children in child: children['取费基数'] = str(sum + zjcs_sum + qtxm_sum) children['金额'] = str(float(sum + zjcs_sum + qtxm_sum) * float(children['费率']) / float(100)) gf_sum = gf_sum + float(children['金额']) for item in gfsj: if item['名称'] == '规费': item['取费基数'] = str(gf_sum) item['金额'] = str(gf_sum) for item in gfsj: if item['名称'] == '税金': item['取费基数'] = str(sum + zjcs_sum + qtxm_sum + gf_sum) item['金额'] = float(item['取费基数']) * float(item['费率']) / float(100) sj_sum = float(item['金额']) for item in gfsj: if item['名称'] == '合计': item['取费基数'] = str(sj_sum + gf_sum) item['金额'] = item['取费基数'] await collection.update_one({'biao_id': biao_id, "Dwgcbh": bh}, {'$set': {'zjcs': result, 'gfsj' : gfsj}})