Skip to content

Commit bac9e80

Browse files
committed
Removed properties if none in sender
1 parent 24b63d9 commit bac9e80

File tree

2 files changed

+27
-12
lines changed

2 files changed

+27
-12
lines changed

splitio/api/impressions.py

Lines changed: 25 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -30,16 +30,7 @@ def _build_bulk(impressions):
3030
{
3131
'f': test_name,
3232
'i': [
33-
{
34-
'k': impression.matching_key,
35-
't': impression.treatment,
36-
'm': impression.time,
37-
'c': impression.change_number,
38-
'r': impression.label,
39-
'b': impression.bucketing_key,
40-
'pt': impression.previous_time,
41-
'properties': impression.properties
42-
}
33+
ImpressionsAPIBase._filter_out_null_prop(impression)
4334
for impression in imps
4435
]
4536
}
@@ -49,6 +40,30 @@ def _build_bulk(impressions):
4940
)
5041
]
5142

43+
@staticmethod
44+
def _filter_out_null_prop(impression):
45+
if impression.properties == None:
46+
return {
47+
'k': impression.matching_key,
48+
't': impression.treatment,
49+
'm': impression.time,
50+
'c': impression.change_number,
51+
'r': impression.label,
52+
'b': impression.bucketing_key,
53+
'pt': impression.previous_time
54+
}
55+
56+
return {
57+
'k': impression.matching_key,
58+
't': impression.treatment,
59+
'm': impression.time,
60+
'c': impression.change_number,
61+
'r': impression.label,
62+
'b': impression.bucketing_key,
63+
'pt': impression.previous_time,
64+
'properties': impression.properties
65+
}
66+
5267
@staticmethod
5368
def _build_counters(counters):
5469
"""

tests/api/test_impressions_api.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,12 +22,12 @@
2222
'f': 'f1',
2323
'i': [
2424
{'k': 'k1', 'b': 'b1', 't': 'on', 'r': 'l1', 'm': 321654, 'c': 123456, 'pt': None, 'properties': {"prop": "val"}},
25-
{'k': 'k3', 'b': 'b1', 't': 'on', 'r': 'l1', 'm': 321654, 'c': 123456, 'pt': None, 'properties': None},
25+
{'k': 'k3', 'b': 'b1', 't': 'on', 'r': 'l1', 'm': 321654, 'c': 123456, 'pt': None},
2626
],
2727
}, {
2828
'f': 'f2',
2929
'i': [
30-
{'k': 'k2', 'b': 'b1', 't': 'off', 'r': 'l1', 'm': 321654, 'c': 123456, 'pt': None, 'properties': None},
30+
{'k': 'k2', 'b': 'b1', 't': 'off', 'r': 'l1', 'm': 321654, 'c': 123456, 'pt': None},
3131
]
3232
}]
3333

0 commit comments

Comments
 (0)