Show import string source

Author: YuriiMotov

src/fastapi_cli/cli.py

@@ -8,7 +8,12 @@
 from rich.tree import Tree
 
 from fastapi_cli.config import FastAPIConfig
-from fastapi_cli.discover import get_import_data, get_import_data_from_import_string
+from fastapi_cli.discover import (
+    AppConfigSource,
+    ModuleConfigSource,
+    get_import_data,
+    get_import_data_from_import_string,
+)
 from fastapi_cli.exceptions import FastAPICLIException
 
 from . import __version__
@@ -22,6 +27,15 @@
 logger = logging.getLogger(__name__)
 
 
+SOURCE_DESCRIPTIONS: dict[ModuleConfigSource | AppConfigSource, str] = {
+    "entrypoint-option": "[blue]--entrypoint[/] option",
+    "entrypoint-pyproject": "[blue]entrypoint[/] in [blue]pyproject.toml[/]",
+    "path-argument": "path argument",
+    "app-option": "[blue]--app[/] option",
+    "auto-discovery": "auto-discovery",
+}
+
+
 try:
     import uvicorn
 except ImportError:  # pragma: no cover
@@ -151,7 +165,9 @@ def _run(
             if path or app:
                 import_data = get_import_data(path=path, app_name=app)
             elif config.entrypoint:
-                import_data = get_import_data_from_import_string(config.entrypoint)
+                import_data = get_import_data_from_import_string(
+                    config.entrypoint, config.from_pyproject
+                )
             else:
                 import_data = get_import_data()
         except FastAPICLIException as e:
@@ -189,6 +205,17 @@ def _run(
             tag="app",
         )
 
+        # Pring configuration sources
+        mod_source_desc = SOURCE_DESCRIPTIONS[import_data.module_config_source]
+        app_source_desc = SOURCE_DESCRIPTIONS[import_data.app_name_config_source]
+        toolkit.print_line()
+        toolkit.print("Configuration sources:", tag="info")
+        if mod_source_desc == app_source_desc:
+            toolkit.print(f" • Import string: {mod_source_desc}")
+        else:
+            toolkit.print(f" • Module: {mod_source_desc}")
+            toolkit.print(f" • App name: {app_source_desc}")
+
         url = f"http://{host}:{port}"
         url_docs = f"{url}/docs"
 

src/fastapi_cli/config.py

@@ -9,6 +9,7 @@
 
 class FastAPIConfig(BaseModel):
     entrypoint: StrictStr | None = None
+    from_pyproject: bool = False
 
     @classmethod
     def _read_pyproject_toml(cls) -> dict[str, Any]:
@@ -39,4 +40,6 @@ def resolve(cls, entrypoint: str | None = None) -> "FastAPIConfig":
         if entrypoint is not None:
             config["entrypoint"] = entrypoint
 
+        config["from_pyproject"] = ("entrypoint" in config) and (entrypoint is None)
+
         return cls.model_validate(config)

src/fastapi_cli/discover.py

@@ -3,6 +3,7 @@
 from dataclasses import dataclass
 from logging import getLogger
 from pathlib import Path
+from typing import Literal, TypeAlias
 
 from fastapi_cli.exceptions import FastAPICLIException
 
@@ -102,18 +103,35 @@ def get_app_name(*, mod_data: ModuleData, app_name: str | None = None) -> str:
     raise FastAPICLIException("Could not find FastAPI app in module, try using --app")
 
 
+ModuleConfigSource: TypeAlias = Literal[
+    "entrypoint-option",
+    "entrypoint-pyproject",
+    "path-argument",
+    "auto-discovery",
+]
+
+AppConfigSource: TypeAlias = Literal[
+    "entrypoint-option", "entrypoint-pyproject", "app-option", "auto-discovery"
+]
+
+
 @dataclass
 class ImportData:
     app_name: str
     module_data: ModuleData
     import_string: str
 
+    module_config_source: ModuleConfigSource
+    app_name_config_source: AppConfigSource
+
 
 def get_import_data(
     *, path: Path | None = None, app_name: str | None = None
 ) -> ImportData:
+    path_config_source: ModuleConfigSource = "path-argument"
     if not path:
         path = get_default_path()
+        path_config_source = "auto-discovery"
 
     logger.debug(f"Using path [blue]{path}[/blue]")
     logger.debug(f"Resolved absolute path {path.resolve()}")
@@ -127,11 +145,17 @@ def get_import_data(
     import_string = f"{mod_data.module_import_str}:{use_app_name}"
 
     return ImportData(
-        app_name=use_app_name, module_data=mod_data, import_string=import_string
+        app_name=use_app_name,
+        module_data=mod_data,
+        import_string=import_string,
+        module_config_source=path_config_source,
+        app_name_config_source="app-option" if app_name else "auto-discovery",
     )
 
 
-def get_import_data_from_import_string(import_string: str) -> ImportData:
+def get_import_data_from_import_string(
+    import_string: str, from_pyproject: bool
+) -> ImportData:
     module_str, _, app_name = import_string.partition(":")
 
     if not module_str or not app_name:
@@ -151,4 +175,10 @@ def get_import_data_from_import_string(import_string: str) -> ImportData:
             module_paths=[],
         ),
         import_string=import_string,
+        module_config_source=(
+            "entrypoint-pyproject" if from_pyproject else "entrypoint-option"
+        ),
+        app_name_config_source=(
+            "entrypoint-pyproject" if from_pyproject else "entrypoint-option"
+        ),
     )

tests/test_discover.py

@@ -12,19 +12,34 @@
 
 
 def test_get_import_data_from_import_string_valid() -> None:
-    result = get_import_data_from_import_string("module.submodule:app")
+    result = get_import_data_from_import_string("module.submodule:app", False)
 
     assert isinstance(result, ImportData)
     assert result.app_name == "app"
     assert result.import_string == "module.submodule:app"
     assert result.module_data.module_import_str == "module.submodule"
     assert result.module_data.extra_sys_path == Path(".").resolve()
     assert result.module_data.module_paths == []
+    assert result.module_config_source == "entrypoint-option"
+    assert result.app_name_config_source == "entrypoint-option"
+
+
+def test_get_import_data_from_import_string_pyproject_valid() -> None:
+    result = get_import_data_from_import_string("module.submodule:app", True)
+
+    assert isinstance(result, ImportData)
+    assert result.app_name == "app"
+    assert result.import_string == "module.submodule:app"
+    assert result.module_data.module_import_str == "module.submodule"
+    assert result.module_data.extra_sys_path == Path(".").resolve()
+    assert result.module_data.module_paths == []
+    assert result.module_config_source == "entrypoint-pyproject"
+    assert result.app_name_config_source == "entrypoint-pyproject"
 
 
 def test_get_import_data_from_import_string_missing_colon() -> None:
     with pytest.raises(FastAPICLIException) as exc_info:
-        get_import_data_from_import_string("module.submodule")
+        get_import_data_from_import_string("module.submodule", False)
 
     assert "Import string must be in the format module.submodule:app_name" in str(
         exc_info.value
@@ -33,7 +48,7 @@ def test_get_import_data_from_import_string_missing_colon() -> None:
 
 def test_get_import_data_from_import_string_missing_app() -> None:
     with pytest.raises(FastAPICLIException) as exc_info:
-        get_import_data_from_import_string("module.submodule:")
+        get_import_data_from_import_string("module.submodule:", False)
 
     assert "Import string must be in the format module.submodule:app_name" in str(
         exc_info.value
@@ -42,7 +57,7 @@ def test_get_import_data_from_import_string_missing_app() -> None:
 
 def test_get_import_data_from_import_string_missing_module() -> None:
     with pytest.raises(FastAPICLIException) as exc_info:
-        get_import_data_from_import_string(":app")
+        get_import_data_from_import_string(":app", False)
 
     assert "Import string must be in the format module.submodule:app_name" in str(
         exc_info.value
@@ -51,8 +66,7 @@ def test_get_import_data_from_import_string_missing_module() -> None:
 
 def test_get_import_data_from_import_string_empty() -> None:
     with pytest.raises(FastAPICLIException) as exc_info:
-        get_import_data_from_import_string("")
-
+        get_import_data_from_import_string("", False)
     assert "Import string must be in the format module.submodule:app_name" in str(
         exc_info.value
     )