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")