blob: b7107cb3dbd294254af53069f0db8bc1b0d3a5c4 [file] [log] [blame] [edit]
#!/usr/bin/env python3
import sys
sys.path.insert(0, 'lib')
from yaml import *
ldr = SafeLoader()
def dice_constructor1(loader, node):
value = loader.construct_scalar(node)
a, b = map(int, value.split('d'))
return [a,b]
def dice_constructor2(loader, node):
value = loader.construct_scalar(node)
a, b = map(int, value.split('d'))
return [b,a]
s = """
- !dice 3d4
"""
add_constructor('!dice', dice_constructor1, SafeLoader)
print()
data = load(s, SafeLoader)
print('1) SafeLoader -> %s' % data)
data = load(s, ldr)
print('1) Instance -> %s' % data)
ldr.add_constructor('!dice', dice_constructor2)
print()
data = load(s, SafeLoader)
print('2) SafeLoader -> %s' % data)
data = load(s, ldr)
print('2) Instance -> %s' % data)
ldr.add_constructor('!dice', None)
print()
data = load(s, SafeLoader)
print('3) SafeLoader -> %s' % data)
data = load(s, ldr)
print('3) Instance -> %s' % data)