Adding TA to uwsgi Split constructor

Author: sarrubia

splitio/uwsgi.py

@@ -377,7 +377,9 @@ def _parse_split(self, split, block_until_ready=False):
             split['name'], split['seed'], split['killed'],
             split['defaultTreatment'], split['trafficTypeName'],
             split['status'], split['changeNumber'],
-            segment_cache=self._segment_cache, algo=split.get('algo')
+            segment_cache=self._segment_cache, algo=split.get('algo'),
+            traffic_allocation=split.get('trafficAllocation'),
+            traffic_allocation_seed=split.get('trafficAllocationSeed')
         )
 
     def _parse_matcher_in_segment(self, partial_split, matcher, block_until_ready=False, *args,
@@ -389,7 +391,9 @@ def _parse_matcher_in_segment(self, partial_split, matcher, block_until_ready=Fa
         return delegate
 
 class UWSGISplit(Split):
-    def __init__(self, name, seed, killed, default_treatment, traffic_type_name, status, change_number, conditions=None, segment_cache=None, algo=None):
+    def __init__(self, name, seed, killed, default_treatment, traffic_type_name, status, change_number, conditions=None, segment_cache=None, algo=None,
+                 traffic_allocation=None,
+                 traffic_allocation_seed=None):
         """A split implementation that mantains a reference to the segment cache so segments can
         be easily pickled and unpickled.
         :param name: Name of the feature
@@ -407,7 +411,8 @@ def __init__(self, name, seed, killed, default_treatment, traffic_type_name, sta
         """
         super(UWSGISplit, self).__init__(
             name, seed, killed, default_treatment, traffic_type_name, status,
-            change_number, conditions, algo)
+            change_number, conditions, algo, traffic_allocation,
+            traffic_allocation_seed)
         self._segment_cache = segment_cache
 
     @property