Compile regex

sebastinas · sebastinas · commit 1e8c8931d570 · 2015-02-02T22:26:38.000+01:00
Signed-off-by: Sebastian Ramacher &lt;sebastian+dev@ramacher.at&gt;
diff --git a/bpython/curtsiesfrontend/parse.py b/bpython/curtsiesfrontend/parse.py
@@ -1,4 +1,5 @@
 
+from bpython.line import LazyReCompile
 from bpython.formatter import BPythonFormatter
 from bpython._py3compat import PythonLexer
 from bpython.config import Struct, loadini, default_config_path
@@ -62,17 +63,21 @@ def fs_from_match(d):
         atts['bold'] = True
     return fmtstr(d['string'], **atts)
 
+
+peel_off_string_re = LazyReCompile(
+    r"""(?P<colormarker>\x01
+            (?P<fg>[krgybmcwdKRGYBMCWD]?)
+            (?P<bg>[krgybmcwdKRGYBMCWDI]?)?)
+        (?P<bold>\x02?)
+        \x03
+        (?P<string>[^\x04]*)
+        \x04
+        (?P<rest>.*)
+        """, re.VERBOSE | re.DOTALL)
+
+
 def peel_off_string(s):
-    p = r"""(?P<colormarker>\x01
-                (?P<fg>[krgybmcwdKRGYBMCWD]?)
-                (?P<bg>[krgybmcwdKRGYBMCWDI]?)?)
-            (?P<bold>\x02?)
-            \x03
-            (?P<string>[^\x04]*)
-            \x04
-            (?P<rest>.*)
-            """
-    m = re.match(p, s, re.VERBOSE | re.DOTALL)
+    m = peel_off_string_re.match(s)
     assert m, repr(s)
     d = m.groupdict()
     rest = d['rest']
diff --git a/bpython/line.py b/bpython/line.py
@@ -10,14 +10,15 @@
 
 class LazyReCompile(object):
 
-    def __init__(self, regex):
+    def __init__(self, regex, flags=0):
         self.regex = regex
+        self.flags = flags
         self.compiled = None
 
     def compile_regex(method):
         def _impl(self, *args, **kwargs):
             if self.compiled is None:
-                self.compiled = re.compile(self.regex)
+                self.compiled = re.compile(self.regex, self.flags)
             return method(self, *args, **kwargs)
         return _impl
 
