Changes to allow passing an open file to the constructor (to support

gward · gward · commit 291ecfe2fd48 · 1999-03-23T14:00:06.000Z
ProcessHierarchy's changes to support reading from a remote URL in ProcessDatabase). git-svn-id: http://svn.python.org/projects/python/trunk@12950 6015fed2-1504-0410-9fe1-9d1591cc4771
diff --git a/Lib/distutils/text_file.py b/Lib/distutils/text_file.py
@@ -13,9 +13,6 @@
 
 
 class TextFile:
-    filename = None
-    file = None
-    current_line = None
 
     default_options = { 'strip_comments': 1,
                         'comment_re':     re.compile (r'\s*#.*'),
@@ -26,7 +23,11 @@ class TextFile:
                         'collapse_ws':    0,
                       }
 
-    def __init__ (self, filename=None, **options):
+    def __init__ (self, filename=None, file=None, **options):
+
+        if filename is None and file is None:
+            raise RuntimeError, \
+                  "you must supply either or both of 'filename' and 'file'" 
 
         # set values for all options -- either from client option hash
         # or fallback to default_options
@@ -45,18 +46,16 @@ def __init__ (self, filename=None, **options):
             if not self.default_options.has_key (opt):
                 raise KeyError, "invalid TextFile option '%s'" % opt
 
-        self.filename = filename
-        if self.filename:
-            self.open ()
-        
-
-    def open (self, filename=None):
-        if not self.filename:
-            if not filename:
-                raise RuntimeError, "must provide a filename somehow"
-
+        if file is None:
+            self.open (filename)
+        else:
             self.filename = filename
+            self.file = file
+            self.current_line = 0       # assuming that file is at BOF!
+        
 
+    def open (self, filename):
+        self.filename = filename
         self.file = open (self.filename, 'r')
         self.current_line = 0
 
