Refactor list_journals()

Create three new methods for each journal list format.
- JSON, YAML, STDOUT
This commit is contained in:
apainintheneck 2022-09-18 10:08:19 -07:00
parent d706865850
commit ebbf66b0e2

View file

@ -25,30 +25,48 @@ def deprecated_cmd(old_cmd, new_cmd, callback=None, **kwargs):
callback(**kwargs) callback(**kwargs)
def journal_list_to_json(journal_list):
import json
return json.dumps(journal_list)
def journal_list_to_yaml(journal_list):
from io import StringIO
from ruamel.yaml import YAML
output = StringIO()
YAML().dump(journal_list, output)
return output.getvalue()
def journal_list_to_stdout(journal_list):
result = f"Journals defined in config ({journal_list['config_path']})\n"
ml = min(max(len(k) for k in journal_list["journals"]), 20)
for journal, cfg in journal_list["journals"].items():
result += " * {:{}} -> {}\n".format(
journal, ml, cfg["journal"] if isinstance(cfg, dict) else cfg
)
return result
def list_journals(configuration, format=None): def list_journals(configuration, format=None):
from jrnl import config from jrnl import config
"""List the journals specified in the configuration file""" """List the journals specified in the configuration file"""
journal_list = {
"config_path": config.get_config_path(),
"journals": configuration["journals"],
}
if format == "json": if format == "json":
import json return journal_list_to_json(journal_list)
return json.dumps(configuration["journals"])
elif format == "yaml": elif format == "yaml":
from io import StringIO return journal_list_to_yaml(journal_list)
from ruamel.yaml import YAML
output = StringIO()
YAML().dump(configuration["journals"], output)
return output.getvalue()
else: else:
result = f"Journals defined in config ({config.get_config_path()})\n" return journal_list_to_stdout(journal_list)
ml = min(max(len(k) for k in configuration["journals"]), 20)
for journal, cfg in configuration["journals"].items():
result += " * {:{}} -> {}\n".format(
journal, ml, cfg["journal"] if isinstance(cfg, dict) else cfg
)
return result
def print_msg(msg: Message, **kwargs) -> Union[None, str]: def print_msg(msg: Message, **kwargs) -> Union[None, str]: