allow for overriding strategy

bonobo/contrib/django/commands.py

@@ -41,6 +41,9 @@ class ETLCommand(BaseCommand):
     def get_services(self):
         return {}
 
+    def get_strategy(self):
+        return None
+
     def info(self, *args, **kwargs):
         self.logger.info(*args, **kwargs)
 
@@ -48,6 +51,7 @@ class ETLCommand(BaseCommand):
         results = []
         with bonobo.parse_args(options) as options:
             services = self.get_services()
+            strategy = self.get_strategy()
             graph_coll = self.get_graph(*args, **options)
 
             if not isinstance(graph_coll, GeneratorType):
@@ -56,7 +60,7 @@ class ETLCommand(BaseCommand):
             for i, graph in enumerate(graph_coll):
                 assert isinstance(graph, bonobo.Graph), 'Invalid graph provided.'
                 print(term.lightwhite('{}. {}'.format(i + 1, graph.name)))
-                result = bonobo.run(graph, services=services)
+                result = bonobo.run(graph, services=services, strategy=strategy)
                 results.append(result)
                 print(term.lightblack(' ... return value: ' + str(result)))
                 print()