streamlit-octostar-utils 0.1.7a7__py3-none-any.whl → 0.1.7a9__py3-none-any.whl

streamlit_octostar_utils/api_crafter/celery.py

@@ -492,9 +492,11 @@ class CeleryExecutor(object):
                 kwargs,
                 task_id,
             )
+
             await asyncio.get_running_loop().run_in_executor(
                 self.set_thread_pool, _send_task, task_fn, task_id, options
             )
+
         except asyncio.CancelledError:
             logger.info(f"Cancelling task {task_id} due to disconnect!")
             await self.terminate_task(task_id)
@@ -683,20 +685,26 @@ class CeleryErrorRoute(DefaultErrorRoute):
     DEFAULT_SILENCED_EXCEPTIONS = {CeleryExecutor.QueueFullException: lambda exc: True}
 
     def add_default_exceptions_handler(
-        fs_app,
-        debug=False,
-        excs_to_status_codes=None,
-        silenced_excs=None,
-        log_filter=None,
+            fs_app,
+            debug=False,
+            excs_to_status_codes=None,
+            silenced_excs=None,
     ):
-        if excs_to_status_codes is None:
-            excs_to_status_codes = {
-                **DefaultErrorRoute.DEFAULT_STATUS_CODE_MAPPINGS,
-                **CeleryErrorRoute.DEFAULT_STATUS_CODE_MAPPINGS,
-            }
-        if silenced_excs is None:
-            silenced_excs = {
-                **DefaultErrorRoute.DEFAULT_SILENCED_EXCEPTIONS,
-                **CeleryErrorRoute.DEFAULT_SILENCED_EXCEPTIONS,
-            }
-        DefaultErrorRoute.add_default_exceptions_handler(fs_app, debug, excs_to_status_codes)
+        extra_status = {CeleryExecutor.QueueFullException: lambda exc: 429}
+        extra_silence = {CeleryExecutor.QueueFullException: lambda exc: True}
+
+        status_codes = {
+            **DefaultErrorRoute.DEFAULT_STATUS_CODE_MAPPINGS,
+            **(excs_to_status_codes or {}),
+            **extra_status,
+        }
+
+        silenced = {
+            **DefaultErrorRoute.DEFAULT_SILENCED_EXCEPTIONS,
+            **(silenced_excs or {}),
+            **extra_silence,
+        }
+
+        super(CeleryErrorRoute, CeleryErrorRoute).add_default_exceptions_handler(
+            fs_app, debug, status_codes, silenced
+        )

streamlit_octostar_utils/api_crafter/fastapi.py

@@ -241,20 +241,30 @@ class CommonModels(object):
             headers["Content-Disposition"] += f' filename="{filename}"'
             return StreamingResponse(zip_buffer, media_type="application/zip", headers=headers)
 
+
 class ErrorLogFilter(logging.Filter):
-    def __init__(self, silenced_excs: dict):
+    def __init__(self, silenced_excs: dict[type, callable]):
         super().__init__()
         self.silenced_excs = silenced_excs
-        
-    def filter(self, record):
-        exc_info = getattr(record, 'exc_info', None)
-        if exc_info:
-            if record.exc_info:
-                exc_type, exc_value, exc_tb = record.exc_info
-                if exc_value:
-                    for etype, cond in self.silenced_excs.items():
-                        if isinstance(exc_value, etype) and cond(exc_value):
-                            return False
+
+    def _silenced(self, exc_value) -> bool:
+        for etype, cond in self.silenced_excs.items():
+            if isinstance(exc_value, etype) and cond(exc_value):
+                return True
+        return False
+
+    def filter(self, record: logging.LogRecord) -> bool:
+        if record.exc_info:
+            exc_type, exc_value, _ = record.exc_info
+            if self._silenced(exc_value):
+                return False
+
+        msg = record.getMessage()
+        for etype in self.silenced_excs:
+            if etype.__name__ in msg:
+                if self.silenced_excs[etype](None):
+                    return False
+
         return True
 
 
@@ -334,10 +344,10 @@ class DefaultErrorRoute:
         return DefaultErrorRoute.format_error(exc, body, debug, excs_to_status_codes)
 
     def add_default_exceptions_handler(
-        fs_app,
-        debug=False,
-        excs_to_status_codes=None,
-        silenced_excs=None,
+            fs_app,
+            debug=False,
+            excs_to_status_codes=None,
+            silenced_excs=None,
     ):
         if excs_to_status_codes is None:
             excs_to_status_codes = DefaultErrorRoute.DEFAULT_STATUS_CODE_MAPPINGS
@@ -351,11 +361,12 @@ class DefaultErrorRoute:
         fs_app.add_exception_handler(RequestValidationError, _async_handle_error)
         fs_app.add_exception_handler(StarletteHTTPException, _async_handle_error)
         fs_app.add_exception_handler(Exception, _async_handle_error)
-        
+
         log_filter = ErrorLogFilter(silenced_excs)
-        logging.getLogger("uvicorn.error").addFilter(log_filter)
-        logging.getLogger("uvicorn").addFilter(log_filter)
-        logging.getLogger("fastapi").addFilter(log_filter)
+        for name in ("uvicorn.error", "uvicorn.access", "fastapi", "uvicorn"):
+            logging.getLogger(name).addFilter(log_filter)
+
+        logging.getLogger().addFilter(log_filter)
 
 class RequestCancelledMiddleware:
     def __init__(self, app):

{streamlit_octostar_utils-0.1.7a7.dist-info → streamlit_octostar_utils-0.1.7a9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: streamlit-octostar-utils
-Version: 0.1.7a7
+Version: 0.1.7a9
 Summary: 
 License: MIT
 Author: Octostar

{streamlit_octostar_utils-0.1.7a7.dist-info → streamlit_octostar_utils-0.1.7a9.dist-info}/RECORD

@@ -1,7 +1,7 @@
 streamlit_octostar_utils/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 streamlit_octostar_utils/api_crafter/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-streamlit_octostar_utils/api_crafter/celery.py,sha256=RRkL2W2L17ML3vsSgg5pbngWLAhubnMvvumkP21szkA,29771
-streamlit_octostar_utils/api_crafter/fastapi.py,sha256=s4WayFitfS2fX_FGtv1s56RmLB9rx7cfwuFTZjzrKoY,14133
+streamlit_octostar_utils/api_crafter/celery.py,sha256=bQq8P95j0XDWqx6Hdnm96PJL8kcp9ZcSKjveZUCQTVk,29874
+streamlit_octostar_utils/api_crafter/fastapi.py,sha256=vuy8cMY8as9GZc5cjBjZJXnudrqjYlPu1QGEyIKYKXw,14365
 streamlit_octostar_utils/api_crafter/nifi.py,sha256=fQ5k9eZl2oSQZ2BexZYwKUiO05-FryUi7wnCd_56P9Y,44375
 streamlit_octostar_utils/api_crafter/parser/__init__.py,sha256=YeYWF6sdQiCFV_RKNW2t9Vs6KJExE2pbXxWTe_DOayY,107
 streamlit_octostar_utils/api_crafter/parser/combine_fields.py,sha256=ddc44xkajw8MU0peAX_263DL7rPXbTKbHUjpOhRgvyU,8790
@@ -37,7 +37,7 @@ streamlit_octostar_utils/threading/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzp
 streamlit_octostar_utils/threading/async_task_manager.py,sha256=q7N6YZwUvIYMzkSHmsJNheNVCv93c03H6Hyg9uH8pvk,4747
 streamlit_octostar_utils/threading/session_callback_manager.py,sha256=LvZVP4g6tvKtYmI13f2j1sX_7hm61Groqp5xJine9_k,3973
 streamlit_octostar_utils/threading/session_state_hot_swapper.py,sha256=6eeCQI6A42hp4DmW2NQw2rbeR-k9N8DhfBKQdN_fbLU,811
-streamlit_octostar_utils-0.1.7a7.dist-info/LICENSE,sha256=dkwVPyV03fPHHtERnF6RnvRXcll__tud9gWca1RcgnQ,1073
-streamlit_octostar_utils-0.1.7a7.dist-info/METADATA,sha256=gziDnnF2kHGt2d5MucoV4I8ELE3tZlDQQWF_ggTGwk0,2267
-streamlit_octostar_utils-0.1.7a7.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-streamlit_octostar_utils-0.1.7a7.dist-info/RECORD,,
+streamlit_octostar_utils-0.1.7a9.dist-info/LICENSE,sha256=dkwVPyV03fPHHtERnF6RnvRXcll__tud9gWca1RcgnQ,1073
+streamlit_octostar_utils-0.1.7a9.dist-info/METADATA,sha256=VhXEOXyzVQhwg8qgd8XdBep95_fyXTk-4Lq2JhdpkzU,2267
+streamlit_octostar_utils-0.1.7a9.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+streamlit_octostar_utils-0.1.7a9.dist-info/RECORD,,