import xml.etree.ElementTree as ET
from typing import Union
from fastapi import FastAPI
import os
import uuid
import re
import zipfile
import json
from fastapi.middleware.cors import CORSMiddleware
from fastapi.middleware.gzip import GZipMiddleware
import base64
from pydantic import BaseModel
from subdir import service
from subdir import db
from subdir import util
import xml.dom.minidom as MD
import numpy as np
from fastapi.staticfiles import StaticFiles
from pymongo import AsyncMongoClient
client = AsyncMongoClient()
from fastapi.responses import FileResponse

from fastapi import WebSocket, WebSocketDisconnect
from fastapi import UploadFile
class ConnectionManager:
    """Class defining socket events"""
    def __init__(self):
        """init method, keeping track of connections"""
        self.active_connections = []
    
    async def connect(self, websocket: WebSocket):
        """connect event"""
        await websocket.accept()
        self.active_connections.append(websocket)

    async def send_personal_message(self, message: str, websocket: WebSocket):
        """Direct Message"""
        await websocket.send_text(message)
    
    def disconnect(self, websocket: WebSocket):
        """disconnect event"""
        self.active_connections.remove(websocket)


app = FastAPI()


origins = [
    "http://localhost.tiangolo.com",
    "https://localhost.tiangolo.com",
    "http://localhost",
    "http://localhost:9002",
    "http://127.0.0.1:9002",
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)
app.add_middleware(GZipMiddleware, minimum_size=1000, compresslevel=5)

app.mount("/static", StaticFiles(directory="front/dist"), name="static")
manager = ConnectionManager()


for root, dirs, files in os.walk("./data", topdown=False):
    for name in files:
        print(os.path.join(root, name))
        file_data = os.path.join(root, name)





def getDetail(root):

    print(root.tag)
    print(root.text)
    print(root.attrib)
    result = []
    result.append(["名称", "金额", "暂估价", "安全文明施工费", "规费"])
    for child in root:
        print(child.tag)
        print(child.attrib)
    
        if child.tag == "TouBiaoXx":
            result.append([child.attrib["Zbr"], child.attrib["Tbzj"], child.attrib["Zgj"], child.attrib["Aqwmf"], child.attrib["Gf"]])
        if child.tag == "Dxgcxx":
            Dxgcbh = child.attrib["Dxgcbh"]
            Dxgcmc = child.attrib["Dxgcmc"]

            result.append([child.attrib["Dxgcmc"], child.attrib["Je"], child.attrib["Zgj"], child.attrib["Aqwmf"], child.attrib["Gf"]])

    return result
def getOutline(root):

    print(root.tag)
    print(root.text)
    print(root.attrib)
    result = []
    for child in root:
        print(child.tag)
        print(child.attrib)
    
        if child.tag == "TouBiaoXx":
            result.append({"id" : "TouBiaoXx", "label" : "投标信息" })
        if child.tag == "Dxgcxx":
            Dxgcbh = child.attrib["Dxgcbh"]
            Dxgcmc = child.attrib["Dxgcmc"]
            result2 = []
            for child1 in child:
                #print("level2===================================")
                #print(child1.tag)
                #print(child1.attrib)
                Dwgcbh = child1.attrib["Dwgcbh"]
                Dwgcmc = child1.attrib["Dwgcmc"]
                Zylb = child1.attrib["Zylb"]
                result2.append({"id" : Dwgcbh,"Zylb":Zylb,  "label": Dwgcmc, "children" : service.getDwgc(root, Dwgcbh, Zylb)})

            result.append({"id" : Dxgcbh, "label" : Dxgcmc, "children" : result2})

    return result

class Info(BaseModel):
    name: str
class DingeshuRequest(BaseModel):
    value: int
class DingeXilieRequest(BaseModel):
    value: int
    id: str
class SingleDingeXilieRequest(BaseModel):
    zhuanye: int
    debh: str

class InfoWithID(BaseModel):
    name: str
    id: str


class Tiaojia(BaseModel):
    biao_id: str
    bh: str
    bm: str
    mingcheng: str
    danwei: str
    jiage: str
    glf: str
    lr: str
    bz: int

class SearchDe(BaseModel):
    zhuanye: str
    text: str

class UpdateZjcs(BaseModel):
    biao_id: str
    bh: str
    mc: str
    fl: str
    

@app.post("/outline")
async def read_root(info: Info):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return getOutline(root)

@app.post("/outline2")
async def read_root2(info: Info):
    
    return await db.getOutline(client, info.name)

@app.post("/detail")
async def read_detail(info: Info):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return getDetail(root)

