X-Git-Url: https://osm.etsi.org/gitweb/?p=osm%2FN2VC.git;a=blobdiff_plain;f=n2vc%2Futils.py;h=6e0f2c053a2029b576bba719ae15d20ca8615b1f;hp=16a47332cf284cba606dc27b5af3e281d902e15a;hb=b816d822dcc61a709b22f475085c3f94dceca61d;hpb=cdf0b8e857e1373410e61bc72cd1a9f5c1ab7ca7 diff --git a/n2vc/utils.py b/n2vc/utils.py index 16a4733..6e0f2c0 100644 --- a/n2vc/utils.py +++ b/n2vc/utils.py @@ -15,6 +15,7 @@ import base64 import re import binascii +import yaml from enum import Enum from juju.machine import Machine from juju.application import Application @@ -120,3 +121,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)