Refactor get_docs_urls to return list of URLs

Author: YuriiMotov

src/fastapi_cli/cli.py

@@ -194,35 +194,25 @@ def _run(
             tag="app",
         )
 
-        docs_urls = get_docs_urls(import_data)
-
         url = f"http://{host}:{port}"
 
         use_root_path = root_path or get_root_path(import_data)
         if use_root_path:
             url += use_root_path
 
-        docs_url = ""
-
-        if docs_urls.openapi_url and (docs_urls.docs_url or docs_urls.redoc_url):
-            if docs_urls.docs_url:
-                docs_url = (
-                    f"[link={url}{docs_urls.docs_url}]{url}{docs_urls.docs_url}[/]"
-                )
-            else:
-                docs_url = (
-                    f"[link={url}{docs_urls.redoc_url}]{url}{docs_urls.redoc_url}[/]"
-                )
-
         toolkit.print_line()
         toolkit.print(
             f"Server started at [link={url}]{url}[/]",
             tag="server",
         )
 
-        if docs_url:
+        docs_urls = get_docs_urls(import_data)
+        docs_links = [
+            f"[link={url}{docs_url}]{url}{docs_url}[/]" for docs_url in docs_urls
+        ]
+        if docs_links:
             toolkit.print(
-                f"Documentation at {docs_url}",
+                f"Documentation at {docs_links[0]}",
                 tag="server",
             )
 

src/fastapi_cli/discover.py

@@ -159,14 +159,7 @@ def get_import_data_from_import_string(import_string: str) -> ImportData:
     )
 
 
-@dataclass
-class DocsURLs:
-    openapi_url: str | None
-    docs_url: str | None
-    redoc_url: str | None
-
-
-def get_docs_urls(import_data: ImportData) -> DocsURLs:
+def get_docs_urls(import_data: ImportData) -> list[str]:
     module = importlib.import_module(import_data.module_data.module_import_str)
     app_name = import_data.app_name
     app = getattr(module, app_name)
@@ -175,11 +168,13 @@ def get_docs_urls(import_data: ImportData) -> DocsURLs:
     assert FastAPI is not None
     assert isinstance(app, FastAPI)
 
-    return DocsURLs(
-        openapi_url=app.openapi_url,
-        docs_url=app.docs_url,
-        redoc_url=app.redoc_url,
-    )
+    docs_urls: list[str] = []
+    if app.openapi_url and app.docs_url:
+        docs_urls.append(app.docs_url)
+    if app.openapi_url and app.redoc_url:
+        docs_urls.append(app.redoc_url)
+
+    return docs_urls
 
 
 def get_root_path(import_data: ImportData) -> str: