fluidattacks-core 5.0.1__py3-none-any.whl → 5.1.0__py3-none-any.whl

fluidattacks_core/logging/filters.py

@@ -1,6 +1,6 @@
 import logging
 
-from fluidattacks_core.logging.sources.utils import get_environment
+from fluidattacks_core.logging.sources.utils import get_env_var, get_environment
 
 
 class NoProductionFilter(logging.Filter):
@@ -11,3 +11,13 @@ class NoProductionFilter(logging.Filter):
 class ProductionOnlyFilter(logging.Filter):
     def filter(self, _record: logging.LogRecord) -> bool:
         return get_environment() == "production"
+
+
+class ErrorOnlyFilter(logging.Filter):
+    def filter(self, _record: logging.LogRecord) -> bool:
+        return _record.levelno >= logging.ERROR
+
+
+class EnabledTelemetryFilter(logging.Filter):
+    def filter(self, _record: logging.LogRecord) -> bool:
+        return str(get_env_var("TELEMETRY_OPT_OUT")).lower() != "true"

fluidattacks_core/logging/handlers.py

@@ -7,7 +7,12 @@ from typing import TextIO
 
 import simplejson as json
 
-from fluidattacks_core.logging.filters import NoProductionFilter, ProductionOnlyFilter
+from fluidattacks_core.logging.filters import (
+    EnabledTelemetryFilter,
+    ErrorOnlyFilter,
+    NoProductionFilter,
+    ProductionOnlyFilter,
+)
 from fluidattacks_core.logging.formatters import ColorfulFormatter, CustomJsonFormatter
 
 
@@ -114,7 +119,14 @@ class DatadogLogsHandler(HTTPHandler):
 
 
 class TelemetryAsyncHandler(CustomQueueHandler):
-    """Logging handler for sending logs to telemetry services asynchronously."""
+    """Logging handler for sending logs to telemetry services asynchronously.
+
+    To enable telemetry, set the `TELEMETRY_OPT_OUT` environment variable different from `true`.
+
+    Includes:
+    - Filters: `ErrorOnlyFilter`, `EnabledTelemetryFilter`
+    - Formatter: `CustomJsonFormatter`
+    """
 
     def __init__(self, service: str, source: str, dd_client_token: str) -> None:
         """Initialize the TelemetryAsyncHandler.
@@ -125,6 +137,9 @@ class TelemetryAsyncHandler(CustomQueueHandler):
             dd_client_token: The Datadog Client Token.
 
         """
+        self.addFilter(ErrorOnlyFilter())
+        self.addFilter(EnabledTelemetryFilter())
+
         handler = DatadogLogsHandler(service, source, dd_client_token)
         handler.setFormatter(CustomJsonFormatter())
 

{fluidattacks_core-5.0.1.dist-info → fluidattacks_core-5.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fluidattacks-core
-Version: 5.0.1
+Version: 5.1.0
 Summary: Fluid Attacks Core Library
 Author-email: Development <development@fluidattacks.com>
 License: MPL-2.0

{fluidattacks_core-5.0.1.dist-info → fluidattacks_core-5.1.0.dist-info}/RECORD

@@ -30,9 +30,9 @@ fluidattacks_core/http/__init__.py,sha256=3Zz90L6e3_z-M-8Bvk_53rv-CFhPThkRGXnxCi
 fluidattacks_core/http/client.py,sha256=jIhtGU2cKi5GZbxHq8WJOPgnk0beScRtxlz9tBSaKuw,2454
 fluidattacks_core/http/validations.py,sha256=h10Hr906KJqda1rJJb8eOqk1Xyyz81lAJ1glXeae4kM,3766
 fluidattacks_core/logging/__init__.py,sha256=y6D12LrvrsMwaveQn5C4Em3RyeS6mP6E9fRpq7gqS4o,1546
-fluidattacks_core/logging/filters.py,sha256=v03EWIbCGLKc6sdSQnO7ealxMdPzcJhd20rGr1PBZrE,388
+fluidattacks_core/logging/filters.py,sha256=jEJkrvFeoQxCJ9vI8rd4N1FWXTbPdiTlv-7kriB8dQ0,725
 fluidattacks_core/logging/formatters.py,sha256=SCA4k9XvHJknmkTv63uiCBU31VrWOsgA7llEXYYj1uQ,6063
-fluidattacks_core/logging/handlers.py,sha256=7fjCG0AqwlF0PRmH6MwkYRO1qdVoRjyYAAwML3A5lXY,4658
+fluidattacks_core/logging/handlers.py,sha256=X-brVAc-QOqSFiEKqbBmxirkLLAaZDi5ID6cYzKdoHg,5028
 fluidattacks_core/logging/presets.py,sha256=KU6d6PI61kklJ_o7NgAzU1DahEPM0KwwjTYHo2naHv8,939
 fluidattacks_core/logging/utils.py,sha256=jbAcwr0L6iPsId3dYEp-vAbtFex2UUU2l2iIk1F60BE,1115
 fluidattacks_core/logging/sources/__init__.py,sha256=bUOsNhmBq9WHLvPxCW3G0Az4D8Wunp0mCCqEYlCsjsI,5937
@@ -44,7 +44,7 @@ fluidattacks_core/semver/match_versions.py,sha256=10Vc3aS5Q2WVeM6UYUNiAo8C4E7zFT
 fluidattacks_core/serializers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 fluidattacks_core/serializers/snippet.py,sha256=e520pZHC-fsNuYVNY30A7TcSugvUlFL6xdr74j5aCDM,12780
 fluidattacks_core/serializers/syntax.py,sha256=eSpoAmWLaVbd8tQ8ghHIvjJlCtieEf_IimonqLpIe3Y,16020
-fluidattacks_core-5.0.1.dist-info/METADATA,sha256=E52TwLj71JPrMUmgZ24JrHHaDPhOH_uTbWAd9dyTPfw,3453
-fluidattacks_core-5.0.1.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
-fluidattacks_core-5.0.1.dist-info/top_level.txt,sha256=m49ZyZ2zPQmDBxkSpjb20wr-ZbGVXdOMFBZrDiP5Lb8,18
-fluidattacks_core-5.0.1.dist-info/RECORD,,
+fluidattacks_core-5.1.0.dist-info/METADATA,sha256=atsMg03afFaIl0BmNom7s3Oecao6wEV7x00__o91GKc,3453
+fluidattacks_core-5.1.0.dist-info/WHEEL,sha256=qELbo2s1Yzl39ZmrAibXA2jjPLUYfnVhUNTlyF1rq0Y,92
+fluidattacks_core-5.1.0.dist-info/top_level.txt,sha256=m49ZyZ2zPQmDBxkSpjb20wr-ZbGVXdOMFBZrDiP5Lb8,18
+fluidattacks_core-5.1.0.dist-info/RECORD,,