canvas 0.34.1__py3-none-any.whl → 0.35.0__py3-none-any.whl

settings.py

@@ -14,6 +14,7 @@ ENV = os.getenv("ENV", "development")
 IS_PRODUCTION = ENV == "production"
 IS_PRODUCTION_CUSTOMER = env_to_bool("IS_PRODUCTION_CUSTOMER", IS_PRODUCTION)
 IS_TESTING = env_to_bool("IS_TESTING", "pytest" in sys.argv[0] or sys.argv[0] == "-c")
+IS_SCRIPT = env_to_bool("IS_SCRIPT", "plugin_runner.py" not in sys.argv[0])
 CUSTOMER_IDENTIFIER = os.getenv("CUSTOMER_IDENTIFIER", "local")
 APP_NAME = os.getenv("APP_NAME")
 
@@ -24,6 +25,9 @@ INTEGRATION_TEST_CLIENT_SECRET = os.getenv("INTEGRATION_TEST_CLIENT_SECRET")
 GRAPHQL_ENDPOINT = os.getenv("GRAPHQL_ENDPOINT", "http://localhost:8000/plugins-graphql")
 REDIS_ENDPOINT = os.getenv("REDIS_ENDPOINT", f"redis://{APP_NAME}-redis:6379")
 
+
+METRICS_ENABLED = env_to_bool("PLUGINS_METRICS_ENABLED", not IS_SCRIPT)
+
 INSTALLED_APPS = [
     "canvas_sdk.v1",
 ]

canvas_sdk/utils/stats.py

@@ -1,74 +0,0 @@
-from datetime import timedelta
-from time import time
-from typing import Any
-
-from statsd.defaults.env import statsd as default_statsd_client
-
-
-def get_duration_ms(start_time: float) -> int:
-    """Get the duration in milliseconds since the given start time."""
-    return int((time() - start_time) * 1000)
-
-
-LINE_PROTOCOL_TRANSLATION = str.maketrans(
-    {
-        ",": r"\,",
-        "=": r"\=",
-        " ": r"\ ",
-        ":": r"__",
-    }
-)
-
-
-def tags_to_line_protocol(tags: dict[str, Any]) -> str:
-    """Generate a tags string compatible with the InfluxDB line protocol.
-
-    See: https://docs.influxdata.com/influxdb/v1.1/write_protocols/line_protocol_tutorial/
-    """
-    return ",".join(
-        f"{tag_name}={str(tag_value).translate(LINE_PROTOCOL_TRANSLATION)}"
-        for tag_name, tag_value in tags.items()
-    )
-
-
-STATS_ENABLED = True
-
-
-class StatsDClientProxy:
-    """Proxy for a StatsD client."""
-
-    def __init__(self) -> None:
-        self.client = default_statsd_client
-
-    def gauge(self, metric_name: str, value: float, tags: dict[str, str]) -> None:
-        """Sends a gauge metric to StatsD with properly formatted tags.
-
-        Args:
-            metric_name (str): The name of the metric.
-            value (float): The value to report.
-            tags (dict[str, str]): Dictionary of tags to attach to the metric.
-        """
-        if not STATS_ENABLED:
-            return
-
-        statsd_tags = tags_to_line_protocol(tags)
-        self.client.gauge(f"{metric_name},{statsd_tags}", value)
-
-    def timing(self, metric_name: str, delta: float | timedelta, tags: dict[str, str]) -> None:
-        """Sends a timing metric to StatsD with properly formatted tags.
-
-        Args:
-            metric_name (str): The name of the metric.
-            delta (float | timedelta): The value to report.
-            tags (dict[str, str]): Dictionary of tags to attach to the metric.
-        """
-        if not STATS_ENABLED:
-            return
-
-        statsd_tags = tags_to_line_protocol(tags)
-        self.client.timing(f"{metric_name},{statsd_tags}", delta)
-
-
-statsd_client = StatsDClientProxy()
-
-__exports__ = ()