misc: migrate to mondrian 0.8, humanize cli commands
This commit is contained in:
@ -7,6 +7,7 @@ from contextlib import contextmanager
|
||||
import bonobo.util.environ
|
||||
from bonobo.util import get_name
|
||||
from bonobo.util.environ import get_argument_parser, parse_args
|
||||
from mondrian import humanizer
|
||||
|
||||
|
||||
class BaseCommand:
|
||||
@ -60,7 +61,8 @@ class BaseGraphCommand(BaseCommand):
|
||||
return options
|
||||
|
||||
def handle(self, file, mod, **options):
|
||||
options = self.parse_options(**options)
|
||||
with humanizer.humanize():
|
||||
options = self.parse_options(**options)
|
||||
with self.read(file, mod, **options) as (graph, graph_execution_options, options):
|
||||
return self.do_handle(graph, **graph_execution_options, **options)
|
||||
|
||||
|
||||
@ -2,6 +2,7 @@ import bonobo
|
||||
from bonobo.commands import BaseCommand
|
||||
from bonobo.registry import READER, WRITER, default_registry
|
||||
from bonobo.util.resolvers import _resolve_options, _resolve_transformations
|
||||
from mondrian import humanizer
|
||||
|
||||
|
||||
class ConvertCommand(BaseCommand):
|
||||
@ -48,6 +49,7 @@ class ConvertCommand(BaseCommand):
|
||||
help="Add a named option to the writer factory.",
|
||||
)
|
||||
|
||||
@humanizer.humanize()
|
||||
def handle(
|
||||
self,
|
||||
input_filename,
|
||||
@ -60,6 +62,8 @@ class ConvertCommand(BaseCommand):
|
||||
limit=None,
|
||||
transformation=None,
|
||||
):
|
||||
graph = bonobo.Graph()
|
||||
|
||||
reader_factory = default_registry.get_reader_factory_for(input_filename, format=reader)
|
||||
reader_kwargs = _resolve_options((option or []) + (reader_option or []))
|
||||
|
||||
@ -78,7 +82,6 @@ class ConvertCommand(BaseCommand):
|
||||
|
||||
transformations += _resolve_transformations(transformation)
|
||||
|
||||
graph = bonobo.Graph()
|
||||
graph.add_chain(
|
||||
reader_factory(input_filename, **reader_kwargs),
|
||||
*transformations,
|
||||
|
||||
@ -3,6 +3,7 @@ import os
|
||||
from jinja2 import Environment, FileSystemLoader
|
||||
|
||||
from bonobo.commands import BaseCommand
|
||||
from mondrian import humanizer
|
||||
|
||||
|
||||
class InitCommand(BaseCommand):
|
||||
@ -30,7 +31,11 @@ class InitCommand(BaseCommand):
|
||||
with open(filename, "w+") as f:
|
||||
f.write(template.render(name=name))
|
||||
|
||||
self.logger.info("Generated {} using template {!r}.".format(filename, template_name))
|
||||
print(
|
||||
humanizer.Success(
|
||||
"Generated {} using template {!r}.".format(filename, template_name)
|
||||
)
|
||||
)
|
||||
|
||||
def create_package(self, *, filename):
|
||||
_, ext = os.path.splitext(filename)
|
||||
@ -52,18 +57,23 @@ class InitCommand(BaseCommand):
|
||||
self.logger.info('Generated "{}" package with medikit.'.format(package_name))
|
||||
self.create_file_from_template(template="default", filename=os.path.join(filename, package_name, "__main__.py"))
|
||||
|
||||
print('Your "{}" package has been created.'.format(package_name))
|
||||
print()
|
||||
print("Install it...")
|
||||
print()
|
||||
print(" pip install --editable {}".format(filename))
|
||||
print()
|
||||
print("Then maybe run the example...")
|
||||
print()
|
||||
print(" python -m {}".format(package_name))
|
||||
print()
|
||||
print("Enjoy!")
|
||||
print(
|
||||
humanizer.Success(
|
||||
'Package "{}" has been created.'.format(package_name),
|
||||
'',
|
||||
"Install it...",
|
||||
'',
|
||||
" $ `pip install --editable {}`".format(filename),
|
||||
'',
|
||||
"Then maybe run the example...",
|
||||
'',
|
||||
" $ `python -m {}`".format(package_name),
|
||||
'',
|
||||
"Enjoy!"
|
||||
)
|
||||
)
|
||||
|
||||
@humanizer.humanize()
|
||||
def handle(self, *, template, filename, package=False, force=False):
|
||||
if os.path.exists(filename) and not force:
|
||||
raise FileExistsError("Target filename already exists, use --force to override.")
|
||||
|
||||
@ -1,4 +1,5 @@
|
||||
from bonobo.commands import BaseCommand
|
||||
from mondrian import humanizer
|
||||
|
||||
|
||||
def get_versions(*, all=False, quiet=None):
|
||||
@ -21,6 +22,7 @@ def get_versions(*, all=False, quiet=None):
|
||||
|
||||
|
||||
class VersionCommand(BaseCommand):
|
||||
@humanizer.humanize()
|
||||
def handle(self, *, all=False, quiet=False):
|
||||
for line in get_versions(all=all, quiet=quiet):
|
||||
print(line)
|
||||
|
||||
Reference in New Issue
Block a user