#!/usr/bin/env python3.5
# -*- coding:utf8 -*-
from xml.etree.ElementTree import ElementTree,Element
def read_xml(in_path):
  """
  讀取并解析XML文件
  :param in_path: XML路徑
  :return:
  """
  tree = ElementTree()
  tree.parse(in_path)
  return tree
def write_xml(tree,out_path):
  """
  將XML文件寫(xiě)出
  :param tree:
  :param out_path:寫(xiě)出路徑
  :return:
  """
  tree.write(out_path,encoding="utf-8",xml_declaration=True)
def if_match(node,kv_map):
  """
  判斷某個(gè)節(jié)點(diǎn)是否包含所有傳入?yún)?shù)屬性
  :param node: 節(jié)點(diǎn)
  :param kv_map: 屬性及屬性值組成的MAP
  :return:
  """
  for key in kv_map:
    if node.get(key) != kv_map.get(key):
      return False
    return True
def find_nodes(tree,path):
  """
  查找某個(gè)路徑匹配的所有節(jié)點(diǎn)
  :param tree:XML樹(shù)
  :param path:節(jié)點(diǎn)路徑
  :return:
  """
  return tree.findall(path)
def get_node_by_keyvalue(nodelist,kv_map):
  """
  根據(jù)屬性及屬性值定位符合的節(jié)點(diǎn)，返回節(jié)點(diǎn)
  :param nodelist: 節(jié)點(diǎn)列表
  :param kv_map: 匹配屬性及屬性值MAP
  :return:
  """
  result_nodes =[]
  for node in nodelist:
    if if_match(node,kv_map):
      result_nodes.append(node)
  return result_nodes
def change_node_properties(nodelist,kv_map,is_delete =False):
  """
  修改、增加、刪除 節(jié)點(diǎn)的屬性及屬性值
  :param nodelist: 節(jié)點(diǎn)列表
  :param kv_map: 屬性及屬性值MAP
  :param is_delete:
  :return:
  """
  for node in nodelist:
    for key in kv_map:
      if is_delete:
        if key in node.attrib:
          del node.attrib[key]
      else:
        node.set(key,kv_map.get(key))
def change_node_text(nodelist,text,is_add=False,is_delete=False):
  """
  改變、增加、刪除一個(gè)節(jié)點(diǎn)的文本
  :param nodelist: 節(jié)點(diǎn)列表
  :param text: 更新后的文本
  :param is_add:
  :param is_delete:
  :return:
  """
  for node in nodelist:
    if is_add:
      node.text += text
    elif is_delete:
      node.text = ""
    else:
      node.text = text
def create_node(tag,property_map,content):
  """
  新造一個(gè)節(jié)點(diǎn)
  :param tag:節(jié)點(diǎn)標(biāo)簽
  :param property_map:屬性及屬性值MAP
  :param content: 節(jié)點(diǎn)閉合標(biāo)簽里的文件內(nèi)容
  :return:新節(jié)點(diǎn)
  """
  element =Element(tag,property_map)
  element.text =content
  return element
def add_child_node(nodelist,element):
  """
  給一個(gè)節(jié)點(diǎn)添加子節(jié)點(diǎn)
  :param nodelist: 節(jié)點(diǎn)列表
  :param element: 子節(jié)點(diǎn)
  :return:
  """
  for node in nodelist:
    node.append(element)
def del_node_by_tagkeyvalue(nodelist,tag,kv_map):
  """
  同過(guò)屬性及屬性值定位一個(gè)節(jié)點(diǎn)，并刪除之
  :param nodelist: 父節(jié)點(diǎn)列表
  :param tag: 子節(jié)點(diǎn)標(biāo)簽
  :param kv_map: 屬性及屬性值列表
  :return:
  """
  for parent_node in nodelist:
    childree = parent_node.getchildren()
    for child in childree:
      if child.tag == tag and if_match(child,kv_map):
        parent_node.remove(child)
def config_file_rw(file):
  """
  對(duì)XML配置文件進(jìn)行修復(fù)以滿足適應(yīng)IIS
  :param file: 目標(biāo)文件
  :return:
  """
  import re
  x =re.compile("<ns0:")
  y = re.compile("</ns0:")
  z = re.compile("xmlns:ns0")
  with open(file,"r",encoding="utf-8") as f:
    txt = f.readlines()
    for i, line in enumerate(txt):
      if x.search(line):
        txt[i] = x.sub("<", line)
      elif y.search(line):
        txt[i] = y.sub("</", line)
      elif z.search(line):
        txt[i] = "<configuration>\n"
  with open(file,"w",encoding="utf-8") as fw:
    fw.writelines(txt)
    fw.close()
    print("配置文件%s，修改成功！"%file)
if __name__ == "__main__":
  #1. 讀取xml文件
  tree = read_xml("web.config")
  # 找到父節(jié)點(diǎn)
  print()
  nodes = find_nodes(tree, "connectionStrings/")
  # 通過(guò)屬性準(zhǔn)確定位子節(jié)點(diǎn)
  result_nodes =(get_node_by_keyvalue(nodes,{"name":"strConnection_HuaChenShiYou"}))
  # 修改節(jié)點(diǎn)屬性
  change_node_properties(result_nodes,{"connectionString":"UwreW/Obe4fGk2CFW4uE6iZWaPAVn0nePXIrtNRApxEGLv6SHetFg6b%2BMLFhg9myAr2EI2b3FgHtKHOKVcjz5DPoV8%2BMAvpzqlEZP2JZqrVEofP3AulFUZbTLfmndYFRqIytlxSCeHr2A79EWHH9/xg0eSgsdvWd"})
  # #2. 屬性修改
  # #A. 找到父節(jié)點(diǎn)
  # nodes = find_nodes(tree, "processers/processer")
  # #B. 通過(guò)屬性準(zhǔn)確定位子節(jié)點(diǎn)
  # result_nodes = get_node_by_keyvalue(nodes, {"name":"BProcesser"})
  # #C. 修改節(jié)點(diǎn)屬性
  # change_node_properties(result_nodes, {"age": "1"})
  # #D. 刪除節(jié)點(diǎn)屬性
  # change_node_properties(result_nodes, {"value":""}, True)
  #
  # #3. 節(jié)點(diǎn)修改
  # #A.新建節(jié)點(diǎn)
  # a = create_node("person", {"age":"15","money":"200000"}, "this is the firest content")
  # #B.插入到父節(jié)點(diǎn)之下
  # add_child_node(result_nodes, a)
  #
  # #4. 刪除節(jié)點(diǎn)
  #  #定位父節(jié)點(diǎn)
  # del_parent_nodes = find_nodes(tree, "processers/services/service")
  #  #準(zhǔn)確定位子節(jié)點(diǎn)并刪除之
  # target_del_node = del_node_by_tagkeyvalue(del_parent_nodes, "chain", {"sequency" : "chain1"})
  #
  # #5. 修改節(jié)點(diǎn)文本
  #  #定位節(jié)點(diǎn)
  # text_nodes = get_node_by_keyvalue(find_nodes(tree, "processers/services/service/chain"), {"sequency":"chain3"})
  # change_node_text(text_nodes, "new text")
  #
  # #6. 輸出到結(jié)果文件
  write_xml(tree, "new.config")
  config_file_rw("new.config")