Verified Commit 04fea99d authored by Peter Stanko's avatar Peter Stanko
Browse files

Updated yaml safe-dump - hotfix

parent 8c6a8051
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
__version__ = '0.5.2'
__version__ = '0.5.3'
+11 −1
Original line number Diff line number Diff line
import json
import sys
from typing import Dict

import click
import yaml
@@ -89,13 +90,22 @@ def entity_printer(entity, output_format: str = 'text'):
    print(f"\n{entity.__class__.__name__} info:\n")
    config = entity.config
    if output_format == 'text' or output_format == 'yaml':
        print(yaml.safe_dump(config, sort_keys=False))
        dump = dump_yaml(config)
        print(dump)
    elif output_format == 'json':
        print(json.dumps(config))
    else:
        print(config)


def dump_yaml(config: Dict) -> str:
    try:
        dump = yaml.safe_dump(config, sort_keys=False)
    except TypeError:
        dump = yaml.safe_dump(config)
    return dump


def edit_entity_using_editor(entity) -> dict:
    """Edits config using the editor
    Args: