Add small optimizations

Author: yugokato

src/openapi_test_client/libraries/api/api_classes/__init__.py

@@ -56,8 +56,8 @@ def init_api_classes(base_api_class: type[APIClassType]) -> list[type[APIClassTy
 
     # Set all API class' Endpoint objects to the base class's endpoint attribute
     base_api_class.endpoints = sorted(
-        list(itertools.chain(*[x.endpoints for x in api_classes if x.endpoints])),
-        key=lambda x: (", ".join(x.tags), x.method, x.path),
+        itertools.chain(*(x.endpoints for x in api_classes if x.endpoints)),
+        key=lambda x: (x.tags, x.method, x.path),
     )
     return sorted(api_classes, key=lambda x: x.TAGs)
 
@@ -71,14 +71,17 @@ def get_api_classes(api_class_dir: Path, base_api_class: type[APIClassType]) ->
         raise RuntimeError(f"Found no API class modules in {api_class_dir}")
 
     api_classes = [
-        getattr(mod, x)
+        obj
         for mod in api_modules
         for x in dir(mod)
-        if inspect.isclass(getattr(mod, x))
-        and issubclass(getattr(mod, x), base_api_class)
+        if not x.startswith("_")
         and x != base_api_class.__name__
+        and (obj := getattr(mod, x))
+        and inspect.isclass(obj)
+        and issubclass(obj, base_api_class)
     ]
-    assert api_classes, (
-        f"Unable to find any API class that is a subclass of {base_api_class.__name__} in {api_class_dir}"
-    )
+    if not api_classes:
+        raise RuntimeError(
+            f"Unable to find any API class that is a subclass of {base_api_class.__name__} in {api_class_dir}"
+        )
     return api_classes

src/openapi_test_client/libraries/api/api_functions/endpoints.py

@@ -38,7 +38,7 @@
 logger = get_logger(__name__)
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, slots=True)
 class Endpoint:
     """An Endpoint class to hold various endpoint data associated to an API class function
 

src/openapi_test_client/libraries/api/api_functions/utils/endpoint_model.py

@@ -88,6 +88,7 @@ def create_endpoint_model(endpoint_func: EndpointFunc, api_spec: dict[str, Any]
             namespace={"content_type": content_type, "endpoint_func": endpoint_func},
             kw_only=True,
             frozen=True,
+            slots=True,
         ),
     )
 

src/openapi_test_client/libraries/api/types.py

@@ -430,14 +430,14 @@ def to_tuple(self) -> tuple[str, str | bytes | str]:
         return astuple(self)  # type: ignore
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, slots=True)
 class ParamAnnotationType:
     """Base class for all custom classes used for annotating endpoint function parameters"""
 
     ...
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, slots=True)
 class Alias(ParamAnnotationType):
     """Alias param name
 
@@ -448,7 +448,7 @@ class Alias(ParamAnnotationType):
     value: str
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, slots=True)
 class Format(ParamAnnotationType):
     """OpenAPI parameter format
 
@@ -458,7 +458,7 @@ class Format(ParamAnnotationType):
     value: str
 
 
-@dataclass(frozen=True)
+@dataclass(frozen=True, slots=True)
 class Constraint(ParamAnnotationType):
     """OpenAPI parameter constraints (maxLength, minLength, etc.)