Python配置XML配置文件

# !/usr/bin/env python
# -*- coding: utf-8 -*-
# author:Rache
# date:2020/3/19 22:02
# desc: Tomcat組件配置初始化，適用版本 apache-tomcat-9.0.33

from xml.etree.ElementTree import ElementTree, Element

def read_xml(in_path):
	'''讀取並解析xml文件
	  in_path: xml路徑
	  return: ElementTree'''
	tree = ElementTree()
	tree.parse(in_path)
	return tree


def write_xml(tree, out_path):
	'''將xml文件寫出
	  tree: xml樹
	  out_path: 寫出路徑'''
	tree.write(out_path, encoding="utf-8", xml_declaration=True)


def if_match(node, kv_map):
	'''判斷某個節點是否包含所有傳入參數屬性
	  node: 節點
	  kv_map: 屬性及屬性值組成的map'''
	for key in kv_map:
		if node.get(key) != kv_map.get(key):
			return False
	return True


# ---------------search -----
def find_nodes(tree, path):
	'''查找某個路徑匹配的所有節點
	  tree: xml樹
	  path: 節點路徑'''
	return tree.findall(path)


def get_node_by_keyvalue(nodelist, kv_map):
	'''根據屬性及屬性值定位符合的節點，返回節點
	  nodelist: 節點列表
	  kv_map: 匹配屬性及屬性值map'''
	result_nodes = []
	for node in nodelist:
		if if_match(node, kv_map):
			result_nodes.append(node)
	return result_nodes


# ---------------change -----
def change_node_properties(nodelist, kv_map, is_delete=False):
	'''修改/增加 /刪除 節點的屬性及屬性值
	  nodelist: 節點列表
	  kv_map:屬性及屬性值map'''
	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):
	'''改變/增加/刪除一個節點的文本
	  nodelist:節點列表
	  text : 更新後的文本'''
	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):
	'''新造一個節點
	  tag:節點標籤
	  property_map:屬性及屬性值map
	  content: 節點閉合標籤裏的文本內容
	  return 新節點'''
	element = Element(tag, property_map)
	element.text = content
	return element


def add_child_node(nodelist, element):
	'''給一個節點添加子節點
	  nodelist: 節點列表
	  element: 子節點'''
	for node in nodelist:
		node.append(element)


def del_node_by_tagkeyvalue(nodelist, tag, kv_map):
	'''同過屬性及屬性值定位一個節點，並刪除之
	  nodelist: 父節點列表
	  tag:子節點標籤
	  kv_map: 屬性及屬性值列表'''
	for parent_node in nodelist:
		children = parent_node.getchildren()
		for child in children:
			if child.tag == tag and if_match(child, kv_map):
				parent_node.remove(child)


if __name__ == "__main__":
	# 1. 讀取xml文件
	tree = read_xml("./server.xml")

	# 2.遍歷節點測試：
	root_node=tree.getroot()
	# print("根節點："+root_node.tag)
	# print("根節點屬性：",root_node.attrib)
	#
	# for child_node in root_node.iter('Service'):
	# 	print(child_node.tag,child_node.attrib)


	# 3.Server節點修改
	server_nodes = get_node_by_keyvalue(find_nodes(tree, "Server"), {"shutdown": "SHUTDOWN"})
	change_node_properties(server_nodes, {"port": "8010"}) # shutdown 默認端口8005改爲8010

	# 3.Connector節點修改,增加些屬性
	# A. 找到父節點
	root_node = tree.getroot()
	for node in root_node.iter('Connector'):
		# 節點的標籤名稱和內容
		print(node.tag, node.attrib)
	# B. 通過屬性準確定位子節點
	connector_nodes = get_node_by_keyvalue(root_node.iter('Connector'), {"port": "8080"})
	change_node_properties(connector_nodes, {"URIEncoding": "UTF-8","minSpareThreads":"25","maxSpareThreads":"300","maxThreads":"1000","acceptCount":"1000","server":"ni chou sha"})

	# 3. Host節點修改，關閉熱部署，增加默認項目路徑
	host_nodes = get_node_by_keyvalue(root_node.iter('Host'), {"name": "localhost"})
	change_node_properties(host_nodes, {"autoDeploy": "false"})
	# A.新建節點
	a = create_node("Context", {"path": "","docBase":"yunmas_biz","reloadable":"true"}, "")
	# B.插入到父節點之下
	add_child_node(host_nodes, a)


	# 6. 輸出到結果文件
	write_xml(tree, "./out.xml")