===================================================================== Big day! V4 is finally live. This INCLUDES some BREAKING CHANGES to ZSH TYPE scwrypts! Please refer to the readme for upgrade details (more specifically docs/upgrade/v3-to-v4.md) Upgrade is SUPER EASY, so please take the time to do so. --- New Features ---------------------------------------------------- - zsh type scwrypts have an upgraded runstring to improve context setup and simplicity to the scwrypt-writer - scwrypts now publishes the package (scwrypts) to PyPi; this provides a simple way to invoke scwrypts from python-based environments as well as the entire scwrypts python library suite pip install scwrypts - scwrypts now publishes the package (scwrypts) to npm; this provides a simple way to invoke scwrypts from nodesjs environments npm install scwrypts --- Bug Fixes ------------------------------------------------------- - scwrypts runner prompts which use the zshbuiltin "read" now appropriately read input from tty, pipe, files, and user input - virtualenv refresh now loads and prepares the scwrypts virtual environments correctly --- Changes --------------------------------------------------------- - created the (-v, --log-level) scwrypts arguments as improvements of and replacements to the --verbose and --no-log flags - (-n) is now an alias for (--log-level 0) - (--no-log) is the same as (-n) for compatibility, but will be removed in 4.2 - zsh/lib/utils/io print functions now *interact with log-level* various log levels will now only display the appropriate console prints for the specified log level - zsh/lib/utils/io:INFO has been renamed to DEBUG to align with log-level output; please use DEBUG for debug messages and REMINDER for important user messages - created zsh/lib/utils/io:FZF_USER_INPUT as a *drop-in replacement* for the confusing FZF_HEAD and FZF_TAIL commands. Update by literally changing any instances of FZF_HEAD or FZF_TAIL with FZF_USER_INPUT - FZF_HEAD and FZF_TAIL will be removed in 4.2 - zsh/lib/utils/io:READ (and other zshbuiltin/read-based prompts) now accept a --force-user-input flag in case important checks should require an admin's approval. This flag will ensure that piped input and the `scwrypts -y` flag are ignored for the single prompt. - zsh/lib/utils/color has been updated to use color names which match the ANSI color names - zsh/hello-world has been reduced to a minimal example; this is to emphasize ease-of-use with v4 - zsh/sanity-check is a scwrypts/run testing helper and detailed starting reference (helpful since hello-world is now minimal) - various refactor, updates, and improvements to the scwrypts runner - migrated all zsh scwrypts and plugins to use v4 runner syntax - zsh - plugins/kubectl - plugins/ci - refactored py/lib into py/lib/scwrypts (PyPi)
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
import csv
|
|
import json
|
|
import yaml
|
|
|
|
|
|
def convert(input_stream, input_type, output_stream, output_type):
|
|
data = convert_input(input_stream, input_type)
|
|
write_output(output_stream, output_type, data)
|
|
|
|
|
|
def convert_input(stream, input_type):
|
|
supported_input_types = {'csv', 'json', 'yaml'}
|
|
|
|
if input_type not in supported_input_types:
|
|
raise ValueError(f'input_type "{input_type}" not supported; must be one of {supported_input_types}')
|
|
|
|
return {
|
|
'csv': _read_csv,
|
|
'json': _read_json,
|
|
'yaml': _read_yaml,
|
|
}[input_type](stream)
|
|
|
|
|
|
def write_output(stream, output_type, data):
|
|
supported_output_types = {'csv', 'json', 'yaml'}
|
|
|
|
if output_type not in supported_output_types:
|
|
raise ValueError(f'output_type "{output_type}" not supported; must be one of {supported_output_types}')
|
|
|
|
return {
|
|
'csv': _write_csv,
|
|
'json': _write_json,
|
|
'yaml': _write_yaml,
|
|
}[output_type](stream, data)
|
|
|
|
|
|
#####################################################################
|
|
|
|
def _read_csv(stream):
|
|
return [dict(line) for line in csv.DictReader(stream)]
|
|
|
|
def _write_csv(stream, data):
|
|
writer = csv.DictWriter(stream, fieldnames=list({
|
|
key
|
|
for dictionary in data
|
|
for key in dictionary.keys()
|
|
}))
|
|
|
|
writer.writeheader()
|
|
|
|
for value in data:
|
|
writer.writerow(value)
|
|
|
|
#####################################################################
|
|
|
|
def _read_json(stream):
|
|
data = json.loads(stream.read())
|
|
return data if isinstance(data, list) else [data]
|
|
|
|
def _write_json(stream, data):
|
|
stream.write(json.dumps(data))
|
|
|
|
#####################################################################
|
|
|
|
def _read_yaml(stream):
|
|
data = yaml.safe_load(stream)
|
|
return data if isinstance(data, list) else [data]
|
|
|
|
def _write_yaml(stream, data):
|
|
yaml.dump(data, stream, default_flow_style=False)
|