ENH TextGrid.read(): accept file-like object

textgrid/textgrid.py

@@ -31,6 +31,7 @@
 
 import re
 import codecs
+import contextlib
 import os.path
 import logging
 
@@ -101,6 +102,26 @@ def detectEncoding(f):
     return encoding
 
 
+@contextlib.contextmanager
+def as_file(obj, encoding):
+    """
+    Accept file-object or path-like object for reading TextGrid
+    """
+    if hasattr(obj, 'readline'):
+        do_close = False
+        file_obj = obj
+    else:
+        do_close = True
+        if encoding is None:
+            encoding = detectEncoding(obj)
+        file_obj = codecs.open(obj, 'r', encoding=encoding)
+    try:
+        yield file_obj
+    finally:
+        if do_close:
+            file_obj.close()
+
+
 class Point(object):
     """
     Represents a point in time with an associated textual mark, as stored
@@ -692,11 +713,10 @@ def pop(self, i=None):
     def read(self, f, round_digits=DEFAULT_TEXTGRID_PRECISION, encoding=None):
         """
         Read the tiers contained in the Praat-formatted TextGrid file
-        indicated by string f. Times are rounded to the specified precision.
+        indicated by path or file-like object f.
+        Times are rounded to the specified precision.
         """
-        if encoding is None:
-            encoding = detectEncoding(f)
-        with codecs.open(f, 'r', encoding=encoding) as source:
+        with as_file(f, encoding) as source:
             file_type, short = parse_header(source)
             if file_type != 'TextGrid':
                 raise TextGridError('The file could not be parsed as a TextGrid as it is lacking a proper header.')