Apply some refactors

Author: Muscaw

gitea_github_sync/cli.py

@@ -1,7 +1,7 @@
 import click
 from rich import print
 
-from . import github
+from . import github, repository
 
 
 @click.group()
@@ -21,10 +21,10 @@ def list_all_repositories(stats: bool) -> None:
         print()
         print("[b]Repository stats[/]")
         number_public_repos = sum(
-            [1 if repo.visibility == github.Visibility.PUBLIC else 0 for repo in repos]
+            1 if repo.visibility == repository.Visibility.PUBLIC else 0 for repo in repos
         )
         number_private_repos = sum(
-            [1 if repo.visibility == github.Visibility.PRIVATE else 0 for repo in repos]
+            1 if repo.visibility == repository.Visibility.PRIVATE else 0 for repo in repos
         )
         number_unknown_repos = len(repos) - number_public_repos - number_private_repos
         print(f"Number of public repos identified: [b red]{number_public_repos}[/]")

gitea_github_sync/github.py

@@ -1,39 +1,11 @@
 from __future__ import annotations
 
-from dataclasses import dataclass
-from enum import Flag, auto
 from typing import List, Optional
 
 from github import Github
 
 from . import config
-
-
-class Visibility(Flag):
-    PUBLIC = auto()
-    PRIVATE = auto()
-    UNKNOWN = auto()
-
-    @staticmethod
-    def from_str(value: str) -> Visibility:
-        if value == "public":
-            return Visibility.PUBLIC
-        elif value == "private":
-            return Visibility.PRIVATE
-        else:
-            return Visibility.UNKNOWN
-
-
-@dataclass(frozen=True)
-class Repository:
-    full_repo_name: str
-    visibility: Visibility
-
-    def get_org_name(self) -> str:
-        return self.full_repo_name.split("/")[0]
-
-    def get_repo_name(self) -> str:
-        return self.full_repo_name.split("/")[1]
+from .repository import Repository, Visibility
 
 
 def get_github(conf: Optional[config.Config] = None) -> Github:
@@ -44,11 +16,10 @@ def get_github(conf: Optional[config.Config] = None) -> Github:
 
 def list_all_repositories(gh: Github) -> List[Repository]:
     repos = gh.get_user().get_repos()
-    output = []
-    for repo in repos:
-        output.append(
-            Repository(
-                full_repo_name=repo.full_name, visibility=Visibility.from_str(repo.visibility)
-            )
+    return [
+        Repository(
+            full_repo_name=repo.full_name,
+            visibility=Visibility.from_str(repo.visibility),
         )
-    return output
+        for repo in repos
+    ]

gitea_github_sync/repository.py

@@ -0,0 +1,31 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from enum import Flag, auto
+
+
+class Visibility(Flag):
+    PUBLIC = auto()
+    PRIVATE = auto()
+    UNKNOWN = auto()
+
+    @staticmethod
+    def from_str(value: str) -> Visibility:
+        if value == "public":
+            return Visibility.PUBLIC
+        elif value == "private":
+            return Visibility.PRIVATE
+        else:
+            return Visibility.UNKNOWN
+
+
+@dataclass(frozen=True)
+class Repository:
+    full_repo_name: str
+    visibility: Visibility
+
+    def get_org_name(self) -> str:
+        return self.full_repo_name.split("/")[0]
+
+    def get_repo_name(self) -> str:
+        return self.full_repo_name.split("/")[1]