Improve the command 'dump_project'

taiga/export_import/management/commands/dump_project.py

@@ -18,25 +18,45 @@
 from django.core.management.base import BaseCommand, CommandError
 
 from taiga.projects.models import Project
-from taiga.export_import.renderers import ExportRenderer
 from taiga.export_import.services import render_project
 
-
+import os
-import resource
 
 
 class Command(BaseCommand):
-    args = '<project_slug project_slug ...>'
+    help = "Export projects to json"
-    help = 'Export a project to json'
+
-    renderer_context = {"indent": 4}
+    def add_arguments(self, parser):
-    renderer = ExportRenderer()
+        parser.add_argument("project_slugs",
+                            nargs="+",
+                            help="<project_slug project_slug ...>")
+
+        parser.add_argument("-d", "--dst_dir",
+                            action="store",
+                            dest="dst_dir",
+                            default="./",
+                            metavar="DIR",
+                            help="Directory to save the json files. ('./' by default)")
 
     def handle(self, *args, **options):
-        for project_slug in args:
+        dst_dir = options["dst_dir"]
+
+        if not os.path.exists(dst_dir):
+            raise CommandError("Directory {} does not exist.".format(dst_dir))
+
+        if not os.path.isdir(dst_dir):
+            raise CommandError("'{}' must be a directory, not a file.".format(dst_dir))
+
+        project_slugs = options["project_slugs"]
+
+        for project_slug in project_slugs:
             try:
                 project = Project.objects.get(slug=project_slug)
             except Project.DoesNotExist:
-                raise CommandError('Project "%s" does not exist' % project_slug)
+                raise CommandError("Project '{}' does not exist".format(project_slug))
 
-            with open('%s.json'%(project_slug), 'w') as outfile:
+            dst_file = os.path.join(dst_dir, "{}.json".format(project_slug))
-                render_project(project, outfile)
+            with open(src_file, "w") as f:
+                render_project(project, f)
+
+            print("-> Generate dump of project '{}' in '{}'".format(project.name, src_file))