Updated Caper to v0.3.1

libs/caper/step.py

@@ -12,13 +12,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from caper.objects import CaptureMatch
 from logr import Logr
 
 
 class CaptureStep(object):
     REPR_KEYS = ['regex', 'func', 'single']
 
-    def __init__(self, capture_group, tag, source, regex=None, func=None, single=None):
+    def __init__(self, capture_group, tag, source, regex=None, func=None, single=None, **kwargs):
         #: @type: CaptureGroup
         self.capture_group = capture_group
 
@@ -33,22 +34,57 @@ class CaptureStep(object):
         #: @type: bool
         self.single = single
 
+        self.kwargs = kwargs
+
+        self.matched = False
+
     def execute(self, fragment):
+        """Execute step on fragment
+
+        :type fragment: CaperFragment
+        :rtype : CaptureMatch
+        """
+
+        match = CaptureMatch(self.tag, self)
+
         if self.regex:
-            weight, match, num_fragments = self.capture_group.parser.matcher.fragment_match(fragment, self.regex)
+            weight, result, num_fragments = self.capture_group.parser.matcher.fragment_match(fragment, self.regex)
             Logr.debug('(execute) [regex] tag: "%s"', self.tag)
-            if match:
-                return True, weight, match, num_fragments
+
+            if not result:
+                return match
+
+            # Populate CaptureMatch
+            match.success = True
+            match.weight = weight
+            match.result = result
+            match.num_fragments = num_fragments
         elif self.func:
-            match = self.func(fragment)
+            result = self.func(fragment)
             Logr.debug('(execute) [func] %s += "%s"', self.tag, match)
-            if match:
-                return True, 1.0, match, 1
+
+            if not result:
+                return match
+
+            # Populate CaptureMatch
+            match.success = True
+            match.weight = 1.0
+            match.result = result
         else:
             Logr.debug('(execute) [raw] %s += "%s"', self.tag, fragment.value)
-            return True, 1.0, fragment.value, 1
 
-        return False, None, None, 1
+            include_separators = self.kwargs.get('include_separators', False)
+
+            # Populate CaptureMatch
+            match.success = True
+            match.weight = 1.0
+
+            if include_separators:
+                match.result = (fragment.left_sep, fragment.value, fragment.right_sep)
+            else:
+                match.result = fragment.value
+
+        return match
 
     def __repr__(self):
         attribute_values = [key + '=' + repr(getattr(self, key))