paulin/bonobo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

extract run function

Browse Source
This commit is contained in:
Ben Rudolph
2018-03-23 17:37:11 -07:00
parent cb972d4a54
commit 3407994e00
1 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
bonobo/contrib/django/commands.py
View File

@ -44,13 +44,7 @@ class ETLCommand(BaseCommand):
def info(self, *args, **kwargs): def info(self, *args, **kwargs):
self.logger.info(*args, **kwargs) self.logger.info(*args, **kwargs)
def handle(self, *args, **options): def run(self, *args, **options):
_stdout_backup, _stderr_backup = self.stdout, self.stderr
self.stdout = OutputWrapper(ConsoleOutputPlugin._stdout, ending=CLEAR_EOL + '\n')
self.stderr = OutputWrapper(ConsoleOutputPlugin._stderr, ending=CLEAR_EOL + '\n')
self.stderr.style_func = lambda x: Fore.LIGHTRED_EX + Back.RED + '!' + Style.RESET_ALL + ' ' + x
with bonobo.parse_args(options) as options: with bonobo.parse_args(options) as options:
services = self.get_services() services = self.get_services()
graph_coll = self.get_graph(*args, **options) graph_coll = self.get_graph(*args, **options)
@ -65,4 +59,13 @@ class ETLCommand(BaseCommand):
print(term.lightblack(' ... return value: ' + str(result))) print(term.lightblack(' ... return value: ' + str(result)))
print() print()
def handle(self, *args, **options):
_stdout_backup, _stderr_backup = self.stdout, self.stderr
self.stdout = OutputWrapper(ConsoleOutputPlugin._stdout, ending=CLEAR_EOL + '\n')
self.stderr = OutputWrapper(ConsoleOutputPlugin._stderr, ending=CLEAR_EOL + '\n')
self.stderr.style_func = lambda x: Fore.LIGHTRED_EX + Back.RED + '!' + Style.RESET_ALL + ' ' + x
self.run(*args, **kwargs)
self.stdout, self.stderr = _stdout_backup, _stderr_backup self.stdout, self.stderr = _stdout_backup, _stderr_backup