X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=n2vc%2Futils.py;fp=n2vc%2Futils.py;h=14ac5d1f63c1f95694d22809fe65c4ad218dcbc6;hb=7a0c8c9d1d1a8f985e7e4c0380a47d283874058e;hp=e8cf64dac123207f6023ca18ac035ebe9f1840e0;hpb=05ddb45012985d97d321499867b40eeee4b6b43d;p=osm%2FN2VC.git diff --git a/n2vc/utils.py b/n2vc/utils.py index e8cf64d..14ac5d1 100644 --- a/n2vc/utils.py +++ b/n2vc/utils.py @@ -17,6 +17,7 @@ from juju.machine import Machine from juju.application import Application from juju.action import Action from juju.unit import Unit +import yaml class N2VCDeploymentStatus(Enum): @@ -100,3 +101,33 @@ DB_DATA = Dict( ) } ) + + +def obj_to_yaml(obj: object) -> str: + """ + Converts object to yaml format + :return: yaml data + """ + # dump to yaml + dump_text = yaml.dump(obj, default_flow_style=False, indent=2) + # split lines + lines = dump_text.splitlines() + # remove !!python/object tags + yaml_text = "" + for line in lines: + index = line.find("!!python/object") + if index >= 0: + line = line[:index] + yaml_text += line + "\n" + return yaml_text + + +def obj_to_dict(obj: object) -> dict: + """ + Converts object to dictionary format + :return: dict data + """ + # convert obj to yaml + yaml_text = obj_to_yaml(obj) + # parse to dict + return yaml.load(yaml_text, Loader=yaml.Loader)