Pull changes from a specific remote defaults to origin (#118)

Author: jackton1

migration_fixer/management/commands/makemigrations.py

@@ -46,6 +46,12 @@ def add_arguments(self, parser):
             help="The name of the default branch.",
             default="main",
         )
+        parser.add_argument(
+            "-r",
+            "--remote",
+            help="Git remote.",
+            default="origin",
+        )
         parser.add_argument(
             "-f",
             "--force-update",
@@ -60,6 +66,7 @@ def handle(self, *app_labels, **options):
         self.fix = options["fix"]
         self.force_update = options["force_update"]
         self.default_branch = options["default_branch"]
+        self.remote = options["remote"]
 
         if self.fix:
             try:
@@ -105,17 +112,17 @@ def handle(self, *app_labels, **options):
                         )
 
                     if current_branch == self.default_branch:  # pragma: no cover
-                        for remote in self.repo.remotes:
-                            remote.pull(
-                                self.default_branch,
-                                force=self.force_update,
-                            )
+                        remote = self.repo.remotes[self.remote]
+                        remote.pull(
+                            self.default_branch,
+                            force=self.force_update,
+                        )
                     else:
-                        for remote in self.repo.remotes:
-                            remote.fetch(
-                                f"{self.default_branch}:{self.default_branch}",
-                                force=self.force_update,
-                            )
+                        remote = self.repo.remotes[self.remote]
+                        remote.fetch(
+                            f"{self.default_branch}:{self.default_branch}",
+                            force=self.force_update,
+                        )
 
                     if self.verbosity >= 2:
                         self.stdout.write(