cleanup

psimsa · Jul 6, 2023 · 7b5d6fa · 7b5d6fa
1 parent 88e6929
commit 7b5d6fa
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 19 deletions.
diff --git a/custom_components/oig_cloud/__init__.py b/custom_components/oig_cloud/__init__.py
@@ -2,7 +2,6 @@
 import hashlib
 
 from opentelemetry import trace
-from opentelemetry.sdk.resources import Resource
 
 from .api import oig_cloud_api
 
@@ -13,7 +12,6 @@
 from .shared.tracing import setup_tracing
 from .shared.logging import setup_otel_logging
 
-from opentelemetry._logs import set_logger_provider, get_logger_provider
 
 tracer = trace.get_tracer(__name__)
 

diff --git a/custom_components/oig_cloud/api/oig_cloud_api.py b/custom_components/oig_cloud/api/oig_cloud_api.py
@@ -38,17 +38,7 @@ def __init__(
 
             self._last_update = datetime.datetime(1, 1, 1, 0, 0)
             self._username = username
-            self._password = password
-
-            if not self._no_telemetry:
-                span.set_attributes(
-                    {
-                        "hass.language": hass.config.language,
-                        "hass.time_zone": hass.config.time_zone,
-                    }
-                )
-                span.add_event("log", {"level": logging.INFO, "msg": "Initializing"})
-
+            self._password = password                
 
             self.last_state = None
             self._logger.debug("OigCloud initialized")
@@ -102,12 +92,11 @@ def get_session(self) -> aiohttp.ClientSession:
     async def get_stats(self) -> object:
         async with lock:
             current_time = datetime.datetime.now()
-            if (current_time - self._last_update).total_seconds() < 10:
+            if (current_time - self._last_update).total_seconds() < 30:
                 self._logger.debug("Using cached stats")
                 return self.last_state
             with tracer.start_as_current_span("get_stats") as span:
                 try:
-
                     to_return: object = None
                     try:
                         to_return = await self.get_stats_internal()

diff --git a/custom_components/oig_cloud/shared/logging.py b/custom_components/oig_cloud/shared/logging.py
@@ -15,9 +15,9 @@
 
 def setup_otel_logging(email_hash:str, hass_id: str) -> LoggingHandler:
 
-    OT_RESOURCE = get_resource(email_hash, hass_id)
+    resource = get_resource(email_hash, hass_id)
 
-    logger_provider = LoggerProvider(resource=OT_RESOURCE)
+    logger_provider = LoggerProvider(resource=resource)
     set_logger_provider(logger_provider)
 
     exporter = OTLPLogExporter(

diff --git a/custom_components/oig_cloud/shared/tracing.py b/custom_components/oig_cloud/shared/tracing.py
@@ -8,9 +8,9 @@
 from .shared import get_resource
 
 def setup_tracing(email_hash:str, hass_id: str):
-    OT_RESOURCE = get_resource(email_hash, hass_id)
+    resource = get_resource(email_hash, hass_id)
 
-    trace_provider = TracerProvider(resource=OT_RESOURCE)
+    trace_provider = TracerProvider(resource=resource)
 
     trace_processor = BatchSpanProcessor(
         OTLPSpanExporter(