manifest.read_template now accepts file objects

2 files changed, 19 insertions, 3 deletions

diff --git a/distutils2/manifest.py b/distutils2/manifest.py
index 27b3dce..6d4694a 100644
--- a/distutils2/manifest.py
+++ b/distutils2/manifest.py
@@ -66,12 +66,17 @@ class Manifest(object):
             if self.files[i] == self.files[i - 1]:
                 del self.files[i]
 
-    def read_template(self, path):
+    def read_template(self, path_or_file):
         """Read and parse a manifest template file.
+        'path' can be a path or a file-like object.
 
         Updates the list accordingly.
         """
-        f = open(path)
+        if isinstance(path_or_file, str):
+            f = open(path_or_file)
+        else:
+            f = path_or_file
+
         try:
             content = f.read()
             # first, let's unwrap collapsed lines
@@ -89,7 +94,7 @@ class Manifest(object):
             try:
                 self._process_template_line(line)
             except DistutilsTemplateError, msg:
-                logging.warning("%s, %s" % (path, msg))
+                logging.warning("%s, %s" % (path_or_file, msg))
 
     def write(self, path):
         """Write the file list in 'self.filelist' (presumably as filled in
diff --git a/distutils2/tests/test_manifest.py b/distutils2/tests/test_manifest.py
index bb8c96c..dcf570f 100644
--- a/distutils2/tests/test_manifest.py
+++ b/distutils2/tests/test_manifest.py
@@ -48,6 +48,17 @@ class ManifestTestCase(support.TempdirManager,
         for warn in warns:
             self.assertIn('warning: no files found matching', warn)
 
+        # manifest also accepts file-like objects
+        old_warn = logging.warning
+        logging.warning = _warn
+        try:
+            manifest.read_template(open(MANIFEST))
+        finally:
+            logging.warning = old_warn
+
+        # the manifest should have been read
+        # and 3 warnings issued (we ddidn't provided the files)
+        self.assertEqual(len(warns), 6)
 
 
 def test_suite():