Transformation de encoding.open_read en un 'context manager'

patacrep/content/include.py

@@ -48,15 +48,12 @@ def parse(keyword, config, argument, contentlist):
         filepath = load_from_datadirs(path, config)
         content_file = None
         try:
-            content_file = encoding.open_read(filepath, 'r')
+            with encoding.open_read(filepath, 'r') as content_file:
-            new_content = json.load(content_file)
+                new_content = json.load(content_file)
         except Exception as error: # pylint: disable=broad-except
             LOGGER.error(error)
             LOGGER.error("Error while loading file '{}'.".format(filepath))
             sys.exit(1)
-        finally:
-            if content_file:
-                content_file.close()
 
         config["datadir"].append(os.path.abspath(os.path.dirname(filepath)))
         new_contentlist += process_content(new_content, config)

patacrep/encoding.py

@@ -5,17 +5,21 @@
 import codecs
 import chardet
 import logging
+import contextlib
 
 LOGGER = logging.getLogger(__name__)
 
+
+@contextlib.contextmanager
 def open_read(filename, mode='r'):
     """Open a file for reading, guessing the right encoding.
 
     Return a fileobject, reading unicode strings.
     """
-    return codecs.open(
+    with codecs.open(
             filename,
             mode=mode,
             encoding=chardet.detect(open(filename, 'rb').read())['encoding'],
             errors='replace',
-            )
+            ) as fileobject:
+        yield fileobject

patacrep/index.py

@@ -29,13 +29,9 @@ def process_sxd(filename):
     """
     data = []
     index_file = None
-    try:
+    with encoding.open_read(filename, 'r') as index_file:
-        index_file = encoding.open_read(filename, 'r')
         for line in index_file:
             data.append(line.strip())
-    finally:
-        if index_file:
-            index_file.close()
 
     i = 1
     idx = Index(data[0])

patacrep/latex/__init__.py

@@ -15,6 +15,7 @@ def parsesong(path):
     """Return a dictonary of data read from the latex file `path`.
 
     """
-    data = syntax_parsesong(encoding.open_read(path).read(), path)
+    with encoding.open_read(path) as songfile:
+        data = syntax_parsesong(songfile.read(), path)
     data['@path'] = path
     return data

patacrep/songbook.py

@@ -100,17 +100,13 @@ def main():
 
     basename = os.path.basename(songbook_path)[:-3]
 
-    songbook_file = None
     try:
-        songbook_file = encoding.open_read(songbook_path)
+        with encoding.open_read(songbook_path) as songbook_file:
-        songbook = json.load(songbook_file)
+            songbook = json.load(songbook_file)
     except Exception as error: # pylint: disable=broad-except
         LOGGER.error(error)
         LOGGER.error("Error while loading file '{}'.".format(songbook_path))
         sys.exit(1)
-    finally:
-        if songbook_file:
-            songbook_file.close()
 
     # Gathering datadirs
     datadirs = []

patacrep/templates.py

@@ -186,32 +186,27 @@ class TexRenderer(object):
         """
 
         subvariables = {}
-        template_file = None
         templatename = self.texenv.get_template(template).filename
-        try:
+        with encoding.open_read(templatename, 'r') as template_file:
-            template_file = encoding.open_read(templatename, 'r')
             content = template_file.read()
-            subtemplates = list(find_templates(self.texenv.parse(content)))
+        subtemplates = list(find_templates(self.texenv.parse(content)))
-            match = re.findall(_VARIABLE_REGEXP, content)
+        match = re.findall(_VARIABLE_REGEXP, content)
-            if match:
+        if match:
-                for var in match:
+            for var in match:
-                    try:
+                try:
-                        subvariables.update(json.loads(var))
+                    subvariables.update(json.loads(var))
-                    except ValueError as exception:
+                except ValueError as exception:
-                        raise errors.TemplateError(
+                    raise errors.TemplateError(
-                                exception,
+                            exception,
-                                (
+                            (
-                                    "Error while parsing json in file "
+                                "Error while parsing json in file "
-                                    "{filename}. The json string was:"
+                                "{filename}. The json string was:"
-                                    "\n'''\n{jsonstring}\n'''"
+                                "\n'''\n{jsonstring}\n'''"
-                                ).format(
+                            ).format(
-                                    filename=templatename,
+                                filename=templatename,
-                                    jsonstring=var,
+                                jsonstring=var,
-                                    )
                                 )
-        finally:
+                            )
-            if template_file:
-                template_file.close()
 
         return (subvariables, subtemplates)