fix types by switching account type to enum

Author: joshuadavidthomas

src/django_github_app/management/commands/github.py

@@ -6,6 +6,7 @@
 from django_typer.management import Typer
 from typer import Option
 
+from django_github_app.models import AccountType
 from django_github_app.models import Installation
 from django_github_app.models import Repository
 
@@ -19,7 +20,7 @@ def github(): ...
 @cli.command()
 def import_app(
     type: Annotated[
-        str,
+        AccountType,
         Option(help="The type of account the GitHub App is installed on"),
     ],
     name: Annotated[

src/django_github_app/models.py

@@ -1,8 +1,8 @@
 from __future__ import annotations
 
 import datetime
+from enum import Enum
 from typing import Any
-from typing import Literal
 
 from asgiref.sync import async_to_sync
 from django.db import models
@@ -113,6 +113,11 @@ def from_event(cls, event: sansio.Event) -> InstallationStatus:
                 raise ValueError(f"Unknown installation action: {action}")
 
 
+class AccountType(str, Enum):
+    ORG = "org"
+    USER = "user"
+
+
 class Installation(models.Model):
     id: int
     installation_id = models.PositiveBigIntegerField(unique=True)
@@ -145,14 +150,12 @@ async def aget_access_token(self, gh: abc.GitHubAPI):  # pragma: no cover
     def get_access_token(self, gh: abc.GitHubAPI):  # pragma: no cover
         return async_to_sync(self.aget_access_token)(gh)
 
-    async def arefresh_from_gh(
-        self, account_type: Literal["org", "user"], account_name: str
-    ):
+    async def arefresh_from_gh(self, account_type: AccountType, account_name: str):
         match account_type:
-            case "org":
+            case AccountType.ORG:
                 endpoint = GitHubAPIEndpoint.ORG_APP_INSTALLATION
                 url_var = "org"
-            case "user":
+            case AccountType.USER:
                 endpoint = GitHubAPIEndpoint.USER_APP_INSTALLATION
                 url_var = "username"
             case _:
@@ -168,7 +171,7 @@ async def arefresh_from_gh(
         self.data = data
         await self.asave()
 
-    def refresh_from_gh(self, account_type: Literal["org", "user"], account_name: str):
+    def refresh_from_gh(self, account_type: AccountType, account_name: str):
         return async_to_sync(self.arefresh_from_gh)(account_type, account_name)
 
     async def aget_repos(self, params: dict[str, Any] | None = None):