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

Adds argument parser support to django extension.

Browse Source
This commit is contained in:
Romain Dorgueil
2017-11-07 08:00:59 +01:00
parent afbc2efce0
commit 1d2916480a
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
bonobo/ext/django.py
View File

@ -1,17 +1,21 @@
from logging import getLogger from logging import getLogger
from colorama import Fore, Back, Style
from django.core.management.base import BaseCommand, OutputWrapper
import bonobo import bonobo
import bonobo.util import bonobo.util
from bonobo.plugins.console import ConsoleOutputPlugin from bonobo.plugins.console import ConsoleOutputPlugin
from bonobo.util.term import CLEAR_EOL from bonobo.util.term import CLEAR_EOL
from colorama import Fore, Back, Style
from django.core.management.base import BaseCommand, OutputWrapper
class ETLCommand(BaseCommand): class ETLCommand(BaseCommand):
GraphType = bonobo.Graph GraphType = bonobo.Graph
def create_parser(self, prog_name, subcommand):
return bonobo.get_argument_parser(
super().create_parser(prog_name, subcommand)
)
def create_or_update(self, model, *, defaults=None, save=True, **kwargs): def create_or_update(self, model, *, defaults=None, save=True, **kwargs):
""" """
Create or update a django model instance. Create or update a django model instance.
@ -63,10 +67,12 @@ class ETLCommand(BaseCommand):
self.stderr = OutputWrapper(ConsoleOutputPlugin._stderr, 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.stderr.style_func = lambda x: Fore.LIGHTRED_EX + Back.RED + '!' + Style.RESET_ALL + ' ' + x
result = bonobo.run( with bonobo.parse_args(options) as options:
self.get_graph(*args, **options), result = bonobo.run(
services=self.get_services(), self.get_graph(*args, **options),
) services=self.get_services(),
self.stdout = _stdout_backup )
self.stdout, self.stderr = _stdout_backup, _stderr_backup
return '\nReturn Value: ' + str(result) return '\nReturn Value: ' + str(result)