Use PyGithub to resolve repository from CLI params

Author: smkent

cookie_python/manage/github.py

@@ -0,0 +1,24 @@
+import contextlib
+import os
+from functools import cached_property, lru_cache
+
+from github import Github
+from github.Repository import Repository
+
+
+class GithubRepo:
+    def __init__(self) -> None:
+        self._gh = Github(os.environ["GITHUB_ACCESS_TOKEN"])
+
+    @cached_property
+    def username(self) -> str:
+        return self._gh.get_user().login
+
+    @lru_cache  # noqa: B019
+    def find_repo(self, search: str) -> Repository:
+        if "/" not in search:
+            search = f"{self.username}/{search}"
+        if "github.com" in search:
+            with contextlib.suppress(IndexError):
+                search = os.path.splitext(search.split(":")[1])[0]
+        return self._gh.get_repo(search)

cookie_python/manage/repo.py

@@ -13,11 +13,13 @@
 
 import loguru
 
+from .github import GithubRepo
+
 
 class RepoSandbox:
     def __init__(self, repo: str, dry_run: bool = False) -> None:
         self._stack = contextlib.ExitStack()
-        self.repo = repo
+        self.repo = self.gh.find_repo(repo)
         self.branch = "update-cookie"
         self.dry_run = dry_run
 
@@ -32,9 +34,13 @@ def __exit__(
     ) -> None:
         self._stack.close()
 
+    @cached_property
+    def gh(self) -> GithubRepo:
+        return GithubRepo()
+
     @cached_property
     def logger(self) -> "loguru.Logger":
-        return loguru.logger.bind(repo=self.repo)
+        return loguru.logger.bind(repo=self.repo.full_name)
 
     @cached_property
     def tempdir(self) -> Path:
@@ -47,7 +53,9 @@ def tempdir(self) -> Path:
     @cached_property
     def clone_path(self) -> Path:
         subprocess.run(
-            ["git", "clone", self.repo, "repo"], cwd=self.tempdir, check=True
+            ["git", "clone", self.repo.ssh_url, "repo"],
+            cwd=self.tempdir,
+            check=True,
         )
         clone_path = self.tempdir / "repo"
         for cmd in (