From 687f01d74ab1712debbe0e7ecc17fcbcd8a1d601 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Luk=C3=A1=C5=A1=20Lalinsk=C3=BD?= <lalinsky@gmail.com>
Date: Mon, 23 Oct 2006 15:47:57 +0200
Subject: [PATCH] Added functions $rsearch and $rreplace.

---
 picard/tagz.py    | 21 ++++++++++++++++++++-
 test/test_tagz.py | 23 +++++++++++++++++++++++
 2 files changed, 43 insertions(+), 1 deletion(-)

diff --git a/picard/tagz.py b/picard/tagz.py
index eb6983538..17087a1a4 100644
--- a/picard/tagz.py
+++ b/picard/tagz.py
@@ -33,7 +33,7 @@ _re_text2 = r"(?:\\.|[^%$,])+"
 _re_args_sep = ","
 _re_name = "\w+"
 _re_var = r"%" + _re_name + r"%"
-_re_func_args = no_parens = r"[^()]"
+_re_func_args = no_parens = r"(?:\(|\)|[^()])"
 for i in range(10): # 10 levels must be enough for everybody ;)
    _re_func_args = r"(\(" + _re_func_args + r"\)|" + no_parens + r")*"
 _re_func = r"\$" + _re_name + r"\(" + _re_func_args + r"\)"
@@ -84,6 +84,16 @@ def func_strip(context, text):
 def func_replace(context, text, old, new):
     return text.replace(old, new)
 
+def func_rreplace(context, text, old, new):
+    return re.sub(old, new, text)
+
+def func_rsearch(context, text, pattern):
+    print pattern
+    match = re.search(pattern, text)
+    if match:
+        return match.group(1)
+    return u""
+
 def func_num(context, text, length):
     format = "%%0%dd" % int(length)
     return format % int(text)
@@ -226,6 +236,8 @@ class TagzBuiltins(Component):
         "pad": func_pad,
         "strip": func_strip,
         "replace": func_replace,
+        "rreplace": func_rreplace,
+        "rsearch": func_rsearch,
         "num": func_num,
 
         "set": func_set,
@@ -264,6 +276,9 @@ class TagzParser(object):
         return "".join(res[0])
 
     def s_text(self, scanner, string):
+        string = string.replace("\\(", "(")
+        string = string.replace("\\)", ")")
+        string = string.replace("\\\\", "\\")
         return string
 
     def s_variable(self, scanner, string):
@@ -289,6 +304,8 @@ class TagzParser(object):
             raise TagzParseError(string.rfind(error))
 
         args = []
+        if name == "replace":
+            print repr(results)
         if results:
             if results[0] == "\0":
                 results.insert(0, "")
@@ -304,6 +321,8 @@ class TagzParser(object):
             results = results[j:]
 
         try:
+            if name == "replace":
+                print args
             return self.functions[name](self.context, *args)
         except KeyError:
             raise TagzUnknownFunction, "Unknown function $%s" % name
diff --git a/test/test_tagz.py b/test/test_tagz.py
index 9753ed6a9..53f26cd79 100644
--- a/test/test_tagz.py
+++ b/test/test_tagz.py
@@ -90,3 +90,26 @@ class MiscModelTest(unittest.TestCase):
     def test_cmd_upper(self):
         self.failUnlessEqual(self.tagz.evaluate_script("$upper(AbeCeDA)"), "ABECEDA")
 
+    def test_cmd_rreplace(self):
+        self.failUnlessEqual(
+            self.tagz.evaluate_script(r"$rreplace(test \(disc 1\),\\s\\(disc \d+\\),)"),
+            "test"
+            )
+
+    def test_cmd_rsearch(self):
+        self.failUnlessEqual(
+            self.tagz.evaluate_script(r"$rsearch(test \(disc 1\),\\(disc \(\d+\)\\))"),
+            "1"
+            )
+
+
+class TagzParserTest(unittest.TestCase):
+
+    def setUp(self):
+        self.tagz = Tagz(ComponentManager())
+
+    def test_arguments(self):
+        self.failUnlessEqual(
+            self.tagz.evaluate_script(r"$rreplace(test \(disc 1\),\\s\\(disc \d+\\),)"),
+            "test"
+            )