Mercurial > sqlpython
annotate sqlpyPlus.py @ 4:23c3a58d7804
about to strip out tselect
| author | devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil |
|---|---|
| date | Tue, 04 Dec 2007 16:28:55 -0500 |
| parents | cd23cd62de3c |
| children | 65ae6cec71c6 |
| rev | line source |
|---|---|
| 0 | 1 """sqlpyPlus - extra features (inspired by Oracle SQL*Plus) for Luca Canali's sqlpython.py |
| 2 | |
| 3 Features include: | |
| 4 - SQL*Plus-style bind variables | |
| 5 - Query result stored in special bind variable ":_" if one row, one item | |
| 6 - SQL buffer with list, run, ed, get, etc.; unlike SQL*Plus, buffer stores session's full history | |
| 7 - @script.sql loads and runs (like SQL*Plus) | |
| 8 - ! runs operating-system command | |
| 9 - show and set to control sqlpython parameters | |
| 10 - SQL*Plus-style describe, spool | |
| 11 - write sends query result directly to file | |
| 12 - comments shows table and column comments | |
| 13 - compare ... to ... graphically compares results of two queries | |
| 14 - commands are case-insensitive | |
| 15 | |
| 16 Use 'help' within sqlpython for details. | |
| 17 | |
| 18 Compatible with sqlpython v1.3 | |
| 19 | |
| 20 Set bind variables the hard (SQL*Plus) way | |
| 21 exec :b = 3 | |
| 22 or with a python-like shorthand | |
| 23 :b = 3 | |
| 24 | |
| 25 - catherinedevlin.blogspot.com May 31, 2006 | |
| 26 """ | |
| 27 # note in cmd.cmd about supporting emacs commands? | |
| 28 | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
29 descQueries = { |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
30 'TABLE': """ |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
31 atc.column_name, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
32 CASE atc.nullable WHEN 'Y' THEN 'NULL' ELSE 'NOT NULL' END "Null?", |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
33 atc.data_type || |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
34 CASE atc.data_type WHEN 'DATE' THEN '' |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
35 ELSE '(' || |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
36 CASE atc.data_type WHEN 'NUMBER' THEN TO_CHAR(atc.data_precision) || |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
37 CASE atc.data_scale WHEN 0 THEN '' |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
38 ELSE ',' || TO_CHAR(atc.data_scale) END |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
39 ELSE TO_CHAR(atc.data_length) END |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
40 END || |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
41 CASE atc.data_type WHEN 'DATE' THEN '' ELSE ')' END |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
42 data_type |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
43 FROM all_tab_columns atc |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
44 WHERE atc.table_name = :object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
45 AND atc.owner = :owner |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
46 ORDER BY atc.column_id;""", |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
47 'PROCEDURE':""" |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
48 argument_name, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
49 data_type, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
50 in_out, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
51 default_value |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
52 FROM all_arguments |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
53 WHERE object_name = :object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
54 AND owner = :owner |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
55 AND package_name IS NULL |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
56 AND argument_name IS NOT NULL |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
57 ORDER BY sequence;""", |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
58 'PackageObjects':""" |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
59 SELECT DISTINCT object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
60 FROM all_arguments |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
61 WHERE package_name = :package_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
62 AND owner = :owner""", |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
63 'PackageObjArgs':""" |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
64 object_name, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
65 argument_name, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
66 data_type, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
67 in_out, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
68 default_value |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
69 FROM all_arguments |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
70 WHERE package_name = :package_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
71 AND object_name = :object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
72 AND owner = :owner |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
73 AND argument_name IS NOT NULL |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
74 ORDER BY sequence""", |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
75 'TRIGGER':""" |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
76 trigger_name, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
77 trigger_type, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
78 triggering_event, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
79 table_owner, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
80 base_object_type, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
81 table_name, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
82 column_name, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
83 when_clause, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
84 status, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
85 description, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
86 action_type, |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
87 crossedition |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
88 FROM all_triggers |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
89 WHERE owner = :owner |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
90 AND trigger_name = :object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
91 \\t |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
92 """ |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
93 } |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
94 descQueries['VIEW'] = descQueries['TABLE'] |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
95 descQueries['FUNCTION'] = descQueries['PROCEDURE'] |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
96 |
| 0 | 97 queries = { |
| 98 'resolve': """ | |
| 99 SELECT object_type, object_name, owner FROM ( | |
| 100 SELECT object_type, object_name, user owner, 1 priority | |
| 101 FROM user_objects | |
| 102 WHERE object_name = :objName | |
| 103 UNION ALL | |
| 104 SELECT ao.object_type, ao.object_name, ao.owner, 2 priority | |
| 105 FROM all_objects ao | |
| 106 JOIN user_synonyms us ON (us.table_owner = ao.owner AND us.table_name = ao.object_name) | |
| 107 WHERE us.synonym_name = :objName | |
| 108 AND ao.object_type != 'SYNONYM' | |
| 109 UNION ALL | |
| 110 SELECT ao.object_type, ao.object_name, ao.owner, 3 priority | |
| 111 FROM all_objects ao | |
| 112 JOIN all_synonyms asyn ON (asyn.table_owner = ao.owner AND asyn.table_name = ao.object_name) | |
| 113 WHERE asyn.synonym_name = :objName | |
| 114 AND ao.object_type != 'SYNONYM' | |
| 115 AND asyn.owner = 'PUBLIC' | |
| 116 ) ORDER BY priority ASC""", | |
| 117 'tabComments': """ | |
| 118 SELECT comments | |
| 119 FROM all_tab_comments | |
| 120 WHERE owner = :owner | |
| 121 AND table_name = :table_name""", | |
| 122 'colComments': """ | |
| 123 atc.column_name, | |
| 124 acc.comments | |
| 125 FROM all_tab_columns atc | |
| 126 JOIN all_col_comments acc ON (atc.owner = acc.owner and atc.table_name = acc.table_name and atc.column_name = acc.column_name) | |
| 127 WHERE atc.table_name = :object_name | |
| 128 AND atc.owner = :owner | |
| 129 ORDER BY atc.column_id;""", | |
| 130 } | |
| 131 | |
| 132 import sys, os, re, sqlpython, cx_Oracle, pyparsing | |
| 133 | |
| 134 if float(sys.version[:3]) < 2.3: | |
| 135 def enumerate(lst): | |
| 136 return zip(range(len(lst)), lst) | |
| 137 | |
| 138 class SoftwareSearcher(object): | |
| 139 def __init__(self, softwareList, purpose): | |
| 140 self.softwareList = softwareList | |
| 141 self.purpose = purpose | |
| 142 self.software = None | |
| 143 def invoke(self, *args): | |
| 144 if not self.software: | |
| 145 (self.software, self.invokeString) = self.find() | |
| 146 argTuple = tuple([self.software] + list(args)) | |
| 147 os.system(self.invokeString % argTuple) | |
| 148 def find(self): | |
| 149 if self.purpose == 'text editor': | |
| 150 software = os.environ.get('EDITOR') | |
| 151 if software: | |
| 152 return (software, '%s %s') | |
| 153 for (n, (software, invokeString)) in enumerate(self.softwareList): | |
| 154 if os.path.exists(software): | |
| 155 if n > (len(self.softwareList) * 0.7): | |
| 156 print """ | |
| 157 | |
| 158 Using %s. Note that there are better options available for %s, | |
| 159 but %s couldn't find a better one in your PATH. | |
| 160 Feel free to open up %s | |
| 161 and customize it to find your favorite %s program. | |
| 162 | |
| 163 """ % (software, self.purpose, __file__, __file__, self.purpose) | |
| 164 return (software, invokeString) | |
| 165 stem = os.path.split(software)[1] | |
| 166 for p in os.environ['PATH'].split(os.pathsep): | |
| 167 if os.path.exists(os.sep.join([p, stem])): | |
| 168 return (stem, invokeString) | |
| 169 raise (OSError, """Could not find any %s programs. You will need to install one, | |
| 170 or customize %s to make it aware of yours. | |
| 171 Looked for these programs: | |
| 172 %s""" % (self.purpose, __file__, "\n".join([s[0] for s in self.softwareList]))) | |
| 173 #v2.4: %s""" % (self.purpose, __file__, "\n".join(s[0] for s in self.softwareList))) | |
| 174 | |
| 175 softwareLists = { | |
| 176 'diff/merge': [ | |
| 177 ('/usr/bin/meld',"%s %s %s"), | |
| 178 ('/usr/bin/kdiff3',"%s %s %s"), | |
| 179 (r'C:\Program Files\Araxis\Araxis Merge v6.5\Merge.exe','"%s" %s %s'), | |
| 180 (r'C:\Program Files\TortoiseSVN\bin\TortoiseMerge.exe', '"%s" /base:"%s" /mine:"%s"'), | |
| 181 ('FileMerge','%s %s %s'), | |
| 182 ('kompare','%s %s %s'), | |
| 183 ('WinMerge','%s %s %s'), | |
| 184 ('xxdiff','%s %s %s'), | |
| 185 ('fldiff','%s %s %s'), | |
| 186 ('gtkdiff','%s %s %s'), | |
| 187 ('tkdiff','%s %s %s'), | |
| 188 ('gvimdiff','%s %s %s'), | |
| 189 ('diff',"%s %s %s"), | |
| 190 (r'c:\windows\system32\comp.exe',"%s %s %s")], | |
| 191 'text editor': [ | |
| 192 ('gedit', '%s %s'), | |
| 193 ('textpad', '%s %s'), | |
| 194 ('notepad.exe', '%s %s'), | |
| 195 ('pico', '%s %s'), | |
| 196 ('emacs', '%s %s'), | |
| 197 ('vim', '%s %s'), | |
| 198 ('vi', '%s %s'), | |
| 199 ('ed', '%s %s'), | |
| 200 ('edlin', '%s %s') | |
| 201 ] | |
| 202 } | |
| 203 | |
| 204 diffMergeSearcher = SoftwareSearcher(softwareLists['diff/merge'],'diff/merge') | |
| 205 editSearcher = SoftwareSearcher(softwareLists['text editor'], 'text editor') | |
| 206 editor = os.environ.get('EDITOR') | |
| 207 if editor: | |
| 208 editSearcher.find = lambda: (editor, "%s %s") | |
| 209 | |
| 210 class CaselessDict(dict): | |
| 211 """dict with case-insensitive keys. | |
| 212 | |
| 213 Posted to ASPN Python Cookbook by Jeff Donner - http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/66315""" | |
| 214 def __init__(self, other=None): | |
| 215 if other: | |
| 216 # Doesn't do keyword args | |
| 217 if isinstance(other, dict): | |
| 218 for k,v in other.items(): | |
| 219 dict.__setitem__(self, k.lower(), v) | |
| 220 else: | |
| 221 for k,v in other: | |
| 222 dict.__setitem__(self, k.lower(), v) | |
| 223 def __getitem__(self, key): | |
| 224 return dict.__getitem__(self, key.lower()) | |
| 225 def __setitem__(self, key, value): | |
| 226 dict.__setitem__(self, key.lower(), value) | |
| 227 def __contains__(self, key): | |
| 228 return dict.__contains__(self, key.lower()) | |
| 229 def has_key(self, key): | |
| 230 return dict.has_key(self, key.lower()) | |
| 231 def get(self, key, def_val=None): | |
| 232 return dict.get(self, key.lower(), def_val) | |
| 233 def setdefault(self, key, def_val=None): | |
| 234 return dict.setdefault(self, key.lower(), def_val) | |
| 235 def update(self, other): | |
| 236 for k,v in other.items(): | |
| 237 dict.__setitem__(self, k.lower(), v) | |
| 238 def fromkeys(self, iterable, value=None): | |
| 239 d = CaselessDict() | |
| 240 for k in iterable: | |
| 241 dict.__setitem__(d, k.lower(), value) | |
| 242 return d | |
| 243 def pop(self, key, def_val=None): | |
| 244 return dict.pop(self, key.lower(), def_val) | |
| 245 | |
| 246 class NotSettableError(Exception): | |
| 247 None | |
| 248 | |
| 249 class Parser(object): | |
| 250 comment_def = "--" + pyparsing.ZeroOrMore(pyparsing.CharsNotIn("\n")) | |
| 251 def __init__(self, scanner, retainSeparator=True): | |
| 252 self.scanner = scanner | |
| 253 self.scanner.ignore(pyparsing.sglQuotedString) | |
| 254 self.scanner.ignore(pyparsing.dblQuotedString) | |
| 255 self.scanner.ignore(self.comment_def) | |
| 256 self.scanner.ignore(pyparsing.cStyleComment) | |
| 257 self.retainSeparator = retainSeparator | |
| 258 def separate(self, txt): | |
| 259 itms = [] | |
| 260 for (sqlcommand, start, end) in self.scanner.scanString(txt): | |
| 261 if sqlcommand: | |
| 262 if type(sqlcommand[0]) == pyparsing.ParseResults: | |
| 263 if self.retainSeparator: | |
| 264 itms.append("".join(sqlcommand[0])) | |
| 265 else: | |
| 266 itms.append(sqlcommand[0][0]) | |
| 267 else: | |
| 268 if sqlcommand[0]: | |
| 269 itms.append(sqlcommand[0]) | |
| 270 return itms | |
| 271 | |
| 272 pipeSeparator = Parser(pyparsing.SkipTo((pyparsing.Literal('|') ^ pyparsing.StringEnd()), include=True), retainSeparator=False) | |
| 273 bindScanner = Parser(pyparsing.Literal(':') + pyparsing.Word( pyparsing.alphanums + "_$#" )) | |
| 274 commandSeparator = Parser(pyparsing.SkipTo((pyparsing.Literal(';') ^ pyparsing.StringEnd()), include=True)) | |
| 275 | |
| 276 def findBinds(target, existingBinds, givenBindVars = {}): | |
| 277 result = givenBindVars | |
| 278 for finding, startat, endat in bindScanner.scanner.scanString(target): | |
| 279 varname = finding[1] | |
| 280 try: | |
| 281 result[varname] = existingBinds[varname] | |
| 282 except KeyError: | |
| 283 if not givenBindVars.has_key(varname): | |
| 284 print 'Bind variable %s not defined.' % (varname) | |
| 285 return result | |
| 286 | |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
287 class HistoryItem(str): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
288 def __init__(self, instr): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
289 str.__init__(self, instr) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
290 self.lowercase = self.lower() |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
291 self.idx = None |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
292 def pr(self): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
293 print '-------------------------[%d]' % (self.idx) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
294 print self |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
295 |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
296 class History(list): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
297 def append(self, new): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
298 new = HistoryItem(new) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
299 list.append(self, new) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
300 new.idx = len(self) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
301 def extend(self, new): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
302 for n in new: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
303 self.append(n) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
304 def get(self, getme): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
305 try: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
306 idx = int(getme) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
307 try: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
308 return [self[idx-1]] |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
309 except IndexError: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
310 return [] |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
311 except ValueError: # search for a string |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
312 try: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
313 getme = getme.strip() |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
314 except: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
315 print "Don't know how to handle %s." % (str(getme)) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
316 return |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
317 if getme.startswith(r'/') and getme.endswith(r'/'): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
318 finder = re.compile(getme[1:-1], re.DOTALL | re.MULTILINE | re.IGNORECASE) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
319 def isin(hi): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
320 return finder.search(hi) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
321 else: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
322 def isin(hi): |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
323 return (getme.lower() in hi.lowercase) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
324 return [itm for itm in self[:-1] if isin(itm)] |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
325 |
| 0 | 326 class sqlpyPlus(sqlpython.sqlpython): |
| 327 def __init__(self): | |
| 328 sqlpython.sqlpython.__init__(self) | |
| 329 self.binds = CaselessDict() | |
| 330 self.sqlBuffer = [] | |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
331 self.history = History() |
| 0 | 332 self.settable = ['maxtselctrows', 'maxfetch', 'autobind', 'failover', 'timeout'] # settables must be lowercase |
| 333 self.stdoutBeforeSpool = sys.stdout | |
| 334 self.spoolFile = None | |
| 335 self.autobind = False | |
| 336 self.failover = False | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
337 self.multiline = '''select insert update delete tselect |
|
2
59903dcaf327
working on history
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
1
diff
changeset
|
338 create drop alter'''.split() |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
339 self.excludeFromHistory = '''run r list l history hi ed'''.split() |
| 0 | 340 |
| 341 def default(self, arg, do_everywhere=False): | |
| 342 sqlpython.sqlpython.default(self, arg, do_everywhere) | |
| 343 self.sqlBuffer.append(self.query) | |
| 344 | |
| 345 # overrides cmd's parseline | |
| 346 shortcuts = {'?': 'help', '@': 'getrun', '!': 'shell', ':': 'setbind', '\\': 'psql'} | |
| 347 def parseline(self, line): | |
| 348 """Parse the line into a command name and a string containing | |
| 349 the arguments. Returns a tuple containing (command, args, line). | |
| 350 'command' and 'args' may be None if the line couldn't be parsed. | |
| 351 Overrides cmd.cmd.parseline to accept variety of shortcuts..""" | |
| 352 line = line.strip() | |
| 353 if not line: | |
| 354 return None, None, line | |
| 355 shortcut = self.shortcuts.get(line[0]) | |
| 356 if shortcut: | |
| 357 cmd, arg = shortcut, line[1:].strip() | |
| 358 else: | |
| 359 i, n = 0, len(line) | |
| 360 while i < n and line[i] in self.identchars: i = i+1 | |
| 361 cmd, arg = line[:i], line[i:].strip() | |
| 362 if cmd.lower() in ('select', 'sleect', 'insert', 'update', 'delete', 'describe', | |
| 363 'desc', 'comments') \ | |
| 364 and not hasattr(self, 'curs'): | |
| 365 print 'Not connected.' | |
| 366 return '', '', '' | |
| 367 return cmd, arg, line | |
| 368 | |
| 369 def precmd(self, line): | |
| 370 """Hook method executed just before the command line is | |
| 371 interpreted, but after the input prompt is generated and issued. | |
| 372 Makes commands case-insensitive (but unfortunately does not alter command completion). | |
| 373 """ | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
374 |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
375 ''' |
| 0 | 376 pipedCommands = pipeSeparator.separate(line) |
| 377 if len(pipedCommands) > 1: | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
378 pipefilename = 'sqlpython.pipe.tmp' |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
379 for (idx, pipedCommand) in enumerate(pipedCommands[:-1]): |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
380 savestdout = sys.stdout |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
381 f = open(pipefilename,'w') |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
382 sys.stdout = f |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
383 self.precmd(pipedCommand) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
384 self.onecmd(pipedCommand) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
385 self.postcmd(False, pipedCommands[0]) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
386 f.close() |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
387 sys.stdout = savestdout |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
388 f = os.popen('%s < %s' % (pipedCommands[idx+1], pipefilename)) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
389 f.read() |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
390 |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
391 ''' |
| 0 | 392 try: |
| 393 args = line.split(None,1) | |
| 394 args[0] = args[0].lower() | |
|
2
59903dcaf327
working on history
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
1
diff
changeset
|
395 statement = ' '.join(args) |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
396 if args[0] in self.multiline: |
|
2
59903dcaf327
working on history
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
1
diff
changeset
|
397 statement = sqlpython.finishStatement(statement) |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
398 if args[0] not in self.excludeFromHistory: |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
399 self.history.append(statement) |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
400 return statement |
| 0 | 401 except Exception: |
| 402 return line | |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
403 |
| 0 | 404 def do_shortcuts(self,arg): |
| 405 """Lists available first-character shortcuts | |
| 406 (i.e. '!dir' is equivalent to 'shell dir')""" | |
| 407 for (scchar, scto) in self.shortcuts.items(): | |
| 408 print '%s: %s' % (scchar, scto) | |
| 409 | |
| 410 | |
| 411 def colnames(self): | |
| 412 return [d[0] for d in curs.description] | |
| 413 | |
| 414 def sql_format_itm(self, itm, needsquotes): | |
| 415 if itm is None: | |
| 416 return 'NULL' | |
| 417 if needsquotes: | |
| 418 return "'%s'" % str(itm) | |
| 419 return str(itm) | |
| 420 def output_as_insert_statements(self): | |
| 421 usequotes = [d[1] != cx_Oracle.NUMBER for d in self.curs.description] | |
| 422 def formatRow(row): | |
| 423 return ','.join(self.sql_format_itm(itm, useq) | |
| 424 for (itm, useq) in zip(row, usequotes)) | |
| 425 result = ['INSERT INTO %s (%s) VALUES (%s);' % | |
| 426 (self.tblname, ','.join(self.colnames), formatRow(row)) | |
| 427 for row in self.rows] | |
| 428 return '\n'.join(result) | |
| 429 | |
| 430 def output_row_as_xml(self, row): | |
| 431 result = [' <%s>\n %s\n </%s>' % | |
| 432 (colname.lower(), str('' if (itm is None) else itm), colname.lower()) | |
| 433 for (itm, colname) in zip(row, self.colnames)] | |
| 434 return '\n'.join(result) | |
| 435 def output_as_xml(self): | |
| 436 result = ['<%s>\n%s\n</%s>' % | |
| 437 (self.tblname, self.output_row_as_xml(row), self.tblname) | |
| 438 for row in self.rows] | |
| 439 return '\n'.join(result) | |
| 440 | |
| 441 def output_as_html_table(self): | |
| 442 result = ''.join('<th>%s</th>' % c for c in self.colnames) | |
| 443 result = [' <tr>\n %s\n </tr>' % result] | |
| 444 for row in self.rows: | |
| 445 result.append(' <tr>\n %s\n </tr>' % | |
| 446 (''.join('<td>%s</td>' % | |
| 447 str('' if (itm is None) else itm) | |
| 448 for itm in row))) | |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
449 result = '''<table id="%s"> |
| 0 | 450 %s |
| 451 </table>''' % (self.tblname, '\n'.join(result)) | |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
452 return result |
| 0 | 453 |
| 454 def output_as_list(self, align): | |
| 455 result = [] | |
| 456 colnamelen = max(len(colname) for colname in self.colnames) + 1 | |
| 457 for (idx, row) in enumerate(self.rows): | |
| 458 result.append('\n**** Row: %d' % (idx+1)) | |
| 459 for (itm, colname) in zip(row, self.colnames): | |
| 460 if align: | |
| 461 colname = colname.ljust(colnamelen) | |
| 462 result.append('%s: %s' % (colname, itm)) | |
| 463 return '\n'.join(result) | |
| 464 | |
| 465 tableNameFinder = re.compile(r'from\s+([\w$#_"]+)', re.IGNORECASE | re.MULTILINE | re.DOTALL) | |
| 466 def output(self, outformat, rowlimit): | |
| 467 self.tblname = self.tableNameFinder.search(self.curs.statement).group(1) | |
| 468 self.colnames = [d[0] for d in self.curs.description] | |
| 469 if outformat == '\\i': | |
| 470 result = self.output_as_insert_statements() | |
| 471 elif outformat == '\\x': | |
| 472 result = self.output_as_xml() | |
| 473 elif outformat == '\\g': | |
| 474 result = self.output_as_list(align=False) | |
| 475 elif outformat == '\\G': | |
| 476 result = self.output_as_list(align=True) | |
| 477 elif outformat in ('\\s', '\\S', '\\c', '\\C'): #csv | |
| 478 result = [] | |
| 479 if outformat in ('\\s', '\\c'): | |
| 480 result.append(','.join('"%s"' % colname for colname in self.colnames)) | |
| 481 for row in self.rows: | |
| 482 result.append(','.join('"%s"' % ('' if itm is None else itm) for itm in row)) | |
| 483 result = '\n'.join(result) | |
| 484 elif outformat == '\\h': | |
| 485 result = self.output_as_html_table() | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
486 elif outformat == '\\t': |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
487 |
| 0 | 488 else: |
| 489 result = sqlpython.pmatrix(self.rows, self.curs.description, self.maxfetch) | |
| 490 return result | |
| 491 | |
| 492 def do_select(self, arg, bindVarsIn=None): | |
| 493 """Fetch rows from a table. | |
| 494 | |
| 495 Limit the number of rows retrieved by appending | |
| 496 an integer after the terminator | |
| 497 (example: SELECT * FROM mytable;10 ) | |
| 498 | |
| 499 Output may be formatted by choosing an alternative terminator | |
| 500 ("help terminators" for details) | |
| 501 """ | |
| 502 bindVarsIn = bindVarsIn or {} | |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
503 self.query = 'select ' + arg |
|
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
504 (self.query, terminator, rowlimit) = sqlpython.findTerminator(self.query) |
|
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
505 rowlimit = int(rowlimit or 0) |
|
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
506 if terminator == '\\t': |
|
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
507 self.do_tselect(' '.join(self.query.split()[1:]) + ';', rowlimit) |
|
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
508 return |
| 0 | 509 else: |
| 510 try: | |
| 511 self.varsUsed = findBinds(self.query, self.binds, bindVarsIn) | |
| 512 self.curs.execute(self.query, self.varsUsed) | |
|
2
59903dcaf327
working on history
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
1
diff
changeset
|
513 self.rows = self.curs.fetchmany(min(self.maxfetch, (rowlimit or self.maxfetch))) |
| 0 | 514 self.desc = self.curs.description |
| 515 self.rc = self.curs.rowcount | |
| 516 if self.rc > 0: | |
|
2
59903dcaf327
working on history
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
1
diff
changeset
|
517 print '\n' + self.output(terminator, rowlimit) |
| 0 | 518 if self.rc == 0: |
| 519 print '\nNo rows Selected.\n' | |
| 520 elif self.rc == 1: | |
| 521 print '\n1 row selected.\n' | |
| 522 if self.autobind: | |
| 523 self.binds.update(dict(zip([d[0] for d in self.desc], self.rows[0]))) | |
| 524 elif self.rc < self.maxfetch: | |
| 525 print '\n%d rows selected.\n' % self.rc | |
| 526 else: | |
| 527 print '\nSelected Max Num rows (%d)' % self.rc | |
| 528 except Exception, e: | |
| 529 print e | |
| 530 import traceback | |
| 531 traceback.print_exc(file=sys.stdout) | |
| 532 self.sqlBuffer.append(self.query) | |
| 533 | |
| 534 def showParam(self, param): | |
| 535 param = param.strip().lower() | |
| 536 if param in self.settable: | |
| 537 val = getattr(self, param) | |
| 538 print '%s: %s' % (param, str(getattr(self, param))) | |
| 539 | |
| 540 def do_show(self, arg): | |
| 541 'Shows value of a (sqlpython, not ORACLE) parameter' | |
| 542 arg = arg.strip().lower() | |
| 543 if arg: | |
| 544 self.showParam(arg) | |
| 545 else: | |
| 546 for param in self.settable: | |
| 547 self.showParam(param) | |
| 548 | |
| 549 def cast(self, current, new): | |
| 550 typ = type(current) | |
| 551 if typ == bool: | |
| 552 new = new.lower() | |
| 553 try: | |
| 554 if (new=='on') or (new[0] in ('y','t')): | |
| 555 return True | |
| 556 return False | |
| 557 except TypeError: | |
| 558 None | |
| 559 try: | |
| 560 return typ(new) | |
| 561 except: | |
| 562 print "Problem setting parameter (now %s) to %s; incorrect type?" % (current, new) | |
| 563 return current | |
| 564 | |
| 565 def do_set(self, arg): | |
| 566 'Sets a (sqlpython, not ORACLE) parameter' | |
| 567 try: | |
| 568 paramName, val = arg.split(None, 1) | |
| 569 except Exception: | |
| 570 self.do_show(arg) | |
| 571 return | |
| 572 paramName = paramName.lower() | |
| 573 try: | |
| 574 current = getattr(self, paramName) | |
| 575 if callable(current): | |
| 576 raise NotSettableError | |
| 577 except (AttributeError, NotSettableError): | |
| 578 self.fail('set %s' % arg) | |
| 579 return | |
| 580 val = self.cast(current, val.strip(';')) | |
| 581 print paramName, ' - was: ', current | |
| 582 setattr(self, paramName.lower(), val) | |
| 583 print 'now: ', val | |
| 584 | |
| 585 def do_describe(self, arg): | |
| 586 "emulates SQL*Plus's DESCRIBE" | |
| 587 object_type, owner, object_name = self.resolve(arg.strip(self.terminator).upper()) | |
| 588 print "%s %s.%s" % (object_type, owner, object_name) | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
589 if object_type in ('TABLE','VIEW','PROCEDURE','FUNCTION','TRIGGER'): |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
590 self.do_select(descQueries[object_type],{'object_name':object_name, 'owner':owner}) |
| 0 | 591 elif object_type == 'PACKAGE': |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
592 self.curs.execute(descQueries['PackageObjects'], {'package_name':object_name, 'owner':owner}) |
| 0 | 593 for (packageObj_name,) in self.curs: |
| 594 print packageObj_name | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
595 self.do_select(descQueries['PackageObjArgs'],{'package_name':object_name, 'owner':owner, 'object_name':packageObj_name}) |
| 0 | 596 do_desc = do_describe |
| 597 | |
| 598 def do_comments(self, arg): | |
| 599 'Prints comments on a table and its columns.' | |
| 600 object_type, owner, object_name = self.resolve(arg.strip(self.terminator).upper()) | |
| 601 if object_type: | |
| 602 self.curs.execute(queries['tabComments'],{'table_name':object_name, 'owner':owner}) | |
| 603 print "%s %s.%s: %s" % (object_type, owner, object_name, self.curs.fetchone()[0]) | |
| 604 self.do_select(queries['colComments'],{'owner':owner, 'object_name': object_name}) | |
| 605 | |
| 606 def resolve(self, identifier): | |
| 607 """Checks (my objects).name, (my synonyms).name, (public synonyms).name | |
| 608 to resolve a database object's name. """ | |
| 609 parts = identifier.split('.') | |
| 610 try: | |
| 611 if len(parts) == 2: | |
| 612 owner, object_name = parts | |
| 613 self.curs.execute('SELECT object_type FROM all_objects WHERE owner = :owner AND object_name = :object_name', | |
| 614 {'owner': owner, 'object_name': object_name}) | |
| 615 object_type = self.curs.fetchone()[0] | |
| 616 elif len(parts) == 1: | |
| 617 object_name = parts[0] | |
| 618 self.curs.execute(queries['resolve'], {'objName':object_name}) | |
| 619 object_type, object_name, owner = self.curs.fetchone() | |
| 620 except TypeError: | |
| 621 print 'Could not resolve object %s.' % identifier | |
| 622 object_type, owner, object_name = '', '', '' | |
| 623 return object_type, owner, object_name | |
| 624 | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
625 def do_resolve(self, arg): |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
626 print self.resolve(arg) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
627 |
| 0 | 628 def do_shell(self, arg): |
| 629 'execute a command as if at the OS prompt.' | |
| 630 os.system(arg) | |
| 631 | |
| 632 def spoolstop(self): | |
| 633 if self.spoolFile: | |
| 634 sys.stdout = self.stdoutBeforeSpool | |
| 635 print 'Finished spooling to ', self.spoolFile.name | |
| 636 self.spoolFile.close() | |
| 637 self.spoolFile = None | |
| 638 | |
| 639 def do_spool(self, arg): | |
| 640 """spool [filename] - begins redirecting output to FILENAME.""" | |
| 641 self.spoolstop() | |
| 642 arg = arg.strip() | |
| 643 if not arg: | |
| 644 arg = 'output.lst' | |
| 645 if arg.lower() != 'off': | |
| 646 if '.' not in arg: | |
| 647 arg = '%s.lst' % arg | |
| 648 print 'Sending output to %s (until SPOOL OFF received)' % (arg) | |
| 649 self.spoolFile = open(arg, 'w') | |
| 650 sys.stdout = self.spoolFile | |
| 651 | |
| 652 def write(self, arg, fname): | |
| 653 originalOut = sys.stdout | |
| 654 f = open(fname, 'w') | |
| 655 sys.stdout = f | |
| 656 self.onecmd(arg) | |
| 657 f.close() | |
| 658 sys.stdout = originalOut | |
| 659 | |
| 660 def do_write(self, args): | |
| 661 'write [filename.extension] query - writes result to a file' | |
| 662 words = args.split(None, 1) | |
| 663 if len(words) > 1 and '.' in words[0]: | |
| 664 fname, command = words | |
| 665 else: | |
| 666 fname, command = 'output.txt', args | |
| 667 self.write(command, fname) | |
| 668 print 'Results written to %s' % os.path.join(os.getcwd(), fname) | |
| 669 | |
| 670 def do_compare(self, args): | |
| 671 """COMPARE query1 TO query2 - uses external tool to display differences. | |
| 672 | |
| 673 Sorting is recommended to avoid false hits.""" | |
| 674 fnames = [] | |
| 675 args2 = args.split(' to ') | |
| 676 for n in range(len(args2)): | |
| 677 query = args2[n] | |
| 678 fnames.append('compare%s.txt' % n) | |
| 679 if query.rstrip()[-1] != self.terminator: | |
| 680 query = '%s%s' % (query, self.terminator) | |
| 681 self.write(query, fnames[n]) | |
| 682 diffMergeSearcher.invoke(fnames[0], fnames[1]) | |
| 683 | |
| 684 bufferPosPattern = re.compile('\d+') | |
| 685 rangeIndicators = ('-',':') | |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
686 |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
687 def last_matching_command(self, arg): |
| 0 | 688 if not arg: |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
689 return self.history[-2] |
| 0 | 690 else: |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
691 history = self.history.get(arg) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
692 if history: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
693 return history[-1] |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
694 return None |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
695 |
| 0 | 696 def do_run(self, arg): |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
697 """run [arg]: re-runs an earlier command |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
698 |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
699 no arg -> run most recent command |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
700 arg is integer -> run one history item, by index |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
701 arg is string -> run most recent command by string search |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
702 arg is /enclosed in forward-slashes/ -> run most recent by regex |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
703 """ |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
704 'run [N]: runs the SQL that was run N commands ago' |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
705 runme = self.last_matching_command(arg) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
706 print runme |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
707 self.onecmd(runme) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
708 do_r = do_run |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
709 def do_history(self, arg): |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
710 """history [arg]: lists past commands issued |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
711 |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
712 no arg -> list all |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
713 arg is integer -> list one history item, by index |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
714 arg is string -> string search |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
715 arg is /enclosed in forward-slashes/ -> regular expression search |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
716 """ |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
717 if arg: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
718 history = self.history.get(arg) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
719 else: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
720 history = self.history |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
721 for hi in history: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
722 hi.pr() |
| 0 | 723 def do_list(self, arg): |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
724 """list: lists single most recent command issued""" |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
725 self.last_matching_command(None).pr() |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
726 do_hi = do_history |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
727 do_l = do_list |
| 0 | 728 def load(self, fname): |
| 729 """Pulls command(s) into sql buffer. Returns number of commands loaded.""" | |
| 730 try: | |
| 731 f = open(fname, 'r') | |
| 732 except IOError, e: | |
| 733 try: | |
| 734 f = open('%s.sql' % fname, 'r') | |
| 735 except: | |
| 736 print 'Problem opening file %s: \n%s' % (fname, e) | |
| 737 return 0 | |
| 738 txt = f.read() | |
| 739 f.close() | |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
740 result = commandSeparator.separate(txt) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
741 self.history.extend(result) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
742 return len(result) |
| 0 | 743 def do_ed(self, arg): |
| 744 'ed [N]: brings up SQL from N commands ago in text editor, and puts result in SQL buffer.' | |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
745 fname = 'sqlpython_temp.sql' |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
746 buffer = self.last_matching_command(arg) |
| 0 | 747 f = open(fname, 'w') |
| 748 f.write(buffer) | |
| 749 f.close() | |
| 750 editSearcher.invoke(fname) | |
| 751 self.load(fname) | |
| 752 do_edit = do_ed | |
| 753 def do_get(self, fname): | |
| 754 'Brings SQL commands from a file to the in-memory SQL buffer.' | |
| 755 commandsLoaded = self.load(fname) | |
| 756 if commandsLoaded: | |
| 757 self.do_list('1-%d' % (commandsLoaded-1)) | |
| 758 def do_getrun(self, fname): | |
| 759 'Brings SQL commands from a file to the in-memory SQL buffer, and executes them.' | |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
760 newCommands = self.load(fname) * -1 |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
761 if newCommands: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
762 for command in self.history[newCommands:]: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
763 self.onecmd(command) |
| 0 | 764 def do_psql(self, arg): |
| 765 '''Shortcut commands emulating psql's backslash commands. | |
| 766 | |
| 767 \c connect | |
| 768 \d desc | |
| 769 \e edit | |
| 770 \g run | |
| 771 \h help | |
| 772 \i getrun | |
| 773 \o spool | |
| 774 \p list | |
| 775 \w save | |
| 776 \? help psql''' | |
| 777 commands = {} | |
| 778 for c in self.do_psql.__doc__.splitlines()[2:]: | |
| 779 (abbrev, command) = c.split(None, 1) | |
| 780 commands[abbrev[1:]] = command | |
| 781 words = arg.split(None,1) | |
| 782 abbrev = words[0] | |
| 783 try: | |
| 784 args = words[1] | |
| 785 except IndexError: | |
| 786 args = '' | |
| 787 try: | |
| 788 self.onecmd('%s %s' % (commands[abbrev], args)) | |
| 789 self.onecmd('q') | |
| 790 except KeyError: | |
| 791 print 'psql command \%s not yet supported.' % abbrev | |
| 792 def do_save(self, fname): | |
| 793 'save FILENAME: Saves most recent SQL command to disk.' | |
|
3
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
794 try: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
795 f = open(fname, 'w') |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
796 f.write(self.sqlBuffer[-1]) |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
797 f.close() |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
798 except Exception, e: |
|
cd23cd62de3c
history working pretty well
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
2
diff
changeset
|
799 print 'Error saving %s: %s' % (fname, str(e)) |
| 0 | 800 |
| 801 def do_print(self, arg): | |
| 802 'print VARNAME: Show current value of bind variable VARNAME.' | |
| 803 if arg: | |
| 804 if arg[0] == ':': | |
| 805 arg = arg[1:] | |
| 806 try: | |
| 807 print self.binds[arg] | |
| 808 except KeyError: | |
| 809 print 'No bind variable ', arg | |
| 810 else: | |
| 811 self.do_setbind('') | |
| 812 def do_setbind(self, arg): | |
| 813 args = arg.split(None, 2) | |
| 814 if len(args) == 0: | |
| 815 for (var, val) in self.binds.items(): | |
| 816 print ':%s = %s' % (var, val) | |
| 817 elif len(args) == 1: | |
| 818 try: | |
| 819 print ':%s = %s' % (args[0], self.binds[args[0]]) | |
| 820 except KeyError, e: | |
| 821 print noSuchBindMsg % args[0] | |
| 822 elif len(args) > 2 and args[1] in ('=',':='): | |
| 823 var, val = args[0], args[2] | |
| 824 if val[0] == val[-1] == "'" and len(val) > 1: | |
| 825 val = val[1:-1] | |
| 826 self.binds[var] = val | |
| 827 else: | |
| 828 print 'Could not parse ', args | |
| 829 def do_exec(self, arg): | |
| 830 if arg[0] == ':': | |
| 831 self.do_setbind(arg[1:]) | |
| 832 else: | |
| 833 self.default('exec %s' % arg) | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
834 |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
835 def do_cat(self, arg): |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
836 targets = arg.split() |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
837 for target in targets: |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
838 self.do_select('* from %s' % target) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
839 |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
840 def do_grep(self, arg): |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
841 """grep PATTERN TABLE - search for term in any of TABLE's fields""" |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
842 targets = arg.split() |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
843 pattern = targets.pop(0) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
844 for target in targets: |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
845 sql = [] |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
846 self.curs.execute('select * from %s where 1=0' % target) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
847 sql = ' or '.join("%s LIKE '%%%s%%'" % (d[0], pattern) for d in self.curs.description) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
848 sql = '* FROM %s WHERE %s' % (target, sql) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
849 self.do_select(sql) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
850 |
| 0 | 851 def _test(): |
| 852 import doctest | |
| 853 doctest.testmod() | |
| 854 | |
| 855 if __name__ == "__main__": | |
| 856 "Silent return implies that all unit tests succeeded. Use -v to see details." | |
| 857 _test() |