File tree 1 file changed +8
-8
lines changed
1 file changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -74,17 +74,17 @@ def to_start_end(self, data):
74
74
return self ._get_date_range (data )
75
75
76
76
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 }}
83
83
else :
84
84
return {}
85
85
86
86
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 ]
88
88
89
89
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 )]
You can’t perform that action at this time.
0 commit comments