X-Git-Url: https://git.madduck.net/etc/taskwarrior.git/blobdiff_plain/31d9529fe14bb21ff87c6e0030890fb32df15834..8078ceed9b50dad2d319a74e2d3014b6ff649b47:/code/taskwarrior/tasklib/tasklib/filters.py

diff --git a/code/taskwarrior/tasklib/tasklib/filters.py b/code/taskwarrior/tasklib/tasklib/filters.py
new file mode 100644
index 0000000..b7cab64
--- /dev/null
+++ b/code/taskwarrior/tasklib/tasklib/filters.py
@@ -0,0 +1,79 @@
+import abc
+import six
+from .serializing import SerializingObject
+
+
+class TaskFilter(object):
+    """
+    Abstract base class that defines interface of a TaskFilter.
+    """
+
+    @abc.abstractmethod
+    def add_filter(self, arg):
+        """
+        Processes an non-keyword filter.
+        """
+        pass
+
+    @abc.abstractmethod
+    def add_filter_param(self, key, value):
+        """
+        Processes a keyword filter.
+        """
+        pass
+
+    @abc.abstractmethod
+    def clone(self):
+        """
+        Returns a new deep copy of itself.
+        """
+        pass
+
+
+class TaskWarriorFilter(TaskFilter, SerializingObject):
+    """
+    A set of parameters to filter the task list with.
+    """
+
+    def __init__(self, backend, filter_params=None):
+        self.filter_params = filter_params or []
+        super(TaskFilter, self).__init__(backend)
+
+    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.backend.version < self.backend.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.backend)
+        c.filter_params = list(self.filter_params)
+        return c