kameleoon-client-python 3.9.0__tar.gz → 3.10.0__tar.gz

{kameleoon-client-python-3.9.0 → kameleoon-client-python-3.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kameleoon-client-python
-Version: 3.9.0
+Version: 3.10.0
 Summary: Kameleoon Client Python Software Development Kit.
 Home-page: https://developers.kameleoon.com/python-sdk.html
 Author: Kameleoon

kameleoon-client-python-3.10.0/kameleoon/data/cbscores.py

@@ -0,0 +1,66 @@
+"""CBScores"""
+
+from typing import Dict, Iterable, List
+from kameleoon.helpers.string_utils import StringUtils
+from kameleoon.data import DataType
+from kameleoon.data.data import BaseData
+
+
+# pylint: disable=R0903
+class ScoredVarId:
+    """ScoredVarId"""
+    def __init__(self, variation_id: int, score: float) -> None:
+        self.variation_id = variation_id
+        self.score = score
+
+
+class VarGroup:
+    """VarGroup"""
+    def __init__(self, ids: List[int]) -> None:
+        ids.sort()
+        self.__ids = ids
+
+    @property
+    def ids(self) -> List[int]:
+        """Returns the IDs"""
+        return self.__ids
+
+    def __str__(self) -> str:
+        return f"VarGroup{{ids:{self.__ids}}}"
+
+
+class CBScores(BaseData):
+    """CBScores"""
+
+    def __init__(self, cbs_map: Dict[int, Iterable[ScoredVarId]]) -> None:
+        super().__init__()
+        self.__values = {cbs_key: self.__extract_var_ids(cbs_value) for cbs_key, cbs_value in cbs_map.items()}
+
+    @property
+    def values(self) -> Dict[int, List[VarGroup]]:
+        """
+        Returns the CB variation groups stored by experiment.
+
+        keys = experiment IDs / values = list of variation IDs ordered descending
+        by score (there may be several variation ids with same score)
+        """
+        return self.__values
+
+    @property
+    def data_type(self) -> DataType:
+        return DataType.CBS
+
+    @staticmethod
+    def __extract_var_ids(scores: Iterable[ScoredVarId]) -> List[VarGroup]:
+        grouped: Dict[float, List[int]] = {}
+        for score in scores:
+            glist = grouped.get(score.score)
+            if glist:
+                glist.append(score.variation_id)
+            else:
+                grouped[score.score] = [score.variation_id]
+        return [VarGroup(grouped[score]) for score in sorted(grouped, reverse=True)]
+
+    def __str__(self) -> str:
+        values = StringUtils.object_to_string(self.__values)
+        return f"CBScores{{values:{values}}}"

{kameleoon-client-python-3.9.0 → kameleoon-client-python-3.10.0}/kameleoon/data/data.py

@@ -22,6 +22,7 @@ class DataType(IntEnum):
     UNIQUE_IDENTIFIER = 13
     FORCED_EXPERIMENT_VARIATION = 14
     FORCED_FEATURE_VARIATION = 15
+    CBS = 16
 
 
 class BaseData:

{kameleoon-client-python-3.9.0 → kameleoon-client-python-3.10.0}/kameleoon/data/manager/visitor.py

@@ -5,6 +5,7 @@ from typing import Any, Callable, Optional, Dict, List, Iterator, Iterable, cast
 from kameleoon.data.visitor_visits import VisitorVisits
 from kameleoon.data.data import Data
 from kameleoon.data.browser import Browser
+from kameleoon.data.cbscores import CBScores
 from kameleoon.data.conversion import Conversion
 from kameleoon.data.cookie import Cookie
 from kameleoon.data.custom_data import CustomData
@@ -172,6 +173,13 @@ class Visitor:  # pylint: disable=R0904
         KameleoonLogger.debug('CALL/RETURN: Visitor.kcs_heat -> (kcs_heat: %s)', kcs_heat)
         return kcs_heat
 
+    @property
+    def cbscores(self) -> Optional[CBScores]:
+        """Returns the CBS data associated with this Visitor."""
+        cbs = self._data.cbscores
+        KameleoonLogger.debug('CALL/RETURN: Visitor.cbscores -> (cbs: %s)', cbs)
+        return cbs
+
     @property
     def visitor_visits(self) -> Optional[VisitorVisits]:
         """Returns the visitor visits data associated with this Visitor."""
@@ -261,6 +269,7 @@ class Visitor:  # pylint: disable=R0904
             self.forced_variations: Optional[Dict[int, ForcedExperimentVariation]] = None
             self.simulated_variations: Optional[Dict[str, ForcedFeatureVariation]] = None
             self.kcs_heat: Optional[KcsHeat] = None
+            self.cbscores: Optional[CBScores] = None
             self.visitor_visits: Optional[VisitorVisits] = None
             self.mapping_identifier: Optional[str] = None
             self.legal_consent = False
@@ -403,6 +412,10 @@ class Visitor:  # pylint: disable=R0904
     def _set_kcs_heat(self, context: DataAddingContext) -> None:
         self._data.kcs_heat = cast(KcsHeat, context.data)
 
+    def _set_cbscores(self, context: DataAddingContext) -> None:
+        if context.overwrite or self._data.cbscores is None:
+            self._data.cbscores = cast(CBScores, context.data)
+
     def _set_visitor_visits(self, context: DataAddingContext) -> None:
         self._data.visitor_visits = cast(VisitorVisits, context.data)
 
@@ -425,6 +438,7 @@ class Visitor:  # pylint: disable=R0904
         PageViewVisit: _add_page_view_visit,
         Conversion: _add_conversion,
         KcsHeat: _set_kcs_heat,
+        CBScores: _set_cbscores,
         VisitorVisits: _set_visitor_visits,
         AssignedVariation: _add_variation,
         ForcedFeatureVariation: _add_forced_feature_variation,