-
Notifications
You must be signed in to change notification settings - Fork 121
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
bumped version to 0.3.39 and updated README
- Loading branch information
Showing
10 changed files
with
285 additions
and
255 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,240 @@ | ||
| import sys | ||
|
|
||
| from pyhocon import ConfigFactory | ||
| from pyhocon.config_tree import ConfigTree | ||
| from pyhocon.config_tree import NoneValue | ||
|
|
||
| try: | ||
| basestring | ||
| except NameError: | ||
| basestring = str | ||
|
|
||
|
|
||
| class HOCONConverter(object): | ||
| @classmethod | ||
| def to_json(cls, config, compact=False, indent=2, level=0): | ||
| """Convert HOCON input into a JSON output | ||
| :return: JSON string representation | ||
| :type return: basestring | ||
| """ | ||
| lines = "" | ||
| if isinstance(config, ConfigTree): | ||
| if len(config) == 0: | ||
| lines += '{}' | ||
| else: | ||
| lines += '{\n' | ||
| bet_lines = [] | ||
| for key, item in config.items(): | ||
| bet_lines.append('{indent}"{key}": {value}'.format( | ||
| indent=''.rjust((level + 1) * indent, ' '), | ||
| key=key.strip('"'), # for dotted keys enclosed with "" to not be interpreted as nested key | ||
| value=cls.to_json(item, compact, indent, level + 1)) | ||
| ) | ||
| lines += ',\n'.join(bet_lines) | ||
| lines += '\n{indent}}}'.format(indent=''.rjust(level * indent, ' ')) | ||
| elif isinstance(config, list): | ||
| if len(config) == 0: | ||
| lines += '[]' | ||
| else: | ||
| lines += '[\n' | ||
| bet_lines = [] | ||
| for item in config: | ||
| bet_lines.append('{indent}{value}'.format( | ||
| indent=''.rjust((level + 1) * indent, ' '), | ||
| value=cls.to_json(item, compact, indent, level + 1)) | ||
| ) | ||
| lines += ',\n'.join(bet_lines) | ||
| lines += '\n{indent}]'.format(indent=''.rjust(level * indent, ' ')) | ||
| elif isinstance(config, basestring): | ||
| lines = '"{value}"'.format(value=config.replace('\n', '\\n').replace('"', '\\"')) | ||
| elif config is None or isinstance(config, NoneValue): | ||
| lines = 'null' | ||
| elif config is True: | ||
| lines = 'true' | ||
| elif config is False: | ||
| lines = 'false' | ||
| else: | ||
| lines = str(config) | ||
| return lines | ||
|
|
||
| @classmethod | ||
| def to_hocon(cls, config, compact=False, indent=2, level=0): | ||
| """Convert HOCON input into a HOCON output | ||
| :return: JSON string representation | ||
| :type return: basestring | ||
| """ | ||
| lines = "" | ||
| if isinstance(config, ConfigTree): | ||
| if len(config) == 0: | ||
| lines += '{}' | ||
| else: | ||
| if level > 0: # don't display { at root level | ||
| lines += '{\n' | ||
| bet_lines = [] | ||
|
|
||
| for key, item in config.items(): | ||
| if compact: | ||
| full_key = key | ||
| while isinstance(item, ConfigTree) and len(item) == 1: | ||
| key, item = next(iter(item.items())) | ||
| full_key += '.' + key | ||
| else: | ||
| full_key = key | ||
|
|
||
| bet_lines.append('{indent}{key}{assign_sign} {value}'.format( | ||
| indent=''.rjust(level * indent, ' '), | ||
| key=full_key, | ||
| assign_sign='' if isinstance(item, dict) else ' =', | ||
| value=cls.to_hocon(item, compact, indent, level + 1)) | ||
| ) | ||
| lines += '\n'.join(bet_lines) | ||
|
|
||
| if level > 0: # don't display { at root level | ||
| lines += '\n{indent}}}'.format(indent=''.rjust((level - 1) * indent, ' ')) | ||
| elif isinstance(config, list): | ||
| if len(config) == 0: | ||
| lines += '[]' | ||
| else: | ||
| lines += '[\n' | ||
| bet_lines = [] | ||
| for item in config: | ||
| bet_lines.append('{indent}{value}'.format(indent=''.rjust(level * indent, ' '), | ||
| value=cls.to_hocon(item, compact, indent, level + 1))) | ||
| lines += '\n'.join(bet_lines) | ||
| lines += '\n{indent}]'.format(indent=''.rjust((level - 1) * indent, ' ')) | ||
| elif isinstance(config, basestring): | ||
| if '\n' in config: | ||
| lines = '"""{value}"""'.format(value=config) # multilines | ||
| else: | ||
| lines = '"{value}"'.format(value=config.replace('\n', '\\n').replace('"', '\\"')) | ||
| elif config is None or isinstance(config, NoneValue): | ||
| lines = 'null' | ||
| elif config is True: | ||
| lines = 'true' | ||
| elif config is False: | ||
| lines = 'false' | ||
| else: | ||
| lines = str(config) | ||
| return lines | ||
|
|
||
| @classmethod | ||
| def to_yaml(cls, config, compact=False, indent=2, level=0): | ||
| """Convert HOCON input into a YAML output | ||
| :return: YAML string representation | ||
| :type return: basestring | ||
| """ | ||
| lines = "" | ||
| if isinstance(config, ConfigTree): | ||
| if len(config) > 0: | ||
| if level > 0: | ||
| lines += '\n' | ||
| bet_lines = [] | ||
| for key, item in config.items(): | ||
| bet_lines.append('{indent}{key}: {value}'.format( | ||
| indent=''.rjust(level * indent, ' '), | ||
| key=key.strip('"'), # for dotted keys enclosed with "" to not be interpreted as nested key, | ||
| value=cls.to_yaml(item, compact, indent, level + 1)) | ||
| ) | ||
| lines += '\n'.join(bet_lines) | ||
| elif isinstance(config, list): | ||
| config_list = [line for line in config if line is not None] | ||
| if len(config_list) == 0: | ||
| lines += '[]' | ||
| else: | ||
| lines += '\n' | ||
| bet_lines = [] | ||
| for item in config_list: | ||
| bet_lines.append('{indent}- {value}'.format(indent=''.rjust(level * indent, ' '), | ||
| value=cls.to_yaml(item, compact, indent, level + 1))) | ||
| lines += '\n'.join(bet_lines) | ||
| elif isinstance(config, basestring): | ||
| # if it contains a \n then it's multiline | ||
| lines = config.split('\n') | ||
| if len(lines) == 1: | ||
| lines = config | ||
| else: | ||
| lines = '|\n' + '\n'.join([line.rjust(level * indent, ' ') for line in lines]) | ||
| elif config is None or isinstance(config, NoneValue): | ||
| lines = 'null' | ||
| elif config is True: | ||
| lines = 'true' | ||
| elif config is False: | ||
| lines = 'false' | ||
| else: | ||
| lines = str(config) | ||
| return lines | ||
|
|
||
| @classmethod | ||
| def to_properties(cls, config, compact=False, indent=2, key_stack=[]): | ||
| """Convert HOCON input into a .properties output | ||
| :return: .properties string representation | ||
| :type return: basestring | ||
| :return: | ||
| """ | ||
|
|
||
| def escape_value(value): | ||
| return value.replace('=', '\\=').replace('!', '\\!').replace('#', '\\#').replace('\n', '\\\n') | ||
|
|
||
| stripped_key_stack = [key.strip('"') for key in key_stack] | ||
| lines = [] | ||
| if isinstance(config, ConfigTree): | ||
| for key, item in config.items(): | ||
| if item is not None: | ||
| lines.append(cls.to_properties(item, compact, indent, stripped_key_stack + [key])) | ||
| elif isinstance(config, list): | ||
| for index, item in enumerate(config): | ||
| if item is not None: | ||
| lines.append(cls.to_properties(item, compact, indent, stripped_key_stack + [str(index)])) | ||
| elif isinstance(config, basestring): | ||
| lines.append('.'.join(stripped_key_stack) + ' = ' + escape_value(config)) | ||
| elif config is True: | ||
| lines.append('.'.join(stripped_key_stack) + ' = true') | ||
| elif config is False: | ||
| lines.append('.'.join(stripped_key_stack) + ' = false') | ||
| elif config is None or isinstance(config, NoneValue): | ||
| pass | ||
| else: | ||
| lines.append('.'.join(stripped_key_stack) + ' = ' + str(config)) | ||
| return '\n'.join([line for line in lines if len(line) > 0]) | ||
|
|
||
| @classmethod | ||
| def convert(cls, config, output_format='json', indent=2, compact=False): | ||
| converters = { | ||
| 'json': cls.to_json, | ||
| 'properties': cls.to_properties, | ||
| 'yaml': cls.to_yaml, | ||
| 'hocon': cls.to_hocon, | ||
| } | ||
|
|
||
| if output_format in converters: | ||
| return converters[output_format](config, compact, indent) | ||
| else: | ||
| raise Exception("Invalid format '{format}'. Format must be 'json', 'properties', 'yaml' or 'hocon'".format( | ||
| format=output_format)) | ||
|
|
||
| @classmethod | ||
| def convert_from_file(cls, input_file=None, output_file=None, output_format='json', indent=2, compact=False): | ||
| """Convert to json, properties or yaml | ||
| :param input_file: input file, if not specified stdin | ||
| :param output_file: output file, if not specified stdout | ||
| :param output_format: json, properties or yaml | ||
| :return: json, properties or yaml string representation | ||
| """ | ||
|
|
||
| if input_file is None: | ||
| content = sys.stdin.read() | ||
| config = ConfigFactory.parse_string(content) | ||
| else: | ||
| config = ConfigFactory.parse_file(input_file) | ||
|
|
||
| res = cls.convert(config, output_format, indent, compact) | ||
| if output_file is None: | ||
| print(res) | ||
| else: | ||
| with open(output_file, "w") as fd: | ||
| fd.write(res) |
Oops, something went wrong.