Allow to remove specific TPs

translate · Feb 10, 2017 · ae9e1ca · ae9e1ca
1 parent 9b20ba0
commit ae9e1ca
Showing 1 changed file with 23 additions and 6 deletions.
diff --git a/pootle/apps/pootle_app/management/commands/project_commands/__init__.py b/pootle/apps/pootle_app/management/commands/project_commands/__init__.py
@@ -168,6 +168,11 @@ class RemoveCommand(TPToolProjectSubCommand):
     help = """Remove project."""
 
     def add_arguments(self, parser):
+        parser.add_argument(
+            'source_project',
+            type=str,
+            help='Source Pootle project',
+        )
         parser.add_argument(
             "--force",
             action="store_true",
@@ -178,12 +183,24 @@ def add_arguments(self, parser):
         super(RemoveCommand, self).add_arguments(parser)
 
     def handle(self, *args, **options):
-        project = self.get_project(options['source_project'])
-        project_path = project.get_real_path()
-        project.delete()
-        if options['force']:
-            shutil.rmtree(project_path)
-        self.stdout.write('Project "%s" has been deleted.' % project)
+        if not options['languages']:
+            project = self.get_project(options['source_project'])
+            project_path = project.get_real_path()
+            project.delete()
+            if options['force']:
+                if os.path.exists(project_path):
+                    shutil.rmtree(project_path)
+            self.stdout.write('Project "%s" has been deleted.' % project)
+            return
+
+        tp_tool = self.get_tp_tool(options['source_project'])
+        tp_query = tp_tool.tp_qs.all()
+        if options['languages']:
+            tp_query = tp_query.filter(language__code__in=options['languages'])
+            for tp in tp_query:
+                tp.delete()
+                self.stdout.write('Translation project "%s" has been deleted.'
+                                  % tp)
 
 
 class UpdateCommand(TPToolProjectSubCommand):