Create and use a type alias for valid Endpoint classes.

cvmfsscraper/http_get_models.py

@@ -2,7 +2,7 @@
 import re
 from datetime import datetime
 from enum import Enum
-from typing import Any, Dict, List, Optional
+from typing import Any, Dict, List, Optional, Union
 
 from pydantic import BaseModel, Field, field_validator, model_validator
 
@@ -299,3 +299,10 @@ def __init__(self, path: str, model_class: type[BaseModel]):
         """Initialize the endpoint."""
         self.path = path
         self.model_class = model_class
+
+
+# Dynamically creating this list based on the Endpoints enum values
+# is not supported by mypy et al, so we have to do it manually.
+EndpointClassesType = Union[
+    GetCVMFSPublished, GetCVMFSRepositoriesJSON, GetCVMFSStatusJSON, GetGeoAPI
+]

cvmfsscraper/server.py

@@ -1,15 +1,15 @@
 """Server class for cvmfs-server-metadata."""
 
 import json
-from typing import Dict, List, Union
+from typing import Dict, List
 from urllib import error, request
 
 from cvmfsscraper.constants import GeoAPIStatus
 from cvmfsscraper.http_get_models import (
+    EndpointClassesType,
     Endpoints,
     GetCVMFSPublished,
     GetCVMFSRepositoriesJSON,
-    GetCVMFSStatusJSON,
     GetGeoAPI,
     RepositoryOrReplica,
 )
@@ -213,9 +213,7 @@ def fetch_endpoint(
         endpoint: Endpoints,
         repo: str = "data",
         geoapi_servers: str = GEOAPI_SERVERS,
-    ) -> Union[
-        GetCVMFSPublished, GetCVMFSRepositoriesJSON, GetCVMFSStatusJSON, GetGeoAPI
-    ]:
+    ) -> EndpointClassesType:
         """Fetch and process a specified URL endpoint.
 
         This function reads the content of a specified URL and ether returns a validated