Fix and refactor caching functions

patacrep/songs/__init__.py

@@ -96,13 +96,21 @@ class Song:
         ]
 
     def __init__(self, subpath, config, *, datadir=None):
-        self.datadir = datadir or ""
+        if datadir is None:
+            self.datadir = ""
+            self.use_cache = False
+        else:
+            self.datadir = datadir
+            # Only songs in datadirs are cached
+            self.use_cache = ('_cache' in config)
+
         self.fullpath = os.path.join(self.datadir, subpath)
+        self.subpath = subpath
         self.encoding = config["encoding"]
         self.default_lang = config["lang"]
         self.config = config
 
-        if self.config['_cache'] and self._cache_retrieved():
+        if self._cache_retrieved():
             return
 
         # Data extraction from the latex song
@@ -113,7 +121,6 @@ class Song:
         self._parse()
 
         # Post processing of data
-        self.subpath = subpath
         self.unprefixed_titles = [
             unprefixed_title(
                 title,
@@ -129,20 +136,18 @@ class Song:
 
         # Cache management
         self._version = self.CACHE_VERSION
-        if self.config['_cache']:
+        self._write_cache()
-            self._write_cache()
 
     def _cache_retrieved(self):
         """If relevant, retrieve self from the cache."""
-        if self.datadir:
+        if self.use_cache:
-            # Only songs in datadirs are cached
             self._filehash = hashlib.md5(
                 open(self.fullpath, 'rb').read()
                 ).hexdigest()
-            if os.path.exists(cached_name(datadir, subpath)):
+            if os.path.exists(cached_name(self.datadir, self.subpath)):
                 try:
                     cached = pickle.load(open(
-                        cached_name(datadir, subpath),
+                        cached_name(self.datadir, self.subpath),
                         'rb',
                         ))
                     if (
@@ -160,7 +165,7 @@ class Song:
 
     def _write_cache(self):
         """If relevant, write a dumbed down version of self to the cache."""
-        if self.datadir:
+        if self.use_cache:
             # Only songs in datadirs can be cached
             cached = {}
             for attribute in self.cached_attributes: