from tkinter.messagebox import NO
from xml.etree import ElementTree as ET
from xml.dom import minidom


class CollectionConvertToXml:

    @classmethod  
    def get_xml_string(cls, element, default_encoding='utf-8'):
        try:  
            rough_string = ET.tostring(element, default_encoding)  
            reparsed = minidom.parseString(rough_string)  
            return reparsed.toprettyxml(indent="  " , encoding=default_encoding)  
        except Exception as e:  
            print("getXml_string:传入的节点不能正确转换为xml，请检查传入的节点是否正确")
            return ""

    @classmethod  
    def process_list(cls, data, ref):
        sub_element = ET.SubElement(ref, data.__class__.__name__)
        for item in data:
            if isinstance(item, list):
                result = cls.process_list(item)
            elif isinstance(item, dict):
                result = cls.process_dict(item)
                sub_element.append(result)
            else:
                # 以下是字符串
                if item is None:
                    continue
                k_ele = ET.Element(item)
                k_ele.text = item
                sub_element.append(k_ele)
        return sub_element

    @classmethod
    def process_dict(cls, data):
        # 进来首先创建一个同名的标签
        element = ET.Element(data.__class__.__name__)
        for k, v in data.items():
            if isinstance(v, list):
                result_list = cls.process_list(v, element)
            elif isinstance(v, dict):
                result = cls.process_dict(v)
                element.append(result)
            else:
                # 以下是字符串
                if v is None:
                    continue
                k_ele = ET.Element(k)
                k_ele.text = str(v)
                element.append(k_ele)
        return element

    @classmethod
    def main(cls, data):
        """"""
        element = ""
        if isinstance(data, list):
            element = cls.process_list(data)
        elif isinstance(data, dict):
            element = cls.process_dict(data)
        return element



class CustomDict(dict):
    __setattr__ = dict.__setitem__
    __getattr__ = dict.__getitem__

    @property
    def __dict__(self):
        return {k: v for k, v in self.items()}


class usp_react_crf_mb_setup_save(CustomDict):
    pass


class CRF_MB_XXK(CustomDict):
    pass


class CRF_MB_FZK(CustomDict):
    pass


class CRF_MB_FZKList(list):
    pass


if __name__ == "__main__":
    crf_mb_fzk = {"fz_id": "b2c90c09-1044-40d0-51b8-47966cca5f22",
      "fz_title": "未命名",
      "fz_description": "",
      "fz_imges_description": "",
      "fz_display_order": 1,
      "fz_display_style": "1",
      "fz_questions_count": 1,
      "mb_id": "193FEF1E-99D8-4188-A524-6A0B75D21721"}
    crf_mb_fzklist = [CRF_MB_FZK(crf_mb_fzk)]
    crf_mb_xxk = {"mb_id": "193FEF1E-99D8-4188-A524-6A0B75D21721",
                "mb_creator_name": "",
                "mb_creator_id": "",
                "mb_create_time": "2022-08-26 16:17:28",
                "mb_hospital_source": "",
                "mb_disease_type_id": "",
                "mb_disease_type_name": "",
                "mb_status": "已创建",
                "mb_title": "测试创建模板",
                "mb_description": "后端测试1",
                "mb_imges_description": "uploadImg\\1661525971081_238379.jpg"}
    new_dict = {"CRF_MB_XXK": CRF_MB_XXK(crf_mb_xxk), "CRF_MB_FZKList": CRF_MB_FZKList(crf_mb_fzklist)}
    personList=usp_react_crf_mb_setup_save(new_dict)
    xml = CollectionConvertToXml.main(personList)
    pretty_xml = CollectionConvertToXml.get_xml_string(xml, default_encoding="utf-16")
    print(pretty_xml.decode("utf-16"))
    # <?xml version="1.0" encoding="utf-16"?>
    # <usp_react_crf_mb_setup_save>
    # <CRF_MB_XXK>
    #     <mb_id>193FEF1E-99D8-4188-A524-6A0B75D21721</mb_id>
    #     <mb_creator_name/>
    #     <mb_creator_id/>
    #     <mb_create_time>2022-08-26 16:17:28</mb_create_time>
    #     <mb_hospital_source/>
    #     <mb_disease_type_id/>
    #     <mb_disease_type_name/>
    #     <mb_status>已创建</mb_status>
    #     <mb_title>测试创建模板</mb_title>
    #     <mb_description>后端测试1</mb_description>
    #     <mb_imges_description>uploadImg\1661525971081_238379.jpg</mb_imges_description>
    # </CRF_MB_XXK>
    # <CRF_MB_FZKList>
    #     <CRF_MB_FZK>
    #     <fz_id>b2c90c09-1044-40d0-51b8-47966cca5f22</fz_id>
    #     <fz_title>未命名</fz_title>
    #     <fz_description/>
    #     <fz_imges_description/>
    #     <fz_display_order>1</fz_display_order>
    #     <fz_display_style>1</fz_display_style>
    #     <fz_questions_count>1</fz_questions_count>
    #     <mb_id>193FEF1E-99D8-4188-A524-6A0B75D21721</mb_id>
    #     </CRF_MB_FZK>
    # </CRF_MB_FZKList>
    # </usp_react_crf_mb_setup_save>