Add is_rate_limit_low method

ogr/abstract/git_service.py

@@ -141,3 +141,13 @@ def get_group(self, group_name: str):
         Get a group by name.
         """
         raise NotImplementedError
+
+    def get_rate_limit_remaining(self) -> Optional[int]:
+        """
+        Get the remaining rate limit.
+
+        Returns:
+            Number of remaining API requests, or None if rate limit information
+            is not available.
+        """
+        raise NotImplementedError

ogr/services/forgejo/service.py

@@ -87,3 +87,9 @@ def get_project_from_url(self, url: str) -> "ForgejoProject":
         repo = path_parts[1]
 
         return self.get_project(repo=repo, namespace=namespace)
+
+    def get_rate_limit_remaining(self) -> Optional[int]:
+        """
+        There is no way to check rate limit status from Forgejo API.
+        """
+        return None

ogr/services/github/service.py

@@ -239,3 +239,15 @@ def list_projects(
             ]
 
         return projects
+
+    def get_rate_limit_remaining(self) -> Optional[int]:
+        rate_limit = self.github.get_rate_limit()
+        # Handle both old and new PyGithub API versions
+        # Old API in f42 and f43: rate_limit.resources.core.remaining
+        # New API in rawhide (f44): rate_limit.core.remaining (or rate_limit.remaining)
+        try:
+            # since PyGithub 2.7.0
+            return rate_limit.resources.core.remaining
+        except AttributeError:
+            return rate_limit.core.remaining
+        return None

ogr/services/gitlab/service.py

@@ -172,3 +172,29 @@ def list_projects(
             )
             for project in projects_to_convert
         ]
+
+    def get_rate_limit_remaining(self) -> Optional[int]:
+        # python-gitlab doesn't have get_rate_limit(), so we make a lightweight
+        # HEAD request to get rate limit headers from GitLab API
+        # GitLab returns rate limit in headers: ratelimit-remaining
+        # Use obey_rate_limit=False to prevent blocking if we've already hit the limit
+        try:
+            headers = self.gitlab_instance.http_head("/user", obey_rate_limit=False)
+            remaining = headers.get("ratelimit-remaining")
+            if remaining:
+                return int(remaining)
+        except gitlab.GitlabHttpError as e:
+            # If we get a 429, we've hit the rate limit
+            if e.response_code == 429:
+                logger.error(
+                    f"Rate limit has been exceeded: {e}",
+                )
+                return 0
+            logger.error(
+                f"Could not get rate limit from GitLab: {e}",
+            )
+        except Exception as e:
+            logger.error(
+                f"Could not get rate limit from GitLab: {e}",
+            )
+        return None

ogr/services/pagure/service.py

@@ -403,3 +403,9 @@ def get_group(self, group_name: str) -> PagureGroup:
         """
         url = self.get_api_url("group", group_name)
         return PagureGroup(group_name, self.call_api(url))
+
+    def get_rate_limit_remaining(self) -> Optional[int]:
+        """
+        There is no way to check rate limit status from Pagure API.
+        """
+        return None

tests/integration/forgejo/test_service.py

@@ -42,3 +42,8 @@ def test_project_create(service, kwargs_):
     # Try to fetch newly created project
     project = service.get_project(**kwargs_fetch)
     assert project.forgejo_repo
+
+
+def test_get_rate_limit_remaining(service):
+    remaining = service.get_rate_limit_remaining()
+    assert remaining is None

tests/integration/github/test_service.py

@@ -131,3 +131,7 @@ def test_wrong_auth_static_method(self):
         # Verify that the exception handler is applied to static methods
         with pytest.raises(GithubAPIException):
             GithubPullRequest.get(self.ogr_project, 1)
+
+    def test_get_rate_limit_remaining(self):
+        remaining = self.service.get_rate_limit_remaining()
+        assert remaining >= 100