from tasklib.serializing import SerializingObject class TaskFilter(SerializingObject): """ A set of parameters to filter the task list with. """ def __init__(self, warrior, filter_params=None): self.filter_params = filter_params or [] super(TaskFilter, self).__init__(warrior) def add_filter(self, filter_str): self.filter_params.append(filter_str) def add_filter_param(self, key, value): key = key.replace('__', '.') # Replace the value with empty string, since that is the # convention in TW for empty values attribute_key = key.split('.')[0] # Since this is user input, we need to normalize before we serialize value = self._normalize(attribute_key, value) value = self._serialize(attribute_key, value) # If we are filtering by uuid:, do not use uuid keyword # due to TW-1452 bug if key == 'uuid': self.filter_params.insert(0, value) else: # Surround value with aphostrophes unless it's a empty string value = "'%s'" % value if value else '' # We enforce equality match by using 'is' (or 'none') modifier # Without using this syntax, filter fails due to TW-1479 # which is, however, fixed in 2.4.5 if self.warrior.version < VERSION_2_4_5: modifier = '.is' if value else '.none' key = key + modifier if '.' not in key else key self.filter_params.append(six.u("{0}:{1}").format(key, value)) def get_filter_params(self): return [f for f in self.filter_params if f] def clone(self): c = self.__class__(self.warrior) c.filter_params = list(self.filter_params) return c