# HG changeset patch
# User devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
# Date 1198089639 18000
# Node ID 8b55aaa52ce95fd667ed89a429f8b8f9b3e9f19e
# Parent  d6d64c2e3b987cf3e53ee2b6c9fcc75d784f4d9b
working on load, and preserving stdin/out

diff -r d6d64c2e3b98 -r 8b55aaa52ce9 cmd2.py
--- a/cmd2.py	Wed Dec 19 11:49:23 2007 -0500
+++ b/cmd2.py	Wed Dec 19 13:40:39 2007 -0500
@@ -4,15 +4,21 @@
 Multi-line commands
 Case-insensitive commands
 Special-character shortcut commands
+
+still to do:
+environment (maxrows, etc.)
+edit
+
 """
 import cmd, re, os
 
 class Cmd(cmd.Cmd):
-    excludeFromHistory = '''run r list l history hi ed li'''.split()
     caseInsensitive = True
     multilineCommands = []
     continuationPrompt = '> '    
-    shortcuts = {'?': 'help', '!': 'shell'}    
+    shortcuts = {'?': 'help', '!': 'shell', '@': 'load'}   
+    excludeFromHistory = '''run r list l history hi ed li'''.split()   
+    defaultExtension = 'txt'
     def __init__(self, *args, **kwargs):	
         cmd.Cmd.__init__(self, *args, **kwargs)
         self.history = History()
@@ -46,12 +52,30 @@
     def finishStatement(self, firstline):
 	statement = firstline
 	while not self.statementHasEnded(statement):
-	    statement = '%s\n%s' % (statement, raw_input(self.continuationPrompt))
+	    statement = '%s\n%s' % (statement, self.pseudo_raw_input(self.continuationPrompt))
 	return statement
 	# assembling a list of lines and joining them at the end would be faster, 
 	# but statementHasEnded needs a string arg; anyway, we're getting
 	# user input and users are slow.
 	
+    def pseudo_raw_input(self, prompt):
+	"""copied from cmd's cmdloop; like raw_input, but accounts for changed stdin, stdout"""
+	
+	if self.use_rawinput:
+	    try:
+		line = raw_input(prompt)
+	    except EOFError:
+		line = 'EOF'
+	else:
+	    self.stdout.write(prompt)
+	    self.stdout.flush()
+	    line = self.stdin.readline()
+	    if not len(line):
+		line = 'EOF'
+	    else:
+		line = line[:-1] # chop \n
+	return line
+			    
     statementEndPattern = re.compile(r'[\n;]\s*$')	
     def statementHasEnded(self, lines):
 	"""This version lets statements end with ; or with a blank line.
@@ -91,7 +115,7 @@
         else:
             history = self.history
         for hi in history:
-            hi.pr()
+            self.stdout.write(hi.pr())
     def last_matching(self, arg):
         try:
             if arg:
@@ -110,12 +134,41 @@
         arg is /enclosed in forward-slashes/ -> regular expression search
         """
         try:
-            self.last_matching(arg).pr()
+            self.stdout.write(self.last_matching(arg).pr())
         except:
             pass
     do_hi = do_history
     do_l = do_list
     do_li = do_list
+    
+    def breakupStatements(self, txt):
+	"""takes text that may include multiple statements and 
+	breaks it into a list of individual statements."""
+	result = ['']
+	for line in txt.splitlines():
+	    result[-1] += line
+	    if self.statementHasEnded(result[-1]):
+		result.append('')
+	
+    def do_load(self, fname):
+        """Runs command(s) from a file."""
+	stdin = self.stdin
+	try:
+	    self.stdin = open(fname, 'r')
+        except IOError, e:
+            try:
+                self.stdin = open('%s.%s' % (fname, self.defaultExtension), 'r')
+            except:
+                print 'Problem opening file %s: \n%s' % (fname, e)
+		self.stdin = stdin
+                return 	    
+	use_rawinput = self.use_rawinput
+	self.use_rawinput = False
+	print 'stdin = ' + str(self.stdin)
+	self.cmdloop()
+	self.stdin.close()
+	self.stdin = stdin
+	self.use_rawinput = use_rawinput
 
 class HistoryItem(str):
     def __init__(self, instr):
@@ -123,8 +176,7 @@
         self.lowercase = self.lower()
         self.idx = None
     def pr(self):
-        print '-------------------------[%d]' % (self.idx)
-        print self
+	return '-------------------------[%d]\n%s\n' % (self.idx, str(self))
         
 class History(list):
     rangeFrom = re.compile(r'^([\d])+\s*\-$')
diff -r d6d64c2e3b98 -r 8b55aaa52ce9 sqlpyPlus.py
--- a/sqlpyPlus.py	Wed Dec 19 11:49:23 2007 -0500
+++ b/sqlpyPlus.py	Wed Dec 19 13:40:39 2007 -0500
@@ -342,6 +342,9 @@
     return result
         
 class sqlpyPlus(sqlpython.sqlpython):
+    defaultExtension = 'sql'
+    multilineCommands = '''select insert update delete tselect
+        create drop alter'''.split()
     def __init__(self):
         sqlpython.sqlpython.__init__(self)
         self.binds = CaselessDict()
@@ -351,9 +354,6 @@
         self.spoolFile = None
         self.autobind = False
         self.failover = False
-        self.multiline = '''select insert update delete tselect
-        create drop alter'''.split()
-
     def default(self, arg, do_everywhere=False):
         sqlpython.sqlpython.default(self, arg, do_everywhere)
         self.sqlBuffer.append(self.query)            
@@ -749,21 +749,6 @@
         self.onecmd_plus_hooks(runme)
     do_r = do_run
 
-    def load(self, fname):
-        """Pulls command(s) into sql buffer.  Returns number of commands loaded."""
-        try:
-            f = open(fname, 'r')
-        except IOError, e:
-            try:
-                f = open('%s.sql' % fname, 'r')
-            except:
-                print 'Problem opening file %s: \n%s' % (fname, e)
-                return 0
-        txt = f.read()
-        f.close()
-        result = commandSeparator.separate(txt)
-        self.history.extend(result) 
-        return len(result)
     def do_ed(self, arg):
         'ed [N]: brings up SQL from N commands ago in text editor, and puts result in SQL buffer.'
         fname = 'sqlpython_temp.sql'