refactor: update profile handling to use BaseProfile and resolve_profile for better structure

Author: gitmobkab

src/copia/cli/config/__init__.py

@@ -1,5 +1,5 @@
 """Holds utilities for enhanced management of the config profile system"""
-from .models import Profile
+from .models import BaseProfile, FileBasedProfile, ServerBasedProfile, resolve_profile
 from .exceptions import *
 from .globals import *
 from .loaders import get_profile

src/copia/cli/config/loaders.py

@@ -12,7 +12,7 @@
 from pydantic import ValidationError
 
 from .globals import  ConfigScope, LOCAL_COPIA_FILE, GLOBAL_COPIA_FILE
-from .models import Profile
+from .models import BaseProfile, resolve_profile
 from .exceptions import (
     FoundProfileIsNotATableError,
     ProfilesKeyIsNotATableError,
@@ -71,7 +71,7 @@ def load_config(scope: ConfigScope) -> TOMLDocument:
         raise InvalidConfigError(f"{scope} config file is not a valid TOML file: {TOMLErr}")
 
 
-def get_profile_from_config(profile_name: str, config: dict) -> Profile:
+def get_profile_from_config(profile_name: str, config: dict) -> BaseProfile:
     """try to get a valid profile from config
 
     config must be a dict representation of the config file
@@ -89,7 +89,7 @@ def get_profile_from_config(profile_name: str, config: dict) -> Profile:
         InvalidProfileError: when the profile isn't valid
 
     Returns:
-        Profile: a valid Profile object
+        BaseProfile: a valid BaseProfile object
     """
     profiles = config.get("profiles")
     if profiles is None:
@@ -103,12 +103,12 @@ def get_profile_from_config(profile_name: str, config: dict) -> Profile:
         raise FoundProfileIsNotATableError(profile_name)
 
     try:
-        return Profile(**profile_data)
+        return resolve_profile(profile_data)
     except ValidationError as Err:
         raise InvalidProfileError(Err)
 
 
-def get_profile(profile_name: str, scope: ConfigScope) -> Profile:
+def get_profile(profile_name: str, scope: ConfigScope) -> BaseProfile:
     """try to get a valid profile from config based on scope
 
     Args:
@@ -124,7 +124,7 @@ def get_profile(profile_name: str, scope: ConfigScope) -> Profile:
         InvalidProfileError: when the profile isn't valid
         
     Returns:
-        Profile: a valid Profile object
+        BaseProfile: a valid BaseProfile object
     """
     config = load_config(scope)
     return get_profile_from_config(profile_name, config)

src/copia/cli/config/models.py

@@ -1,6 +1,6 @@
 from typing import Annotated, Union
 
-from pydantic import BaseModel, Field, StringConstraints, ConfigDict
+from pydantic import BaseModel, Field, StringConstraints
 from pydantic.networks import IPvAnyAddress
 
 from .globals import Adapter, PORT_MIN_VAL, PORT_MAX_VAL
@@ -16,11 +16,12 @@
     strict=True,
     description="The Database Port"
 )]
-
-class Profile(BaseModel):
-    model_config = ConfigDict(extra="forbid")
 
+class BaseProfile(BaseModel):
     adapter: Adapter = Field(description="The copia Adapter to use")
+
+class ServerBasedProfile(BaseProfile):
+    
     host: Union[IPvAnyAddress, NotEmptyStr] = Field(description="The database hostname")
     port: Port
     database: NotEmptyStr = Field(description="The Database Port")
@@ -34,3 +35,15 @@ def __str__(self) -> str:
     def __repr__(self) -> str:
         return self.__str__()
 
+class FileBasedProfile(BaseProfile):
+    database: str
+    
+    def __str__(self) -> str:
+        return f"{self.database}"
+
+def resolve_profile(payload: dict) -> BaseProfile:
+    basic_info = BaseProfile(**payload)
+    if basic_info.adapter == "mysql":
+        return FileBasedProfile(**payload)
+    else:
+        return ServerBasedProfile(**payload)