style: switching from yapf to isort/black
This commit is contained in:
@ -42,6 +42,7 @@ def entrypoint(args=None):
|
||||
logger.exception('Error while loading command {}.'.format(ext.name))
|
||||
|
||||
from stevedore import ExtensionManager
|
||||
|
||||
mgr = ExtensionManager(namespace='bonobo.commands')
|
||||
mgr.map(register_extension)
|
||||
|
||||
|
||||
@ -41,6 +41,7 @@ class BaseGraphCommand(BaseCommand):
|
||||
Base class for CLI commands that depends on a graph definition, either from a file or from a module.
|
||||
|
||||
"""
|
||||
|
||||
required = True
|
||||
handler = None
|
||||
|
||||
|
||||
@ -1,7 +1,7 @@
|
||||
import bonobo
|
||||
from bonobo.commands import BaseCommand
|
||||
from bonobo.registry import READER, WRITER, default_registry
|
||||
from bonobo.util.resolvers import _resolve_transformations, _resolve_options
|
||||
from bonobo.util.resolvers import _resolve_options, _resolve_transformations
|
||||
|
||||
|
||||
class ConvertCommand(BaseCommand):
|
||||
@ -11,21 +11,14 @@ class ConvertCommand(BaseCommand):
|
||||
parser.add_argument(
|
||||
'--' + READER,
|
||||
'-r',
|
||||
help='Choose the reader factory if it cannot be detected from extension, or if detection is wrong.'
|
||||
help='Choose the reader factory if it cannot be detected from extension, or if detection is wrong.',
|
||||
)
|
||||
parser.add_argument(
|
||||
'--' + WRITER,
|
||||
'-w',
|
||||
help=
|
||||
'Choose the writer factory if it cannot be detected from extension, or if detection is wrong (use - for console pretty print).'
|
||||
)
|
||||
parser.add_argument(
|
||||
'--limit',
|
||||
'-l',
|
||||
type=int,
|
||||
help='Adds a Limit() after the reader instance.',
|
||||
default=None,
|
||||
help='Choose the writer factory if it cannot be detected from extension, or if detection is wrong (use - for console pretty print).',
|
||||
)
|
||||
parser.add_argument('--limit', '-l', type=int, help='Adds a Limit() after the reader instance.', default=None)
|
||||
parser.add_argument(
|
||||
'--transformation',
|
||||
'-t',
|
||||
@ -56,16 +49,16 @@ class ConvertCommand(BaseCommand):
|
||||
)
|
||||
|
||||
def handle(
|
||||
self,
|
||||
input_filename,
|
||||
output_filename,
|
||||
reader=None,
|
||||
reader_option=None,
|
||||
writer=None,
|
||||
writer_option=None,
|
||||
option=None,
|
||||
limit=None,
|
||||
transformation=None,
|
||||
self,
|
||||
input_filename,
|
||||
output_filename,
|
||||
reader=None,
|
||||
reader_option=None,
|
||||
writer=None,
|
||||
writer_option=None,
|
||||
option=None,
|
||||
limit=None,
|
||||
transformation=None,
|
||||
):
|
||||
reader_factory = default_registry.get_reader_factory_for(input_filename, format=reader)
|
||||
reader_kwargs = _resolve_options((option or []) + (reader_option or []))
|
||||
@ -75,13 +68,13 @@ class ConvertCommand(BaseCommand):
|
||||
writer_args = ()
|
||||
else:
|
||||
writer_factory = default_registry.get_writer_factory_for(output_filename, format=writer)
|
||||
writer_args = (output_filename, )
|
||||
writer_args = (output_filename,)
|
||||
writer_kwargs = _resolve_options((option or []) + (writer_option or []))
|
||||
|
||||
transformations = ()
|
||||
|
||||
if limit:
|
||||
transformations += (bonobo.Limit(limit), )
|
||||
transformations += (bonobo.Limit(limit),)
|
||||
|
||||
transformations += _resolve_transformations(transformation)
|
||||
|
||||
@ -92,8 +85,4 @@ class ConvertCommand(BaseCommand):
|
||||
writer_factory(*writer_args, **writer_kwargs),
|
||||
)
|
||||
|
||||
return bonobo.run(
|
||||
graph, services={
|
||||
'fs': bonobo.open_fs(),
|
||||
}
|
||||
)
|
||||
return bonobo.run(graph, services={'fs': bonobo.open_fs()})
|
||||
|
||||
@ -19,6 +19,7 @@ class RunCommand(BaseGraphCommand):
|
||||
|
||||
def parse_options(self, *, quiet=False, verbose=False, install=False, **options):
|
||||
from bonobo import settings
|
||||
|
||||
settings.QUIET.set_if_true(quiet)
|
||||
settings.DEBUG.set_if_true(verbose)
|
||||
self.install = install
|
||||
@ -65,4 +66,5 @@ def _install_requirements(requirements):
|
||||
# python interpreter.
|
||||
pip.utils.pkg_resources = importlib.reload(pip.utils.pkg_resources)
|
||||
import site
|
||||
|
||||
importlib.reload(site)
|
||||
|
||||
@ -32,10 +32,11 @@ class VersionCommand(BaseCommand):
|
||||
|
||||
def _format_version(mod, *, name=None, quiet=False):
|
||||
from bonobo.util.pkgs import bonobo_packages
|
||||
|
||||
args = {
|
||||
'name': name or mod.__name__,
|
||||
'version': mod.__version__,
|
||||
'location': bonobo_packages[name or mod.__name__].location
|
||||
'location': bonobo_packages[name or mod.__name__].location,
|
||||
}
|
||||
|
||||
if not quiet:
|
||||
|
||||
Reference in New Issue
Block a user