feat(api/commands)!: added project base command

Author: c0rydoras

api/outdated/commands.py

@@ -1,10 +1,48 @@
-from asyncio import run
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
 
 from django.core.management.base import BaseCommand
 
+from outdated.outdated.models import Project
+
+if TYPE_CHECKING: # pragma: no cover
+    from django.core.management.base import CommandParser
+
 
-class AsyncCommand(BaseCommand):
-    """Base command to run async code."""
+class ProjectCommand(BaseCommand):
+    def add_arguments(self, parser: CommandParser):
+        projects = parser.add_mutually_exclusive_group(required=True)
+        projects.add_argument(
+            "--all",
+            action="store_true",
+            help="Affect all projects",
+        )
+        projects.add_argument("projects", nargs="*", type=str, default=[])
 
     def handle(self, *args, **options):
-        run(self._handle(*args, **options))
+        projects = []
+        if not options["all"]:
+            nonexistant_projects = []
+            project_names = options["projects"]
+            for name in project_names:
+                try:
+                    projects.append(Project.objects.get(name__iexact=name))
+                except Project.DoesNotExist:
+                    nonexistant_projects.append(name)
+
+            if nonexistant_projects:
+                self.stderr.write(
+                    f"Projects with names {nonexistant_projects} do not exist"
+                )
+                return
+        projects = (
+            Project.objects.filter(id__in=[project.pk for project in projects])
+            or Project.objects.all()
+        )
+
+        for project in projects:
+            self._handle(project)
+
+    def _handle(self, project: Project): # pragma: no cover
+        raise NotImplementedError()

api/outdated/outdated/management/commands/syncproject.py

@@ -1,21 +1,11 @@
-from django.core.management.base import BaseCommand
+from outdated.commands import ProjectCommand
+from outdated.outdated.synchroniser import Synchroniser
 
-from ...models import Project
-from ...synchroniser import Synchroniser
 
-
-class Command(BaseCommand):
+class Command(ProjectCommand):
     help = "Syncs the given project with its remote counterpart."
 
-    def add_arguments(self, parser):
-        parser.add_argument("project_name", type=str)
-
-    def handle(self, *args, **options):
-        project_name = options["project_name"]
-        try:
-            project = Project.objects.get(name__iexact=project_name)
-            self.stdout.write(f"Syncing project {project}")
-            Synchroniser(project).sync()
-            self.stdout.write(f"Finished syncing {project}")
-        except Project.DoesNotExist:
-            self.stdout.write(f"Project {project_name} not found")
+    def _handle(self, project):
+        self.stdout.write(f"Syncing project {project}")
+        Synchroniser(project).sync()
+        self.stdout.write(f"Finished syncing {project}")