chore: more lazy importing

src/githarbor/providers/gitearepository.py

@@ -5,8 +5,6 @@
 from typing import TYPE_CHECKING, Any, ClassVar, Literal
 from urllib.parse import urlparse
 
-import giteapy
-from giteapy.rest import ApiException
 import upath
 
 from githarbor.core.base import BaseRepository
@@ -18,6 +16,8 @@
     from collections.abc import Iterator
     from datetime import datetime
 
+    import giteapy
+
     from githarbor.core.models import (
         Branch,
         Commit,
@@ -46,6 +46,9 @@ def __init__(
         token: str | None = None,
         url: str = "https://gitea.com",
     ):
+        import giteapy
+        from giteapy.rest import ApiException
+
         t = token or os.getenv("GITEA_TOKEN")
         if not t:
             msg = "Gitea token is required"

src/githarbor/providers/githubrepository.py

@@ -6,9 +6,6 @@
 from typing import TYPE_CHECKING, Any, ClassVar, Literal
 from urllib.parse import urlparse
 
-from github import Auth, Github, NamedUser
-from github.GithubException import GithubException
-
 from githarbor.core.base import BaseRepository
 from githarbor.exceptions import AuthenticationError, ResourceNotFoundError
 from githarbor.providers import githubtools
@@ -46,6 +43,9 @@ class GitHubRepository(BaseRepository):
 
     def __init__(self, owner: str, name: str, token: str | None = None):
         """Initialize GitHub repository."""
+        from github import Auth, Github, NamedUser
+        from github.GithubException import GithubException
+
         try:
             t = token or TOKEN
             if t is None:
@@ -310,6 +310,8 @@ def get_release(self, tag: str) -> Release:
     @githubtools.handle_github_errors("Failed to get tag {name}")
     def get_tag(self, name: str) -> Tag:
         """Get a specific tag by name."""
+        from github.GithubException import GithubException
+
         try:
             tag = self._repo.get_git_ref(f"tags/{name}")
             tag_obj = self._repo.get_git_tag(tag.object.sha)

src/githarbor/providers/gitlabrepository.py

@@ -5,9 +5,6 @@
 from typing import TYPE_CHECKING, Any, ClassVar, Literal
 from urllib.parse import urlparse
 
-import gitlab
-from gitlab.exceptions import GitlabAuthenticationError
-
 from githarbor.core.base import BaseRepository
 from githarbor.core.models import (
     Branch,
@@ -43,6 +40,9 @@ def __init__(
         token: str | None = None,
         url: str = "https://gitlab.com",
     ):
+        import gitlab
+        from gitlab.exceptions import GitlabAuthenticationError
+
         try:
             t = token or os.getenv("GITLAB_TOKEN")
             if not t:
@@ -258,7 +258,8 @@ def get_contributors(
 
     @gitlabtools.handle_gitlab_errors("Failed to get languages")
     def get_languages(self) -> dict[str, int]:
-        return self._repo.languages()
+        response = self._repo.languages()
+        return response if isinstance(response, dict) else response.json()
 
     @gitlabtools.handle_gitlab_errors("Failed to compare branches {base} and {head}")
     def compare_branches(

src/githarbor/providers/localrepository.py

@@ -7,8 +7,6 @@
 import pathlib
 from typing import TYPE_CHECKING, Any, ClassVar
 
-import git
-
 from githarbor.core.base import BaseRepository
 from githarbor.exceptions import ResourceNotFoundError
 from githarbor.providers import localtools
@@ -29,6 +27,8 @@ class LocalRepository(BaseRepository):
     url_patterns: ClassVar[list[str]] = []  # Local repos don't have URL patterns
 
     def __init__(self, path: str | os.PathLike[str]) -> None:
+        import git
+
         try:
             self.path = pathlib.Path(path)
             self.repo = git.Repo(self.path)

tests/test_github.py

@@ -17,7 +17,7 @@
 
 @pytest.fixture
 def mock_github():
-    with patch("githarbor.providers.githubrepository.Github") as mock:
+    with patch("github.Github") as mock:
         yield mock
 
 

tests/test_gitlab.py

@@ -21,7 +21,7 @@ def mock_gitlab_client():
 
 @pytest.fixture
 def gitlab_repo(mock_gitlab_client) -> GitLabRepository:
-    with patch("githarbor.providers.gitlabrepository.gitlab.Gitlab") as mock_gitlab:
+    with patch("gitlab.Gitlab") as mock_gitlab:
         mock_gitlab.return_value = mock_gitlab_client
         provider = GitLabRepository("phil65", "test")
         provider._gl = mock_gitlab_client