qingdan/subdir/util.py

import xml.etree.ElementTree as ET
from typing import Union
from fastapi import FastAPI
import os
import uuid
import re
from fastapi.middleware.cors import CORSMiddleware
from pydantic import BaseModel
import pandas as pd
from numpy import float64

def huizongrcj(data):
    huizong = {}
    result = [
    [ "ID", "编码", "名称", "规格型号", "单位", "数量", "单价", "合价", "产地", 
    "供应厂商", "人材机类别", "主要材料标志", "主材标志", "设备标志", "甲供标志", "暂估价标志"]]
    for entry in data:
        bm = entry[1]
        name = entry[2]
        danwei = entry[4]
        key = bm + name + danwei
        if key in huizong:
            huizong[key].append(entry)
        else:
            huizong[key] = [entry]
    for key,value in huizong.items():
        result_hl = 0
        for i in range(0, len(value)):
            result_hl = result_hl + float(value[i][10])
        result_dj = float(value[0][5])
        result_hj = result_hl * result_dj
        result.append([value[0][0], value[0][1], value[0][2], value[0][3], value[0][4], result_hl,
                       value[0][5], result_hj, value[0][6], value[0][7], value[0][8], value[0][13],
                       value[0][14], value[0][15], value[0][9], value[0][12]])
    return result