工具编写思路

原创于 2023-01-09 18:33:06 发布 · 115 阅读

CC 4.0 BY-SA版权

文章标签：

该代码定义了两个函数，generate_xml和remove_empty_nodes。generate_xml函数从模板路径读取XML模板，用给定参数替换占位符，然后删除空节点，最后保存为XML文件。remove_empty_nodes函数递归地从XML元素树中移除没有文本内容的节点。

import os
from lxml import etree

def generate_xml(template_path, params):
# 读取模板文件
with open(template_path, 'r') as f:
template = f.read()
# 使用参数替换模板中的占位符
xml = template.format(**params)
# 将字符串解析为 xml 元素树
root = etree.fromstring(xml)
# 删除缺少参数的节点
remove_empty_nodes(root)
# 将 xml 元素树转换为字符串
xml = etree.tostring(root, encoding='unicode', pretty_print=True)
# 保存 xml 文件
filename = os.path.splitext(os.path.basename(template_path))[0] + '.xml'
with open(filename, 'w') as f:
f.write(xml)

def remove_empty_nodes(node):
# 递归删除缺少参数的节点
for child in node:
remove_empty_nodes(child)
if not node.text and len(node) == 0:
parent = node.getparent()
parent.remove(node)

13 条评论

金金金白水 2023.06.17
import com.alibaba.fastjson.JSON; import com.alibaba.fastjson.JSONArray; import com.alibaba.fastjson.JSONObject; public class Main { public static void main(String[] args) { JSONObject obj1 = JSON.parseObject("{\"person\":{\"name\":\"John\", \"age\":30}, \"city\":\"New York\"}"); JSONObject obj2 = JSON.parseObject("{\"city\":\"New York\", \"person\":{\"age\":30, \"name\":\"John\"}}"); if (compareJsonStructure(obj1, obj2)) { System.out.println("The structures of the two JSON objects are the same."); } else { System.out.println("The structures of the two JSON objects are different."); } } public static boolean compareJsonStructure(Object json1, Object json2) { if (json1 instanceof JSONObject && json2 instanceof JSONObject) { JSONObject jsonObj1 = (JSONObject) json1; JSONObject jsonObj2 = (JSONObject) json2; if (jsonObj1.size() != jsonObj2.size()) { return false; } for (String key : jsonObj1.keySet()) { if (!jsonObj2.containsKey(key)) { return false; } if (!compareJsonStructure(jsonObj1.get(key), jsonObj2.get(key))) { return false; } } return true; } else if (json1 instanceof JSONArray && json2 instanceof JSONArray) { JSONArray jsonArr1 = (JSONArray) json1; JSONArray jsonArr2 = (JSONArray) json2; if (jsonArr1.size() != jsonArr2.size()) { return false; } for (int i = 0; i < jsonArr1.size(); i++) { if (!compareJsonStructure(jsonArr1.get(i), jsonArr2.get(i))) { return false; } } return true; } else { return json1.getClass().equals(json2.getClass()); } } }

金金金白水 2023.02.15
@Entity public class Position { @Id @GeneratedValue(strategy = GenerationType.IDENTITY) private Long id; private String name; @ManyToOne private Position parent; @OneToMany(mappedBy = "parent") private List<Position> children; // 省略构造方法、getter和setter方法 }

金金金白水 2023.02.15
@Entity @Table(name = "my_table") public class MyTable { @Id @GeneratedValue(strategy = GenerationType.IDENTITY) private Long id; private String name; @ManyToOne(fetch = FetchType.LAZY) @JoinColumn(name = "parent_id") private MyTable parent; // getter and setter }

金金金白水 2023.01.12
import json # 载入 JSON 数据 data = json.loads('{"a": 1, "b": {"c": 2, "d": [3, 4, 5]}, "e": 6}') def remove_key(data, key_to_remove): if isinstance(data, list): return [v for v in (remove_key(v, key_to_remove) for v in data) if v] elif isinstance(data, dict): return {k: v for k, v in ((k, remove_key(v, key_to_remove)) for k, v in data.items()) if k != key_to_remove} else: return data # 删除所有名为 "b" 的元素 data = remove_key(data, 'b') # 输出结果 print(json.dumps(data))

金金金白水 2023.01.09
<?xml version="1.0" encoding="UTF-8"?> <xsl:stylesheet version="1.0" xmlns:xsl="http://www.w3.org/1999/XSL/Transform"> <xsl:param name="param1"/> <xsl:param name="param2"/> <xsl:template match="/"> <root> <element1> <xsl:attribute name="attr1"> <xsl:value-of select="$param1"/> </xsl:attribute> <xsl:attribute name="attr2"> <xsl:value-of select="$param2"/> </xsl:attribute> </element1> </root> </xsl:template> </xsl:stylesheet>

金金金白水 2023.01.09
from lxml import etree def read_template(template_path): # 加载xlst模板文件 xslt = etree.XSLT(etree.parse(template_path)) return xslt def generate_xml(xslt, params): # 存储xlst模板文件中使用的参数名称 param_names = {} for param in xslt.xpath("//xsl:param", namespaces=xslt.nsmap): param_name = param.get("name") if param_name: param_names[param_name] = True for key, value in params.items(): # 如果参数名称在xlst模板文件中使用过，则将参数值设置为该参数的值 if key in param_names: xslt = xslt(**{key: etree.XSLT.strparam(value)}) # 将生成的xml元素树写入文件 with open("output.xml", "w") as f: f.write(str(xslt)) def main(template_path, params): xslt = read_template(template_path) generate_xml(xslt, params) if __name__ == "__main__": # 使用示例 template_path = "template.xlst" params = {"param1": "value1", "param2": "value2"} main(template_path, params)

金金金白水 2023.01.09
from lxml import etree def read_template(template_path): # 加载xlst模板文件 xslt = etree.XSLT(etree.parse(template_path)) return xslt def generate_xml(xslt, params): # 存储xlst模板文件中使用的参数名称 param_names = {} for param in xslt.xpath("//xsl:param", namespaces=xslt.nsmap): param_name = param.get("name") if param_name: param_names[param_name] = True for key, value in params.items(): # 如果参数名称在xlst模板文件中使用过，则将参数值设置为该参数的值 if key in param_names: xslt = xslt(**{key: etree.XSLT.strparam(value)}) # 将生成的xml元素树写入文件 with open("output.xml", "w") as f: f.write(str(xslt)) def main(template_path, params): xslt = read_template(template_path) generate_xml(xslt, params) if __name__ == "__main__": # 使用示例 template_path = "template.xlst" params = {"param1": "value1", "param2": "value2"} main(template_path, params)

金金金白水 2023.01.09
param_names = {} for param in xslt.xpath("//xsl:param", namespaces=xslt.nsmap): param_name = param.get("name") if param_name: param_names[param_name] = True