feat(processors): re_groups

wind2sing · Oct 25, 2019 · 157e481 · 157e481
1 parent 9c1fa22
commit 157e481
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/acrawler/processors.py b/acrawler/processors.py
@@ -93,6 +93,26 @@ def _f(value):
 
         return _f
 
+    @staticmethod
+    def re_groups(regex, default=None):
+        def _f(value):
+            match = re.search(regex, value)
+            if match:
+                return match.groups(default)
+            return None
+
+        return _f
+
+    @staticmethod
+    def re_groupdict(regex, default=None):
+        def _f(value):
+            match = re.search(regex, value)
+            if match:
+                return match.groupdict(default)
+            return None
+
+        return _f
+
     @staticmethod
     def default(default, fn=bool):
         def _f(value):