tracktolib 0.45.0__py3-none-any.whl → 0.46.1__py3-none-any.whl

tracktolib/api.py

@@ -24,6 +24,7 @@ from .utils import json_serial
 
 try:
     from fastapi import params, APIRouter
+    from fastapi.routing import APIRoute
     from fastapi.responses import JSONResponse
     from pydantic.alias_generators import to_camel
     from pydantic import BaseModel, ConfigDict
@@ -205,14 +206,56 @@ _NoneType = type(None)
 
 class IgnoreConfig(BaseModel):
     endpoints: dict[str, dict[Method, bool]]
-    ignore_default: bool = True
+    ignore_missing: bool = True
 
 
-def _get_ignore_config() -> IgnoreConfig | None:
+def get_ignore_config() -> IgnoreConfig | None:
     _config = os.getenv("IGNORE_CONFIG")
     return IgnoreConfig.model_validate_json(_config) if _config else None
 
 
+def set_ignore_config(config: str | IgnoreConfig):
+    if isinstance(config, str):
+        config = IgnoreConfig.model_validate_json(config)
+    os.environ["IGNORE_CONFIG"] = config.model_dump_json()
+
+
+def _filter_route(route: APIRoute, ignored_route: dict[Method, bool], ignore_missing: bool) -> APIRoute | None:
+    # If no config is provided and default is to ignore missing, return the route
+    if ignored_route is None and not ignore_missing:
+        return route
+
+    has_methods = False
+    enabled_methods = {method for method, has_access in ignored_route.items() if has_access}
+    for method in list(route.methods):
+        # If the config is not specified, we remove the method if ignore_missing is True
+        if method not in ignored_route:
+            if ignore_missing:
+                route.methods -= {method}
+                continue
+            else:
+                has_methods = True
+                continue
+        elif method not in enabled_methods:
+            route.methods -= {method}
+        else:
+            has_methods = True
+
+    if not has_methods:
+        return None
+    return route
+
+
+def filter_routes(routes: list[APIRoute], ignore_config: IgnoreConfig) -> list[APIRoute]:
+    _routes = []
+    for route in routes:
+        _ignored_route = ignore_config.endpoints.get(route.path)
+        _route = _filter_route(route, _ignored_route or {}, ignore_missing=ignore_config.ignore_missing)
+        if _route is not None:
+            _routes.append(_route)
+    return _routes
+
+
 def _get_return_type(fn):
     return_type = get_type_hints(fn)["return"]
     _args = get_args(return_type)
@@ -228,14 +271,8 @@ def add_endpoint(
     *,
     dependencies: Dependencies = None,
 ):
-    _ignore_config = _get_ignore_config()
+    _ignore_config = get_ignore_config()
     for _method, _meta in endpoint.methods.items():
-        # Do not add endpoint if it is not in the ignore config
-        if _ignore_config is not None:
-            _has_access = _ignore_config.endpoints.get(path, {}).get(_method, not _ignore_config.ignore_default)
-            if not _has_access:
-                continue
-        #
         _fn = _meta["fn"]
         _status_code = _meta["status_code"]
         _dependencies = _meta["dependencies"]

{tracktolib-0.45.0.dist-info → tracktolib-0.46.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tracktolib
-Version: 0.45.0
+Version: 0.46.1
 Summary: Utility library for python
 Home-page: https://github.com/tracktor/tracktolib
 License: MIT

{tracktolib-0.45.0.dist-info → tracktolib-0.46.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
 tracktolib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tracktolib/api.py,sha256=uJYVqRnHE_QTi6g2eD-zDpP-GsVPmq40h3hy1v3Agms,8558
+tracktolib/api.py,sha256=DILKtPQn6HRXmQk5FgYHA9o5ToSNGudHsMEF_ZXEuI8,9826
 tracktolib/http_utils.py,sha256=dqjRJTNCo1qDnvQBTcImw1ibi1LnqETrpWDAU5txBHs,2633
 tracktolib/logs.py,sha256=enezPjBU4v2s8knG_V6kIbQsoyxfNIKEpKVSSkr_2U8,2212
 tracktolib/pg/__init__.py,sha256=32kKBJzIDyppszZMAOSMgM3ggz7Ow0TaNKiMs740gyw,333
@@ -13,7 +13,7 @@ tracktolib/s3/minio.py,sha256=JaRpHOM6NqpMieplp58Fpc7uTh7PItyP8On7jMXhB-Q,1234
 tracktolib/s3/s3.py,sha256=0aPUo3Vnfy5Rk_Qp_y0KkCZa2kuMWw2JU2xrodn7oAE,3486
 tracktolib/tests.py,sha256=FS0uae3ShiiJOD7WE2WEv9h3QwRJCbLDHpe_8Ikc0j0,554
 tracktolib/utils.py,sha256=YZZROGgOioblJkqgneHGZRbof45nmZ-VLSk9LqcVkkE,4847
-tracktolib-0.45.0.dist-info/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
-tracktolib-0.45.0.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
-tracktolib-0.45.0.dist-info/METADATA,sha256=j7mvkYhNZZN0JdGdpoU73sfWg3nyBj2xgmTnZND5gZI,3610
-tracktolib-0.45.0.dist-info/RECORD,,
+tracktolib-0.46.1.dist-info/LICENSE,sha256=uUanH0X7SeZEPdsRTHegMSMTiIHMurt9H0jSwEwKE1Y,1081
+tracktolib-0.46.1.dist-info/WHEEL,sha256=vxFmldFsRN_Hx10GDvsdv1wroKq8r5Lzvjp6GZ4OO8c,88
+tracktolib-0.46.1.dist-info/METADATA,sha256=wi_XoRBTxEhgO2DvDPf1h_cf4XL8mcd0JP4i76_lHu0,3610
+tracktolib-0.46.1.dist-info/RECORD,,