Skip to content

Commit c15e337

Browse files
committed
polish
1 parent 8a48ab4 commit c15e337

File tree

1 file changed

+39
-39
lines changed

1 file changed

+39
-39
lines changed

splitio/models/telemetry.py

Lines changed: 39 additions & 39 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ class CounterConstants(Enum):
2727
EVENTS_QUEUED = 'eventsQueued'
2828
EVENTS_DROPPED = 'eventsDropped'
2929

30-
class ConfigParams(Enum):
30+
class _ConfigParams(Enum):
3131
"""Config parameters constants"""
3232
SPLITS_REFRESH_RATE = 'featuresRefreshRate'
3333
SEGMENTS_REFRESH_RATE = 'segmentsRefreshRate'
@@ -42,7 +42,7 @@ class ConfigParams(Enum):
4242
IMPRESSIONS_MODE = 'impressionsMode'
4343
IMPRESSIONS_LISTENER = 'impressionListener'
4444

45-
class ExtraConfig(Enum):
45+
class _ExtraConfig(Enum):
4646
"""Extra config constants"""
4747
ACTIVE_FACTORY_COUNT = 'activeFactoryCount'
4848
REDUNDANT_FACTORY_COUNT = 'redundantFactoryCount'
@@ -53,7 +53,7 @@ class ExtraConfig(Enum):
5353
HTTP_PROXY = 'httpProxy'
5454
HTTPS_PROXY_ENV = 'HTTPS_PROXY'
5555

56-
class ApiURLs(Enum):
56+
class _ApiURLs(Enum):
5757
"""Api URL constants"""
5858
SDK_URL = 'sdk_url'
5959
EVENTS_URL = 'events_url'
@@ -88,7 +88,7 @@ class MethodExceptionsAndLatencies(Enum):
8888
TREATMENTS_WITH_CONFIG_BY_FLAG_SETS = 'treatments_with_config_by_flag_sets'
8989
TRACK = 'track'
9090

91-
class LastSynchronizationConstants(Enum):
91+
class _LastSynchronizationConstants(Enum):
9292
"""Last sync constants"""
9393
LAST_SYNCHRONIZATIONS = 'lastSynchronizations'
9494

@@ -108,7 +108,7 @@ class SSESyncMode(Enum):
108108
STREAMING = 0
109109
POLLING = 1
110110

111-
class StreamingEventsConstant(Enum):
111+
class _StreamingEventsConstant(Enum):
112112
"""Storage types constant"""
113113
STREAMING_EVENTS = 'streamingEvents'
114114

