diff cmd2.py @ 41:1275d59c08a7

do_load accepts file object
author catherine@cordelia
date Thu, 22 May 2008 20:07:07 -0400
parents 71151157ac7c
children b3200c6e5763
line wrap: on
line diff
--- a/cmd2.py	Tue May 20 21:50:00 2008 -0400
+++ b/cmd2.py	Thu May 22 20:07:07 2008 -0400
@@ -487,15 +487,18 @@
         if fname is None:
             fname = self.defaultFileName        
         keepstate = Statekeeper(self, ('stdin','use_rawinput','prompt','continuationPrompt'))
-        try:
-            self.stdin = open(fname, 'r')
-        except IOError, e:
+        if isinstance(fname, file):
+            self.stdin = fname
+        else:           
             try:
-                self.stdin = open('%s.%s' % (fname, self.defaultExtension), 'r')
-            except IOError:
-                print 'Problem opening file %s: \n%s' % (fname, e)
-                keepstate.restore()
-                return
+                self.stdin = open(fname, 'r')
+            except IOError, e:
+                try:
+                    self.stdin = open('%s.%s' % (fname, self.defaultExtension), 'r')
+                except IOError:
+                    print 'Problem opening file %s: \n%s' % (fname, e)
+                    keepstate.restore()
+                    return
         self.use_rawinput = False
         self.prompt = self.continuationPrompt = ''
         self.cmdloop()