有多种方法可以做你想做的事。如果您不想影响解析器的正常行为,您应该使用替代的and子类化RoundTripLoader
and 。但这需要注册所有构造函数和表示器,并且非常冗长。RoundTripConstructor
RoundTripConstructor
RoundTripRepresenter
如果您不关心能够在程序中稍后在原始功能中加载具有前导零的十六进制标量整数的其他 YAML 文档,您只需将新的构造函数和表示器添加到RoundTripConstructor
and RoundTripRepresenter
。
最简单的部分是根据值和宽度获取格式。如果您仍然使用,则不需要zfill()
也不需要:upper()
format
'0x{:0{}X}'.format(value, width)
做这项工作。
您的代码不起作用的主要原因是因为您的代码从不构造 a HexWInt
,因为RoundTripLoader
不知道它应该这样做。我也不会将宽度硬编码为 8,而是从输入(使用len()
)导出它,并保留它。
import sys
import ruamel.yaml
class HexWInt(ruamel.yaml.scalarint.ScalarInt):
def __new__(cls, value, width):
x = ruamel.yaml.scalarint.ScalarInt.__new__(cls, value)
x._width = width # keep the original width
return x
def __isub__(self, a):
return HexWInt(self - a, self._width)
def alt_construct_yaml_int(constructor, node):
# check for 0x0 starting hex integers
value_s = ruamel.yaml.compat.to_str(constructor.construct_scalar(node))
if not value_s.startswith('0x0'):
return constructor.construct_yaml_int(node)
return HexWInt(int(value_s[2:], 16), len(value_s[2:]))
ruamel.yaml.constructor.RoundTripConstructor.add_constructor(
u'tag:yaml.org,2002:int', alt_construct_yaml_int)
def represent_hexw_int(representer, data):
return representer.represent_scalar(u'tag:yaml.org,2002:int',
'0x{:0{}X}'.format(data, data._width))
ruamel.yaml.representer.RoundTripRepresenter.add_representer(HexWInt, represent_hexw_int)
yaml_text = """\
hexa: 0x0123ABCD
hexb: 0x02AD
"""
yaml = ruamel.yaml.YAML()
data = yaml.load(yaml_text)
data['hexc'] = HexWInt(0xa1, 8)
data['hexb'] -= 3
yaml.dump(data, sys.stdout)
HexWInt
存储值和宽度。alt_construct_yaml_int
除了construct_yaml_int
标量以0x0
. 它是add_constructor()
根据解析器完成的基于正常正则表达式的匹配注册的。表示器将值和宽度组合回一个字符串。上面的输出是:
hexa: 0x0123ABCD
hexb: 0x02AD
hexc: 0x000000A1
请注意,您不能执行以下操作:
data['hexb'] -= 3
as ScalarInt
(确实有方法__isub__
)不知道 width 属性。要使上述方法起作用,您必须实现适当的方法,ScalarInt
就像HexWInt
. 例如:
def __isub__(self, a):
return HexWInt(self - a, self._width)
上述的增强版本(也保留_
整数并支持八进制和二进制整数)被合并到ruamel.yaml>=0.14.7