@@ -426,7 +426,7 @@ def get_all(self):
426426
:rtype: dict
427427
"""
428428
with self._lock:
429-
return {LastSynchronizationConstants.LAST_SYNCHRONIZATIONS.value: {HTTPExceptionsAndLatencies.SPLIT.value: self._split, HTTPExceptionsAndLatencies.SEGMENT.value: self._segment, HTTPExceptionsAndLatencies.IMPRESSION.value: self._impression,
429+
return {_LastSynchronizationConstants.LAST_SYNCHRONIZATIONS.value: {HTTPExceptionsAndLatencies.SPLIT.value: self._split, HTTPExceptionsAndLatencies.SEGMENT.value: self._segment, HTTPExceptionsAndLatencies.IMPRESSION.value: self._impression,
430430
HTTPExceptionsAndLatencies.IMPRESSION_COUNT.value: self._impression_count, HTTPExceptionsAndLatencies.EVENT.value: self._event,
431431
HTTPExceptionsAndLatencies.TELEMETRY.value: self._telemetry, HTTPExceptionsAndLatencies.TOKEN.value: self._token}
432432
}
@@ -760,7 +760,7 @@ def pop_streaming_events(self):
760760
with self._lock:
761761
streaming_events = self._streaming_events
762762
self._streaming_events = []
763-
return {StreamingEventsConstant.STREAMING_EVENTS.value: [{'e': streaming_event.type, 'd': streaming_event.data,
763+
return {_StreamingEventsConstant.STREAMING_EVENTS.value: [{'e': streaming_event.type, 'd': streaming_event.data,
764764
't': streaming_event.time} for streaming_event in streaming_events]}
765765

766766
class TelemetryConfig(object):
@@ -782,10 +782,10 @@ def _reset_all(self):
782782
self._operation_mode = None
783783
self._storage_type = None
784784
self._streaming_enabled = None
785-
self._refresh_rate = {ConfigParams.SPLITS_REFRESH_RATE.value: 0, ConfigParams.SEGMENTS_REFRESH_RATE.value: 0,
786-
ConfigParams.IMPRESSIONS_REFRESH_RATE.value: 0, ConfigParams.EVENTS_REFRESH_RATE.value: 0, ConfigParams.TELEMETRY_REFRESH_RATE.value: 0}
787-
self._url_override = {ApiURLs.SDK_URL.value: False, ApiURLs.EVENTS_URL.value: False, ApiURLs.AUTH_URL.value: False,
788-
ApiURLs.STREAMING_URL.value: False, ApiURLs.TELEMETRY_URL.value: False}
785+
self._refresh_rate = {_ConfigParams.SPLITS_REFRESH_RATE.value: 0, _ConfigParams.SEGMENTS_REFRESH_RATE.value: 0,
786+
_ConfigParams.IMPRESSIONS_REFRESH_RATE.value: 0, _ConfigParams.EVENTS_REFRESH_RATE.value: 0, _ConfigParams.TELEMETRY_REFRESH_RATE.value: 0}
787+
self._url_override = {_ApiURLs.SDK_URL.value: False, _ApiURLs.EVENTS_URL.value: False, _ApiURLs.AUTH_URL.value: False,
788+
_ApiURLs.STREAMING_URL.value: False, _ApiURLs.TELEMETRY_URL.value: False}
789789
self._impressions_queue_size = 0
790790
self._events_queue_size = 0
791791
self._impressions_mode = None
@@ -818,15 +818,15 @@ def record_config(self, config, extra_config):
818818
:type config: dict
819819
"""
820820
with self._lock:
821-
self._operation_mode = self._get_operation_mode(config[ConfigParams.OPERATION_MODE.value])
822-
self._storage_type = self._get_storage_type(config[ConfigParams.OPERATION_MODE.value], config[ConfigParams.STORAGE_TYPE.value])
823-
self._streaming_enabled = config[ConfigParams.STREAMING_ENABLED.value]
821+
self._operation_mode = self._get_operation_mode(config[_ConfigParams.OPERATION_MODE.value])
822+
self._storage_type = self._get_storage_type(config[_ConfigParams.OPERATION_MODE.value], config[_ConfigParams.STORAGE_TYPE.value])
823+
self._streaming_enabled = config[_ConfigParams.STREAMING_ENABLED.value]
824824
self._refresh_rate = self._get_refresh_rates(config)
825825
self._url_override = self._get_url_overrides(extra_config)
826-
self._impressions_queue_size = config[ConfigParams.IMPRESSIONS_QUEUE_SIZE.value]
827-
self._events_queue_size = config[ConfigParams.EVENTS_QUEUE_SIZE.value]
828-
self._impressions_mode = self._get_impressions_mode(config[ConfigParams.IMPRESSIONS_MODE.value])
829-
self._impression_listener = True if config[ConfigParams.IMPRESSIONS_LISTENER.value] is not None else False
826+
self._impressions_queue_size = config[_ConfigParams.IMPRESSIONS_QUEUE_SIZE.value]
827+
self._events_queue_size = config[_ConfigParams.EVENTS_QUEUE_SIZE.value]
828+
self._impressions_mode = self._get_impressions_mode(config[_ConfigParams.IMPRESSIONS_MODE.value])
829+
self._impression_listener = True if config[_ConfigParams.IMPRESSIONS_LISTENER.value] is not None else False
830830
self._http_proxy = self._check_if_proxy_detected()
831831

