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)
         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)
     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,10 +186,8 @@ 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)))
         match = re.findall(_VARIABLE_REGEXP, content)
@@ -209,9 +207,6 @@ class TexRenderer(object):
                                 jsonstring=var,
                                 )
                             )
-        finally:
-            if template_file:
-                template_file.close()
 
         return (subvariables, subtemplates)