clean up

Author: chillaq

splitio/models/telemetry.py

@@ -411,6 +411,7 @@ def add_error(self, resource, status):
         :param status: http error code
         :type status: str
         """
+        status = str(status)
         with self._lock:
             if resource == HTTPExceptionsAndLatencies.SPLIT:
                 if status not in self._split:

splitio/push/status_tracker.py

@@ -81,11 +81,11 @@ def handle_occupancy(self, event):
         self._timestamps.occupancy = event.timestamp
 
         self._publishers[event.channel] = event.publishers
-        if event.channel[-3:] == 'pri':
-            event_type = StreamingEventTypes.OCCUPANCY_PRI
-        else:
-            event_type = StreamingEventTypes.OCCUPANCY_SEC
-        self._telemetry_runtime_producer.record_streaming_event((event_type, len(self._publishers), event.timestamp))
+        self._telemetry_runtime_producer.record_streaming_event((
+            StreamingEventTypes.OCCUPANCY_PRI if event.channel[-3:] == 'pri' else StreamingEventTypes.OCCUPANCY_SEC,
+            len(self._publishers),
+            event.timestamp
+        ))
         return self._update_status()
 
     def handle_control_message(self, event):

splitio/storage/inmemmory.py

@@ -532,7 +532,7 @@ def record_successful_sync(self, resource, time):
 
     def record_sync_error(self, resource, status):
         """Record sync http error."""
-        self._http_sync_errors.add_error(resource, str(status))
+        self._http_sync_errors.add_error(resource, status)
 
     def record_sync_latency(self, resource, latency):
         """Record latency time."""