832832
def record_active_and_redundant_factories(self, active_factory_count, redundant_factory_count):
@@ -913,16 +913,16 @@ def get_stats(self):
913913
'oM': self._operation_mode,
914914
'sT': self._storage_type,
915915
'sE': self._streaming_enabled,
916-
'rR': {'sp': self._refresh_rate[ConfigParams.SPLITS_REFRESH_RATE.value],
917-
'se': self._refresh_rate[ConfigParams.SEGMENTS_REFRESH_RATE.value],
918-
'im': self._refresh_rate[ConfigParams.IMPRESSIONS_REFRESH_RATE.value],
919-
'ev': self._refresh_rate[ConfigParams.EVENTS_REFRESH_RATE.value],
920-
'te': self._refresh_rate[ConfigParams.TELEMETRY_REFRESH_RATE.value]},
921-
'uO': {'s': self._url_override[ApiURLs.SDK_URL.value],
922-
'e': self._url_override[ApiURLs.EVENTS_URL.value],
923-
'a': self._url_override[ApiURLs.AUTH_URL.value],
924-
'st': self._url_override[ApiURLs.STREAMING_URL.value],
925-
't': self._url_override[ApiURLs.TELEMETRY_URL.value]},
916+
'rR': {'sp': self._refresh_rate[_ConfigParams.SPLITS_REFRESH_RATE.value],
917+
'se': self._refresh_rate[_ConfigParams.SEGMENTS_REFRESH_RATE.value],
918+
'im': self._refresh_rate[_ConfigParams.IMPRESSIONS_REFRESH_RATE.value],
919+
'ev': self._refresh_rate[_ConfigParams.EVENTS_REFRESH_RATE.value],
920+
'te': self._refresh_rate[_ConfigParams.TELEMETRY_REFRESH_RATE.value]},
921+
'uO': {'s': self._url_override[_ApiURLs.SDK_URL.value],
922+
'e': self._url_override[_ApiURLs.EVENTS_URL.value],
923+
'a': self._url_override[_ApiURLs.AUTH_URL.value],
924+
'st': self._url_override[_ApiURLs.STREAMING_URL.value],
925+
't': self._url_override[_ApiURLs.TELEMETRY_URL.value]},
926926
'iQ': self._impressions_queue_size,
927927
'eQ': self._events_queue_size,
928928
'iM': self._impressions_mode,
@@ -981,11 +981,11 @@ def _get_refresh_rates(self, config):
981981
"""
982982
with self._lock:
983983
return {
984-
ConfigParams.SPLITS_REFRESH_RATE.value: config[ConfigParams.SPLITS_REFRESH_RATE.value],
985-
ConfigParams.SEGMENTS_REFRESH_RATE.value: config[ConfigParams.SEGMENTS_REFRESH_RATE.value],
986-
ConfigParams.IMPRESSIONS_REFRESH_RATE.value: config[ConfigParams.IMPRESSIONS_REFRESH_RATE.value],
987-
ConfigParams.EVENTS_REFRESH_RATE.value: config[ConfigParams.EVENTS_REFRESH_RATE.value],
988-
ConfigParams.TELEMETRY_REFRESH_RATE.value: config[ConfigParams.TELEMETRY_REFRESH_RATE.value]
984+
_ConfigParams.SPLITS_REFRESH_RATE.value: config[_ConfigParams.SPLITS_REFRESH_RATE.value],
985+
_ConfigParams.SEGMENTS_REFRESH_RATE.value: config[_ConfigParams.SEGMENTS_REFRESH_RATE.value],
986+
_ConfigParams.IMPRESSIONS_REFRESH_RATE.value: config[_ConfigParams.IMPRESSIONS_REFRESH_RATE.value],
987+
_ConfigParams.EVENTS_REFRESH_RATE.value: config[_ConfigParams.EVENTS_REFRESH_RATE.value],
988+
_ConfigParams.TELEMETRY_REFRESH_RATE.value: config[_ConfigParams.TELEMETRY_REFRESH_RATE.value]
989989
}
990990

991991
def _get_url_overrides(self, config):
@@ -1000,11 +1000,11 @@ def _get_url_overrides(self, config):
10001000
"""
10011001
with self._lock:
10021002
return {
1003-
ApiURLs.SDK_URL.value: True if ApiURLs.SDK_URL.value in config else False,
1004-
ApiURLs.EVENTS_URL.value: True if ApiURLs.EVENTS_URL.value in config else False,
1005-
ApiURLs.AUTH_URL.value: True if ApiURLs.AUTH_URL.value in config else False,
1006-
ApiURLs.STREAMING_URL.value: True if ApiURLs.STREAMING_URL.value in config else False,
1007-
ApiURLs.TELEMETRY_URL.value: True if ApiURLs.TELEMETRY_URL.value in config else False
1003+
_ApiURLs.SDK_URL.value: True if _ApiURLs.SDK_URL.value in config else False,
1004+
_ApiURLs.EVENTS_URL.value: True if _ApiURLs.EVENTS_URL.value in config else False,
1005+
_ApiURLs.AUTH_URL.value: True if _ApiURLs.AUTH_URL.value in config else False,
1006+
_ApiURLs.STREAMING_URL.value: True if _ApiURLs.STREAMING_URL.value in config else False,
1007+
_ApiURLs.TELEMETRY_URL.value: True if _ApiURLs.TELEMETRY_URL.value in config else False
10081008
}
10091009

10101010
def _get_impressions_mode(self, imp_mode):
@@ -1034,6 +1034,6 @@ def _check_if_proxy_detected(self):
10341034
"""
10351035
with self._lock:
10361036
for x in os.environ:
1037-
if x.upper() == ExtraConfig.HTTPS_PROXY_ENV.value:
1037+
if x.upper() == _ExtraConfig.HTTPS_PROXY_ENV.value:
10381038
return True
10391039
return False

0 commit comments

Comments
 (0)