@app.post("/detail2")
async def read_detail2(info: Info):
    
    return await db.getDetail(client, info.name)

@app.post("/baojiahuizong/")
async def read_bjhz(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getBjhz(root, info.id)

@app.post("/baojiahuizong2/")
async def read_bjhz2(info: InfoWithID):
   
    raw = await db.getBjhz(client, info.name, info.id)
    raw2 = []
    for entry in raw:
        if "__children" in entry:
            entry["_children"] = entry["__children"]
            for grandchild in entry["_children"]:
                if "__children" in grandchild:
                    grandchild['_children'] = grandchild['__children']
                    del grandchild['__children']
            del entry["__children"]
        raw2.append(entry)
    return raw2

@app.post("/guifeishuijin/")
async def read_gfsj(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getGfsj(root, info.id)

@app.post("/guifeishuijin2/")
async def read_gfsj2(info: InfoWithID):
    raw = await db.getGfsj(client, info.name, info.id)
    raw2 = []
    for entry in raw:
        if "__children" in entry:
            entry["_children"] = entry["__children"]
            del entry["__children"]
        raw2.append(entry)
    return raw2


@app.post("/qitaxiangmu/")
async def read_qtxm(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getQtxm(root, info.id)

@app.post("/qitaxiangmu2/")
async def read_qtxm2(info: InfoWithID):
    raw = await db.getQtxm(client, info.name, info.id)
    raw2 = []
    for entry in raw:
        if "__children" in entry:
            entry["_children"] = entry["__children"]
            del entry["__children"]
        raw2.append(entry)
    return raw2

@app.post("/zygczgj/")
async def zygczgj(info: InfoWithID):
    return await db.getZygczgj(client, info.name, info.id)

@app.post("/zanliejine/")
async def read_zlje(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getZlje(root, info.id)


@app.post("/zanliejine2/")
async def read_zlje2(info: InfoWithID):
    return await db.getZlje(client, info.name, info.id)

@app.post("/jirigong/")
async def read_jrg(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getJrg(root, info.id)

@app.post("/jirigong2/")
async def read_jrg2(info: InfoWithID):
    return await db.getJrg(client, info.name, info.id)

@app.post("/zongchengbaofuwufei/")
async def read_zcbfwf(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getZcbfwf(root, info.id)

@app.post("/zongchengbaofuwufei2/")
async def read_zcbfwf2(info: InfoWithID):
    return await db.getZcbfwf(client, info.name, info.id)


@app.post("/fabaorengongyingcailiao/")
async def read_fbrgycl(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getFbrgycl(root, info.id)

@app.post("/fabaorengongyingcailiao2/")
async def read_fbrgycl2(info: InfoWithID):
    return await db.getFbrgycl(client, info.name, info.id)


@app.post("/rencaijihuizong/")
async def read_rcjhz(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getRcjhz(root, info.id)

@app.post("/rencaijihuizong2/")
async def read_rcjhz2(info: InfoWithID):
    return await db.getRcjhz(client, info.name, info.id)


@app.post("/qingdanxiangmu/")
async def read_qdxm(info: InfoWithID):
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            if name == info.name:
                print(os.path.join(root, name))
                file_data = os.path.join(root, name)
    tree = ET.parse(file_data)
    root = tree.getroot()
    return service.getQdxm(root, info.id)
    #return []

@app.post("/qingdanxiangmu2/")
async def read_qdxm2(info: InfoWithID):
    
    return await db.getQdxm(client, info.name, info.id)

class Item(BaseModel):
    bh: str
    bt: str
    name: str
class Rcj(BaseModel):
    bh: str
    bt: str
    bm: str
    name: str
class Dercj(BaseModel):
    bh: str
    bt: str
    qdbm: str
    debm: str
    name: str
class Zjcs(BaseModel):
    bh: str
    name: str


@app.post("/qingdanmingxi/")
async def read_qdmx(item : Item):

    return await db.getQdmx(client, item.name, item.bh, item.bt)

@app.post("/qingdanrcj/")
async def read_rcj(item : Rcj):

    if item.bt == "Djcs":
        return await db.getDjcsQingdanrcj(client, item.name, item.bh, item.bt, item.bm)
    return await db.getQingdanrcj(client, item.name, item.bh, item.bt, item.bm)

@app.post("/qingdantuijian/")
async def read_tuijian(item : Rcj):
    
        
    return service.getQingdanTuijian(item.bh, item.bt, item.bm)


@app.post("/dingercj/")
async def read_dercj(item : Dercj):
    if item.bt == "Djcs":
        return await db.getDjcsDingercj(client, item.name, item.bh, item.bt, item.qdbm, item.debm)

    
    return await db.getDingercj(client, item.name, item.bh, item.bt, item.qdbm, item.debm)


@app.post("/zjcs/")
async def read_zjcs(item : Zjcs):
    raw = await db.getZjcs(client, item.name, item.bh)
    raw2 = []
    for entry in raw:
        if "__children" in entry:
            entry["_children"] = entry["__children"]
            del entry["__children"]
        raw2.append(entry)
    return raw2


@app.post("/djcs/")
async def read_djcs(item : Zjcs):
    raw = await db.getDjcs(client, item.name, item.bh)
    raw2 = []
    for entry in raw:
        if "__children" in entry:
            entry["_children"] = entry["__children"]
            del entry["__children"]
        raw2.append(entry)
    return raw2

@app.post("/files/")
async def read_files():
    result = []
    for root, dirs, files in os.walk("./data", topdown=False):
        for name in files:
            print(os.path.join(root, name))
            result.append([name, "", ""])
    return result

@app.post("/files2/")
async def read_files2():
    result = await db.list_files(client)
    return result

@app.post("/deleteFile/")
async def delete(r: Info):
    result = await db.delete_files(client, r.name)
    return result

@app.post("/des/")
async def read_des(r: DingeshuRequest):
    result = service.getDes(r.value)
    #print(result)
    return result

@app.post("/pbs/")
async def read_pbs(r: DingeshuRequest):
    result = service.getPbs(r.value)
    #print(result)
    result.insert(0, {"id": "0", "label": "全部"})
    return result

@app.post("/pbxl/")
async def read_pbxl(r: Info):
    result = service.getPbxl(r.name)
    #print(result)
    return result

@app.post("/cankao/")
async def cankao():
    result = service.getCankao()
    #print(result)
    return result

@app.post("/qufei/")
async def read_qufei(r: Info):
    return await db.getQufei(client, r.name)

@app.post("/dexilie/")
async def read_dexilie(r: DingeXilieRequest):
    result = service.getDeXilie(r.value, r.id)
    #print(result)
    return result

@app.post("/singledexilie/")
async def read_singledexilie(r: SingleDingeXilieRequest):
    if r.debh.startswith("D") :
        return json.dumps({
            "reverse": "None", 
            "rgde": None,
            "jxde": None,
            "clde": None,
            "actual_zhuanye": r.zhuanye,
            "bz_selected": {"BZBH": {}},
            "bz_selected2": {"BZBH": {}}
        }, ensure_ascii=False)
    
    result1, result2, rgde, jxde, clde, bz_selected, bz_selected2, actual_zhuanye= service.getSingleDeXilie(r.zhuanye, r.debh)
    print("get result ***************************************")
    if result1:
        result3 = json.loads(result1)
    else:
        result3 = {}
    result3["reverse"] = str(result2)
    result3["rgde"] = rgde
    result3["jxde"] = jxde
    result3["clde"] = clde
    result3["actual_zhuanye"] = actual_zhuanye
    if bz_selected != None:
        result3["bz_selected"] = json.loads(bz_selected)
    else:
        result3["bz_selected"] = {"BZBH": {}}
    if bz_selected2 != None:
        result3["bz_selected2"] = json.loads(bz_selected2)
    else:
        result3["bz_selected2"] = {"BZBH": {}}
    if "[" in r.debh:
        position3 = r.debh.find("*")
        coef = 1
        if position3 == -1:
            pass
        else:
            coef = r.debh[1 + position3:]
            print(coef)
            tail = 0
            for i in range(0, len(coef) + 1):
                if i == len(coef):
                    tail = i
                    break
                if coef[i] > '9' or coef[i] < '0':
                    tail = i
                    break
            if tail == 0:
                print("1113111")
                result3["reverse"] = 'None'
                return json.dumps(result3, ensure_ascii=False)
            coef = int(coef[0:tail])

        position1 = r.debh.find("[")
        position2 = r.debh.find("]")
        if position2 == -1:
            print("1121111")
            result3["reverse"] = 'None'
            return json.dumps(result3, ensure_ascii=False)
        if r.debh[position1-1] == "+":
            pass
        elif r.debh[position1-1] == "-":
            coef = -coef
        else:
            print("111111")
            result3["reverse"] = 'None'
            return json.dumps(result3, ensure_ascii=False)
        debh = r.debh[position1+1: position2]
        if result3['GLDE']:
            hit = False
            for key, value in result3['GLDE'].items():
                if value == debh:
                    hit = True
            if not hit:
                result3["reverse"] = 'None'
                return json.dumps(result3, ensure_ascii=False)
        else:
            result3["reverse"] = 'None'
            return json.dumps(result3, ensure_ascii=False)
        result1_, result2_, rgde_, jxde_, clde_, bz_selected_, bz_selected2_, actual_zhuanye_ = service.getSingleDeXilie(r.zhuanye, debh)
        if result1_:
            util.mergerg(rgde, rgde_, coef)
            util.mergejx(jxde, jxde_, coef)
            util.mergecl(clde, clde_, coef)
            result3["rgde"] = rgde
            result3["jxde"] = jxde
            result3["clde"] = clde
            item_ = None
            for item in result3['DW'].keys():
                item_ = item
            result3['DEBH'] = {item_: util.cleanBM(r.debh)}
        else:
            result3["reverse"] = 'None'
        
    return json.dumps(result3, ensure_ascii=False)


@app.post("/save/")
async def save(r: Info):
    data = json.loads(r.name)
    print(data)
 
    return await db.save(client, data)


@app.post("/savedjcs/")
async def savedjcs(r: Info):
    data = json.loads(r.name)
    print(data)
 
    return await db.savedjcs(client, data)


@app.post("/applyFL/")
async def applyFL(r: InfoWithID):
    data = json.loads(r.name)
    print(data)
 
    return await db.applyFL(client, r.id, data)


@app.post("/tiaojia/")
async def tiaojia(r: Tiaojia):
    ##return []
    glf = r.glf
    lr = r.lr
    if r.glf == '':
        glf = '0'
    if r.lr == "":
        lr = '0'
    return await db.tiaojia(client, r.biao_id, r.bh, r.bm, r.mingcheng, r.danwei, r.jiage, glf, lr, r.bz)

@app.post("/searchde/")
async def searchde(r: SearchDe):
    ##return []
    
    return await db.searchde(client, r.zhuanye, r.text)

@app.post("/updatezjcs/")
async def updatezjcs(r: UpdateZjcs):
    ##return []
    
    return await db.updatezjcs(client, r.biao_id, r.bh, r.mc, r.fl)


@app.post("/upload/")
async def upload(file: UploadFile):
    if file.filename.endswith('13jz') or file.filename.endswith("13jt"):
        content = await file.read()
        string = content.decode('utf-8')
        await resolve(string)
        return [file.filename]
    elif file.filename.endswith("jszf"):
        content = await file.read()
        string = content.decode('utf-8')
        root = ET.fromstring(string)
        for child in root:
            print(child)
            if child.tag == "ZBFileContent":
                base64_bytes = child.text.encode("ascii")

                sample_string_bytes = base64.b64decode(base64_bytes)
        tmp = str(uuid.uuid4())
        os.mkdir(os.path.join("tmp", tmp))
        with open(tmp, "wb") as file_:
            file_.write(sample_string_bytes)
        with zipfile.ZipFile(tmp, 'r') as zip_ref:
            zip_ref.extractall(os.path.join("tmp", tmp))
        dir_list = os.listdir(os.path.join("tmp", tmp))
        print(dir_list)
        for entry in dir_list:
            if entry.endswith("13jz"):
                with open(os.path.join("tmp", tmp, entry), 'r') as f:
                    await resolve(f.read()) 

        return [file.filename]
    else:
        return [file.filename]
        
        
@app.get("/download/{item_id}")
async def download_file(item_id):
    file_path = "export/contacts.xml.b"
    # Create the root element
    root = ET.Element("JingJiBiao")

    # Create a sub-element
    ##TouBiaoXx = ET.SubElement(root, "TouBiaoXx")
    await db.build(client, root, item_id)

    

    # Create the tree and write to a file
    tree = ET.ElementTree(root)
    tree.write("export/contacts.xml", encoding="utf-8")
    dom = MD.parse("export/contacts.xml")
   
    with open("export/contacts.xml.b", 'w') as f:
        f.write("<?xml version=\"1.0\" encoding=\"UTF-8\" ?>\n")
        content = (dom.toprettyxml(indent="  "))
        content_ = content.split("\n")
        f.write("\n".join(content_[1:]))
    

    return FileResponse(file_path, media_type='application/octet-stream', filename="contacts.xml")
        







async def resolve(data):
    await db.resolve(data, client)##manager.send_personal_message(f"You wrote: {data}", websocket)


@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await manager.connect(websocket)
    try:
        while True:
            data = await websocket.receive_text()
            await resolve(websocket, data)
            ##await manager.broadcast(f"Client #{client_id} says: {data}")
    except WebSocketDisconnect:
        manager.disconnect(websocket)
        ##await manager.broadcast(f"Client  left the chat")