Skip to content

Commit 2b619e1

Browse files
committed
Address code review comments
1 parent 7dc9881 commit 2b619e1

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

arctic/chunkstore/date_chunker.py

+8-8
Original file line numberDiff line numberDiff line change
@@ -74,17 +74,17 @@ def to_start_end(self, data):
7474
return self._get_date_range(data)
7575

7676
def to_mongo(self, range_obj):
77-
if range_obj[0] and range_obj[1]:
78-
return {'$and': [{'start': {'$lte': range_obj[1]}}, {'end': {'$gte': range_obj[0]}}]}
79-
elif range_obj[0]:
80-
return {'end': {'$gte': range_obj[0]}}
81-
elif range_obj[1]:
82-
return {'start': {'$lte': range_obj[1]}}
77+
if range_obj.start and range_obj.end:
78+
return {'$and': [{'start': {'$lte': range_obj.end}}, {'end': {'$gte': range_obj.start}}]}
79+
elif range_obj.start:
80+
return {'end': {'$gte': range_obj.start}}
81+
elif range_obj.end:
82+
return {'start': {'$lte': range_obj.end}}
8383
else:
8484
return {}
8585

8686
def filter(self, data, range_obj):
87-
return data.ix[range_obj[0]:range_obj[1]]
87+
return data.ix[range_obj.start:range_obj.end]
8888

8989
def exclude(self, data, range_obj):
90-
return data[(data.index.get_level_values('date') < range_obj[0]) | (data.index.get_level_values('date') > range_obj[1])]
90+
return data[(data.index.get_level_values('date') < range_obj.start) | (data.index.get_level_values('date') > range_obj.end)]

0 commit comments

Comments
 (0)