Fix formatting, and add python2 support.

beetbox · May 9, 2019 · 7df4e23 · 7df4e23
1 parent c5075b2
commit 7df4e23
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/beets/util/functemplate.py b/beets/util/functemplate.py
@@ -553,12 +553,21 @@ def _parse(template):
         parts.append(remainder)
     return Expression(parts)
 
-@functools.lru_cache(maxsize=128)
+
+# Decorator that enables lru_cache on py3, and no caching on py2.
+def cached(func):
+    if six.PY2:
+        # Sorry python2 users, no caching for you :(
+        return func
+    return functools.lru_cache(maxsize=128)(func)
+
+
+@cached
 def template(fmt):
     return Template(fmt)
 
-# External interface.
 
+# External interface.
 class Template(object):
     """A string template, including text, Symbols, and Calls.
     """