feat: management command fixed for re-runs

Author: marslanabdulrauf

src/ol_openedx_git_auto_export/management/commands/migrate_giturl.py

@@ -1,6 +1,9 @@
 from django.core.management.base import BaseCommand
 from ol_openedx_git_auto_export.models import CourseGitRepo
+from ol_openedx_git_auto_export.signals import listen_for_course_created
 from openedx.core.djangoapps.content.course_overviews.models import CourseOverview
+from openedx_events.content_authoring.data import CourseData
+from xmodule.modulestore.django import modulestore
 
 
 class Command(BaseCommand):
@@ -20,9 +23,31 @@ def handle(self, *args, **options):  # noqa: ARG002
         if course_ids:
             courses = courses.filter(id__in=course_ids)
 
+        seen_giturls = set()
         for course in courses:
-            if course.git_url:
+            course_module = modulestore().get_course(course.id, depth=1)
+            giturl = course_module.giturl
+            if giturl and giturl not in seen_giturls:
+                seen_giturls.add(giturl)
+                self.stdout.write(
+                    self.style.SUCCESS(f"Course {course.id} has giturl: {giturl}")
+                )
                 CourseGitRepo.objects.get_or_create(
                     course_id=course.id,
-                    defaults={"git_url": course.git_url},
+                    defaults={"git_url": giturl},
+                )
+            elif giturl and giturl in seen_giturls:
+                self.stdout.write(
+                    self.style.WARNING(
+                        f"Course {course.id} has a duplicate giturl: {giturl}"
+                    )
                 )
+                ssh_url = listen_for_course_created(CourseData(course_key=course.id))
+                if ssh_url:
+                    seen_giturls.add(ssh_url)
+            else:
+                self.stdout.write(
+                    self.style.WARNING(f"Course {course.id} does not have a giturl.")
+                )
+
+        self.stdout.write(self.style.SUCCESS("Git URLs migrated successfully."))

src/ol_openedx_git_auto_export/signals.py

@@ -49,22 +49,22 @@ def listen_for_course_created(**kwargs):
             "GitHub repo creation is disabled. Skipping GitHub repo creation for course %s",  # noqa: E501
             course_id,
         )
-        return
+        return None
 
     # SignalHandler.course_published is called before COURSE_CREATED signal
     if CourseGitRepo.objects.filter(course_id=str(course_id)).exists():
         log.info(
             "GitHub repository already exists for course %s. Skipping creation.",
             course_id,
         )
-        return
+        return None
 
     gh_access_token = settings.GITHUB_ACCESS_TOKEN
     if not settings.GITHUB_ORG_API_URL or not gh_access_token:
         log.error(
             "GITHUB_ORG_API_URL or GITHUB_ACCESS_TOKEN is not set in settings. Skipping GitHub repo creation."  # noqa: E501
         )
-        return
+        return None
 
     url = f"{settings.GITHUB_ORG_API_URL}/repos"
     headers = {
@@ -88,7 +88,7 @@ def listen_for_course_created(**kwargs):
             course_id,
             response.json(),
         )
-        return
+        return None
 
     repo_data = response.json()
     ssh_url = repo_data.get("ssh_url")
@@ -107,4 +107,5 @@ def listen_for_course_created(**kwargs):
             "Failed to retrieve URL for GitHub repository for course %s",
             course_id,
         )
-        return
+
+    return ssh_url