Skip to content

Commit

Permalink
Making sure all Commands use run in __call__.
Browse files Browse the repository at this point in the history
This way subclasses don't have to also set `__call__ = run`,
they can just over-ride `run`.
  • Loading branch information
dhermes committed Sep 18, 2017
1 parent d2d44a0 commit 9ba4d98
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions nox/command.py
Expand Up @@ -112,7 +112,8 @@ def run(self, path_override=None, env_fallback=None, **kwargs):
logger.error('Interrupted...')
raise

__call__ = run
def __call__(self, *args, **kwargs):
return self.run(*args, **kwargs)


class FunctionCommand(object):
Expand Down Expand Up @@ -141,7 +142,8 @@ def run(self):

raise CommandFailed(e)

__call__ = run
def __call__(self, *args, **kwargs):
return self.run(*args, **kwargs)

def __str__(self):
return '{}(args={!r}, kwargs={!r})'.format(
Expand All @@ -165,7 +167,8 @@ def __init__(self, deps):
def run(self, venv):
venv.install(*self.deps)

__call__ = run
def __call__(self, *args, **kwargs):
return self.run(*args, **kwargs)

def __str__(self):
return ' '.join(self.deps)
Expand Down

0 comments on commit 9ba4d98

Please sign in to comment.