Mercurial > sqlpython
annotate sqlpyPlus.py @ 113:d48034a42b26
slash-endings working
| author | catherine@Elli.myhome.westell.com |
|---|---|
| date | Mon, 23 Jun 2008 22:01:51 -0400 |
| parents | 5e57765073f4 |
| children | c1c05670b4e5 |
| 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 | |
|
112
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
13 - compare ... to ... grweaphically compares results of two queries |
| 0 | 14 - commands are case-insensitive |
| 80 | 15 |
| 0 | 16 Use 'help' within sqlpython for details. |
| 17 | |
| 18 Set bind variables the hard (SQL*Plus) way | |
| 19 exec :b = 3 | |
| 20 or with a python-like shorthand | |
| 21 :b = 3 | |
| 22 | |
| 23 - catherinedevlin.blogspot.com May 31, 2006 | |
| 24 """ | |
| 25 # note in cmd.cmd about supporting emacs commands? | |
| 26 | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
27 descQueries = { |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
28 'TABLE': (""" |
| 80 | 29 atc.column_name, |
| 30 CASE atc.nullable WHEN 'Y' THEN 'NULL' ELSE 'NOT NULL' END "Null?", | |
| 31 atc.data_type || | |
| 32 CASE atc.data_type WHEN 'DATE' THEN '' | |
| 33 ELSE '(' || | |
| 34 CASE atc.data_type WHEN 'NUMBER' THEN TO_CHAR(atc.data_precision) || | |
| 35 CASE atc.data_scale WHEN 0 THEN '' | |
| 36 ELSE ',' || TO_CHAR(atc.data_scale) END | |
| 37 ELSE TO_CHAR(atc.data_length) END | |
| 38 END || | |
| 39 CASE atc.data_type WHEN 'DATE' THEN '' ELSE ')' END | |
| 40 data_type | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
41 FROM all_tab_columns atc |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
42 WHERE atc.table_name = :object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
43 AND atc.owner = :owner |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
44 ORDER BY atc.column_id;""",), |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
45 'PROCEDURE': (""" |
| 80 | 46 argument_name, |
| 47 data_type, | |
| 48 in_out, | |
| 49 default_value | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
50 FROM all_arguments |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
51 WHERE object_name = :object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
52 AND owner = :owner |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
53 AND package_name IS NULL |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
54 AND argument_name IS NOT NULL |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
55 ORDER BY sequence;""",), |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
56 'PackageObjects':(""" |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
57 SELECT DISTINCT object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
58 FROM all_arguments |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
59 WHERE package_name = :package_name |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
60 AND owner = :owner""",), |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
61 'PackageObjArgs':(""" |
| 80 | 62 object_name, |
| 63 argument_name, | |
| 64 data_type, | |
| 65 in_out, | |
| 66 default_value | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
67 FROM all_arguments |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
68 WHERE package_name = :package_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
69 AND object_name = :object_name |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
70 AND owner = :owner |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
71 AND argument_name IS NOT NULL |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
72 ORDER BY sequence""",), |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
73 'TRIGGER':(""" |
| 80 | 74 description |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
75 FROM all_triggers |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
76 WHERE owner = :owner |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
77 AND trigger_name = :object_name |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
78 """, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
79 """ |
| 80 | 80 table_owner, |
| 81 base_object_type, | |
| 82 table_name, | |
| 83 column_name, | |
| 84 when_clause, | |
| 85 status, | |
| 86 action_type, | |
| 87 crossedition | |
|
4
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 |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
92 """, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
93 ), |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
94 'INDEX':(""" |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
95 index_type, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
96 table_owner, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
97 table_name, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
98 table_type, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
99 uniqueness, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
100 compression, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
101 partitioned, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
102 temporary, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
103 generated, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
104 secondary, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
105 dropped, |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
106 visibility |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
107 FROM all_indexes |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
108 WHERE owner = :owner |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
109 AND index_name = :object_name |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
110 \\t |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
111 """,) |
| 80 | 112 } |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
113 descQueries['VIEW'] = descQueries['TABLE'] |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
114 descQueries['FUNCTION'] = descQueries['PROCEDURE'] |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
115 |
| 0 | 116 queries = { |
| 117 'resolve': """ | |
| 118 SELECT object_type, object_name, owner FROM ( | |
| 80 | 119 SELECT object_type, object_name, user owner, 1 priority |
| 120 FROM user_objects | |
| 121 WHERE object_name = :objName | |
| 122 UNION ALL | |
| 123 SELECT ao.object_type, ao.object_name, ao.owner, 2 priority | |
| 124 FROM all_objects ao | |
| 125 JOIN user_synonyms us ON (us.table_owner = ao.owner AND us.table_name = ao.object_name) | |
| 126 WHERE us.synonym_name = :objName | |
| 127 AND ao.object_type != 'SYNONYM' | |
| 128 UNION ALL | |
| 129 SELECT ao.object_type, ao.object_name, ao.owner, 3 priority | |
| 130 FROM all_objects ao | |
| 131 JOIN all_synonyms asyn ON (asyn.table_owner = ao.owner AND asyn.table_name = ao.object_name) | |
| 132 WHERE asyn.synonym_name = :objName | |
| 133 AND ao.object_type != 'SYNONYM' | |
| 134 AND asyn.owner = 'PUBLIC' | |
| 135 UNION ALL | |
| 136 SELECT 'DIRECTORY' object_type, dir.directory_name, dir.owner, 6 priority | |
| 137 FROM all_directories dir | |
| 138 WHERE dir.directory_name = :objName | |
| 139 UNION ALL | |
| 140 SELECT 'DATABASE LINK' object_type, db_link, owner, 7 priority | |
| 141 FROM all_db_links dbl | |
| 142 WHERE dbl.db_link = :objName | |
| 0 | 143 ) ORDER BY priority ASC""", |
| 144 'tabComments': """ | |
| 145 SELECT comments | |
| 146 FROM all_tab_comments | |
| 147 WHERE owner = :owner | |
| 148 AND table_name = :table_name""", | |
| 149 'colComments': """ | |
| 150 atc.column_name, | |
| 80 | 151 acc.comments |
| 0 | 152 FROM all_tab_columns atc |
| 153 JOIN all_col_comments acc ON (atc.owner = acc.owner and atc.table_name = acc.table_name and atc.column_name = acc.column_name) | |
| 154 WHERE atc.table_name = :object_name | |
| 155 AND atc.owner = :owner | |
| 156 ORDER BY atc.column_id;""", | |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
157 #thanks to Senora.pm for "refs" |
|
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
158 'refs': """ |
| 80 | 159 NULL referenced_by, |
| 160 c2.table_name references, | |
| 161 c1.constraint_name constraint | |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
162 FROM |
| 80 | 163 user_constraints c1, |
| 164 user_constraints c2 | |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
165 WHERE |
| 80 | 166 c1.table_name = :object_name |
| 167 and c1.constraint_type ='R' | |
| 168 and c1.r_constraint_name = c2.constraint_name | |
| 169 and c1.r_owner = c2.owner | |
| 170 and c1.owner = :owner | |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
171 UNION |
|
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
172 SELECT c1.table_name referenced_by, |
| 80 | 173 NULL references, |
| 174 c1.constraint_name constraint | |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
175 FROM |
| 80 | 176 user_constraints c1, |
| 177 user_constraints c2 | |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
178 WHERE |
| 80 | 179 c2.table_name = :object_name |
| 180 and c1.constraint_type ='R' | |
| 181 and c1.r_constraint_name = c2.constraint_name | |
| 182 and c1.r_owner = c2.owner | |
| 183 and c1.owner = :owner | |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
184 """ |
| 0 | 185 } |
| 186 | |
|
39
5d29e6a21c6f
ready to package cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
34
diff
changeset
|
187 import sys, os, re, sqlpython, cx_Oracle, pyparsing |
| 81 | 188 from cmd2 import Cmd, make_option, options |
| 0 | 189 |
| 190 if float(sys.version[:3]) < 2.3: | |
| 191 def enumerate(lst): | |
| 192 return zip(range(len(lst)), lst) | |
| 80 | 193 |
| 0 | 194 class SoftwareSearcher(object): |
| 195 def __init__(self, softwareList, purpose): | |
| 196 self.softwareList = softwareList | |
| 197 self.purpose = purpose | |
| 198 self.software = None | |
| 199 def invoke(self, *args): | |
| 200 if not self.software: | |
| 201 (self.software, self.invokeString) = self.find() | |
| 202 argTuple = tuple([self.software] + list(args)) | |
| 203 os.system(self.invokeString % argTuple) | |
| 204 def find(self): | |
| 205 if self.purpose == 'text editor': | |
| 206 software = os.environ.get('EDITOR') | |
| 207 if software: | |
| 208 return (software, '%s %s') | |
| 209 for (n, (software, invokeString)) in enumerate(self.softwareList): | |
| 210 if os.path.exists(software): | |
| 211 if n > (len(self.softwareList) * 0.7): | |
| 212 print """ | |
| 80 | 213 |
| 214 Using %s. Note that there are better options available for %s, | |
| 215 but %s couldn't find a better one in your PATH. | |
| 216 Feel free to open up %s | |
| 217 and customize it to find your favorite %s program. | |
| 218 | |
| 219 """ % (software, self.purpose, __file__, __file__, self.purpose) | |
| 0 | 220 return (software, invokeString) |
| 221 stem = os.path.split(software)[1] | |
| 222 for p in os.environ['PATH'].split(os.pathsep): | |
| 223 if os.path.exists(os.sep.join([p, stem])): | |
| 224 return (stem, invokeString) | |
| 225 raise (OSError, """Could not find any %s programs. You will need to install one, | |
| 80 | 226 or customize %s to make it aware of yours. |
| 227 Looked for these programs: | |
| 228 %s""" % (self.purpose, __file__, "\n".join([s[0] for s in self.softwareList]))) | |
| 0 | 229 #v2.4: %s""" % (self.purpose, __file__, "\n".join(s[0] for s in self.softwareList))) |
| 230 | |
| 231 softwareLists = { | |
| 232 'diff/merge': [ | |
| 80 | 233 ('/usr/bin/meld',"%s %s %s"), |
| 234 ('/usr/bin/kdiff3',"%s %s %s"), | |
| 235 (r'C:\Program Files\Araxis\Araxis Merge v6.5\Merge.exe','"%s" %s %s'), | |
| 236 (r'C:\Program Files\TortoiseSVN\bin\TortoiseMerge.exe', '"%s" /base:"%s" /mine:"%s"'), | |
| 237 ('FileMerge','%s %s %s'), | |
| 238 ('kompare','%s %s %s'), | |
| 239 ('WinMerge','%s %s %s'), | |
| 240 ('xxdiff','%s %s %s'), | |
| 241 ('fldiff','%s %s %s'), | |
| 242 ('gtkdiff','%s %s %s'), | |
| 243 ('tkdiff','%s %s %s'), | |
| 244 ('gvimdiff','%s %s %s'), | |
| 245 ('diff',"%s %s %s"), | |
| 246 (r'c:\windows\system32\comp.exe',"%s %s %s")], | |
| 247 'text editor': [ | |
| 248 ('gedit', '%s %s'), | |
| 249 ('textpad', '%s %s'), | |
| 250 ('notepad.exe', '%s %s'), | |
| 251 ('pico', '%s %s'), | |
| 252 ('emacs', '%s %s'), | |
| 253 ('vim', '%s %s'), | |
| 254 ('vi', '%s %s'), | |
| 255 ('ed', '%s %s'), | |
| 256 ('edlin', '%s %s') | |
| 257 ] | |
| 0 | 258 } |
| 259 | |
| 260 diffMergeSearcher = SoftwareSearcher(softwareLists['diff/merge'],'diff/merge') | |
| 261 editSearcher = SoftwareSearcher(softwareLists['text editor'], 'text editor') | |
| 262 editor = os.environ.get('EDITOR') | |
| 263 if editor: | |
| 264 editSearcher.find = lambda: (editor, "%s %s") | |
| 265 | |
| 266 class CaselessDict(dict): | |
| 267 """dict with case-insensitive keys. | |
| 268 | |
| 269 Posted to ASPN Python Cookbook by Jeff Donner - http://aspn.activestate.com/ASPN/Cookbook/Python/Recipe/66315""" | |
| 270 def __init__(self, other=None): | |
| 271 if other: | |
| 272 # Doesn't do keyword args | |
| 273 if isinstance(other, dict): | |
| 274 for k,v in other.items(): | |
| 275 dict.__setitem__(self, k.lower(), v) | |
| 276 else: | |
| 277 for k,v in other: | |
| 278 dict.__setitem__(self, k.lower(), v) | |
| 279 def __getitem__(self, key): | |
| 280 return dict.__getitem__(self, key.lower()) | |
| 281 def __setitem__(self, key, value): | |
| 282 dict.__setitem__(self, key.lower(), value) | |
| 283 def __contains__(self, key): | |
| 284 return dict.__contains__(self, key.lower()) | |
| 285 def has_key(self, key): | |
| 286 return dict.has_key(self, key.lower()) | |
| 287 def get(self, key, def_val=None): | |
| 288 return dict.get(self, key.lower(), def_val) | |
| 289 def setdefault(self, key, def_val=None): | |
| 290 return dict.setdefault(self, key.lower(), def_val) | |
| 291 def update(self, other): | |
| 292 for k,v in other.items(): | |
| 293 dict.__setitem__(self, k.lower(), v) | |
| 294 def fromkeys(self, iterable, value=None): | |
| 295 d = CaselessDict() | |
| 296 for k in iterable: | |
| 297 dict.__setitem__(d, k.lower(), value) | |
| 298 return d | |
| 299 def pop(self, key, def_val=None): | |
| 300 return dict.pop(self, key.lower(), def_val) | |
| 301 | |
| 302 class Parser(object): | |
| 303 comment_def = "--" + pyparsing.ZeroOrMore(pyparsing.CharsNotIn("\n")) | |
| 304 def __init__(self, scanner, retainSeparator=True): | |
| 305 self.scanner = scanner | |
| 306 self.scanner.ignore(pyparsing.sglQuotedString) | |
| 307 self.scanner.ignore(pyparsing.dblQuotedString) | |
| 308 self.scanner.ignore(self.comment_def) | |
| 309 self.scanner.ignore(pyparsing.cStyleComment) | |
| 310 self.retainSeparator = retainSeparator | |
| 311 def separate(self, txt): | |
| 312 itms = [] | |
| 313 for (sqlcommand, start, end) in self.scanner.scanString(txt): | |
| 314 if sqlcommand: | |
| 315 if type(sqlcommand[0]) == pyparsing.ParseResults: | |
| 316 if self.retainSeparator: | |
| 317 itms.append("".join(sqlcommand[0])) | |
| 318 else: | |
| 319 itms.append(sqlcommand[0][0]) | |
| 320 else: | |
| 321 if sqlcommand[0]: | |
| 322 itms.append(sqlcommand[0]) | |
| 323 return itms | |
| 324 | |
| 325 bindScanner = Parser(pyparsing.Literal(':') + pyparsing.Word( pyparsing.alphanums + "_$#" )) | |
| 326 | |
| 327 def findBinds(target, existingBinds, givenBindVars = {}): | |
| 328 result = givenBindVars | |
| 329 for finding, startat, endat in bindScanner.scanner.scanString(target): | |
| 330 varname = finding[1] | |
| 331 try: | |
| 332 result[varname] = existingBinds[varname] | |
| 333 except KeyError: | |
| 334 if not givenBindVars.has_key(varname): | |
| 335 print 'Bind variable %s not defined.' % (varname) | |
| 336 return result | |
| 80 | 337 |
| 0 | 338 class sqlpyPlus(sqlpython.sqlpython): |
|
21
8b55aaa52ce9
working on load, and preserving stdin/out
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
20
diff
changeset
|
339 defaultExtension = 'sql' |
| 92 | 340 sqlpython.sqlpython.shortcuts.update({':': 'setbind', '\\': 'psql', '@': '_load'}) |
|
21
8b55aaa52ce9
working on load, and preserving stdin/out
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
20
diff
changeset
|
341 multilineCommands = '''select insert update delete tselect |
| 80 | 342 create drop alter'''.split() |
|
112
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
343 terminatorKeepingCommands = ['select'] |
|
42
05c20d6bcec4
picking up from hiatus
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
41
diff
changeset
|
344 defaultFileName = 'afiedt.buf' |
| 0 | 345 def __init__(self): |
| 346 sqlpython.sqlpython.__init__(self) | |
| 347 self.binds = CaselessDict() | |
| 348 self.sqlBuffer = [] | |
| 349 self.settable = ['maxtselctrows', 'maxfetch', 'autobind', 'failover', 'timeout'] # settables must be lowercase | |
| 350 self.stdoutBeforeSpool = sys.stdout | |
| 351 self.spoolFile = None | |
| 352 self.autobind = False | |
| 353 self.failover = False | |
| 354 def default(self, arg, do_everywhere=False): | |
| 355 sqlpython.sqlpython.default(self, arg, do_everywhere) | |
| 356 self.sqlBuffer.append(self.query) | |
| 357 | |
| 358 # overrides cmd's parseline | |
| 359 def parseline(self, line): | |
| 360 """Parse the line into a command name and a string containing | |
| 361 the arguments. Returns a tuple containing (command, args, line). | |
|
40
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
362 'command' and 'args' may be None if the line couldn't be parsed. |
| 0 | 363 Overrides cmd.cmd.parseline to accept variety of shortcuts..""" |
| 80 | 364 |
|
41
33c9bc61db66
separation surgery successful?
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
40
diff
changeset
|
365 cmd, arg, line = sqlpython.sqlpython.parseline(self, line) |
|
20
d6d64c2e3b98
shortcuts
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
16
diff
changeset
|
366 if cmd in ('select', 'sleect', 'insert', 'update', 'delete', 'describe', |
| 80 | 367 'desc', 'comments', 'pull', 'refs', 'desc', 'triggers', 'find') \ |
| 368 and not hasattr(self, 'curs'): | |
| 0 | 369 print 'Not connected.' |
| 370 return '', '', '' | |
| 371 return cmd, arg, line | |
| 92 | 372 |
| 373 do__load = Cmd.do_load | |
| 80 | 374 |
|
7
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
375 def onecmd_plus_hooks(self, line): |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
376 line = self.precmd(line) |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
377 stop = self.onecmd(line) |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
378 stop = self.postcmd(stop, line) |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
379 |
| 0 | 380 def do_shortcuts(self,arg): |
| 381 """Lists available first-character shortcuts | |
| 382 (i.e. '!dir' is equivalent to 'shell dir')""" | |
| 383 for (scchar, scto) in self.shortcuts.items(): | |
| 58 | 384 print '%s: %s' % (scchar, scto) |
| 0 | 385 |
| 386 def colnames(self): | |
| 387 return [d[0] for d in curs.description] | |
| 388 | |
| 389 def sql_format_itm(self, itm, needsquotes): | |
| 390 if itm is None: | |
| 391 return 'NULL' | |
| 392 if needsquotes: | |
| 393 return "'%s'" % str(itm) | |
| 394 return str(itm) | |
| 105 | 395 def str_or_empty(self, itm): |
| 396 if itm is None: | |
| 397 return '' | |
| 398 return str(itm) | |
| 0 | 399 def output_as_insert_statements(self): |
| 400 usequotes = [d[1] != cx_Oracle.NUMBER for d in self.curs.description] | |
| 401 def formatRow(row): | |
| 402 return ','.join(self.sql_format_itm(itm, useq) | |
| 403 for (itm, useq) in zip(row, usequotes)) | |
| 404 result = ['INSERT INTO %s (%s) VALUES (%s);' % | |
| 405 (self.tblname, ','.join(self.colnames), formatRow(row)) | |
| 406 for row in self.rows] | |
| 407 return '\n'.join(result) | |
| 80 | 408 |
| 0 | 409 def output_row_as_xml(self, row): |
| 410 result = [' <%s>\n %s\n </%s>' % | |
| 105 | 411 (colname.lower(), self.str_or_empty(itm), colname.lower()) |
| 0 | 412 for (itm, colname) in zip(row, self.colnames)] |
| 413 return '\n'.join(result) | |
| 414 def output_as_xml(self): | |
| 415 result = ['<%s>\n%s\n</%s>' % | |
| 416 (self.tblname, self.output_row_as_xml(row), self.tblname) | |
| 417 for row in self.rows] | |
| 418 return '\n'.join(result) | |
|
111
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
419 |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
420 html_template = """<html> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
421 <head> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
422 <title py:content="tblname">Table Name</title> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
423 </head> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
424 <body> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
425 <table py:attr={'id':tblname}> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
426 <tr> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
427 <th py:for="colname in colnames"> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
428 <span py:replace="colname">Column Name</span> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
429 </th> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
430 </tr> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
431 <tr py:for="row in rows"> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
432 <td py:for="itm in row"> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
433 <span py:replace="str_or_empty(itm)">Value</span> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
434 </td> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
435 </tr> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
436 </table> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
437 </body> |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
438 </html>""" |
| 0 | 439 def output_as_html_table(self): |
| 440 result = ''.join('<th>%s</th>' % c for c in self.colnames) | |
| 441 result = [' <tr>\n %s\n </tr>' % result] | |
| 442 for row in self.rows: | |
| 443 result.append(' <tr>\n %s\n </tr>' % | |
| 444 (''.join('<td>%s</td>' % | |
| 105 | 445 self.str_or_empty(itm) |
| 80 | 446 for itm in row))) |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
447 result = '''<table id="%s"> |
| 0 | 448 %s |
| 449 </table>''' % (self.tblname, '\n'.join(result)) | |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
450 return result |
|
111
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
451 |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
452 #TODO: use serious templating to make these user-tweakable |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
453 |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
454 def output_as_markup(self, genshi_template): |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
455 return None |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
456 #self.tblname, self.colnames, self.rows |
|
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
457 |
| 0 | 458 def output_as_list(self, align): |
| 459 result = [] | |
| 460 colnamelen = max(len(colname) for colname in self.colnames) + 1 | |
| 461 for (idx, row) in enumerate(self.rows): | |
| 462 result.append('\n**** Row: %d' % (idx+1)) | |
| 463 for (itm, colname) in zip(row, self.colnames): | |
| 464 if align: | |
| 465 colname = colname.ljust(colnamelen) | |
| 466 result.append('%s: %s' % (colname, itm)) | |
| 467 return '\n'.join(result) | |
| 468 | |
|
112
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
469 |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
470 rowlimit_pattern = pyparsing.Optional(pyparsing.Word(pyparsing.nums)) |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
471 terminators = [pyparsing.Literal(';') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
472 pyparsing.Literal('\\i') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
473 pyparsing.Literal('\\x') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
474 pyparsing.Literal('\\g') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
475 pyparsing.Literal('\\G') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
476 pyparsing.Literal('\\s') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
477 pyparsing.Literal('\\S') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
478 pyparsing.Literal('\\c') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
479 pyparsing.Literal('\\C') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
480 pyparsing.Literal('\\h') + rowlimit_pattern, |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
481 pyparsing.Literal('\\t') + rowlimit_pattern, |
| 113 | 482 pyparsing.LineEnd() + '/' + pyparsing.LineEnd() |
|
112
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
483 ] |
| 0 | 484 tableNameFinder = re.compile(r'from\s+([\w$#_"]+)', re.IGNORECASE | re.MULTILINE | re.DOTALL) |
| 485 def output(self, outformat, rowlimit): | |
| 486 self.tblname = self.tableNameFinder.search(self.curs.statement).group(1) | |
| 487 self.colnames = [d[0] for d in self.curs.description] | |
| 488 if outformat == '\\i': | |
| 489 result = self.output_as_insert_statements() | |
| 490 elif outformat == '\\x': | |
| 491 result = self.output_as_xml() | |
| 492 elif outformat == '\\g': | |
| 493 result = self.output_as_list(align=False) | |
| 494 elif outformat == '\\G': | |
| 495 result = self.output_as_list(align=True) | |
| 496 elif outformat in ('\\s', '\\S', '\\c', '\\C'): #csv | |
| 497 result = [] | |
| 498 if outformat in ('\\s', '\\c'): | |
| 499 result.append(','.join('"%s"' % colname for colname in self.colnames)) | |
| 500 for row in self.rows: | |
| 106 | 501 result.append(','.join('"%s"' % self.str_or_empty(itm) for itm in row)) |
| 0 | 502 result = '\n'.join(result) |
| 503 elif outformat == '\\h': | |
| 504 result = self.output_as_html_table() | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
505 elif outformat == '\\t': |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
506 rows = [self.colnames] |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
507 rows.extend(list(self.rows)) |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
508 transpr = [[rows[y][x] for y in range(len(rows))]for x in range(len(rows[0]))] # matrix transpose |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
509 newdesc = [['ROW N.'+str(y),10] for y in range(len(rows))] |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
510 for x in range(len(self.curs.description)): |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
511 if str(self.curs.description[x][1]) == "<type 'cx_Oracle.BINARY'>": # handles RAW columns |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
512 rname = transpr[x][0] |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
513 transpr[x] = map(binascii.b2a_hex, transpr[x]) |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
514 transpr[x][0] = rname |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
515 newdesc[0][0] = 'COLUMN NAME' |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
516 result = '\n' + sqlpython.pmatrix(transpr,newdesc) |
| 0 | 517 else: |
| 518 result = sqlpython.pmatrix(self.rows, self.curs.description, self.maxfetch) | |
| 519 return result | |
| 80 | 520 |
| 89 | 521 statementEndPattern = re.compile(r'(.*)(;|\\[gGhtxicCsS])\s*(\d*)\s*$', re.DOTALL | re.MULTILINE) |
| 45 | 522 # what about quote-enclosed? |
| 80 | 523 |
|
42
05c20d6bcec4
picking up from hiatus
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
41
diff
changeset
|
524 legalOracle = re.compile('[a-zA-Z_$#]') |
| 80 | 525 |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
526 def do_select(self, arg, bindVarsIn=None, override_terminator=None): |
| 0 | 527 """Fetch rows from a table. |
| 80 | 528 |
| 0 | 529 Limit the number of rows retrieved by appending |
| 530 an integer after the terminator | |
| 531 (example: SELECT * FROM mytable;10 ) | |
| 80 | 532 |
| 0 | 533 Output may be formatted by choosing an alternative terminator |
| 534 ("help terminators" for details) | |
| 535 """ | |
| 536 bindVarsIn = bindVarsIn or {} | |
|
1
8fa146b9a2d7
reworking multiline
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
0
diff
changeset
|
537 self.query = 'select ' + arg |
|
112
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
538 terminator = self.commmand_terminator_finder(self.query) |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
539 if terminator: |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
540 (self.query, terminator, dummy) = terminator |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
541 else: |
|
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
542 terminator = [';'] |
| 113 | 543 try: |
| 544 terminator, rowlimit = terminator[0], int(terminator[1]) | |
| 545 except (IndexError, ValueError): | |
|
112
5e57765073f4
worked out terminator keeping
catherine@Elli.myhome.westell.com
parents:
111
diff
changeset
|
546 terminator, rowlimit = terminator[0], 0 |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
547 if override_terminator: |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
548 terminator = override_terminator |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
549 try: |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
550 self.varsUsed = findBinds(self.query, self.binds, bindVarsIn) |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
551 self.curs.execute(self.query, self.varsUsed) |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
552 self.rows = self.curs.fetchmany(min(self.maxfetch, (rowlimit or self.maxfetch))) |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
553 self.desc = self.curs.description |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
554 self.rc = self.curs.rowcount |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
555 if self.rc > 0: |
| 48 | 556 self.stdout.write('\n%s\n' % (self.output(terminator, rowlimit))) |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
557 if self.rc == 0: |
|
47
46b7b31dc395
feedback should print, not stdout
catherine.devlin@gmail.com
parents:
46
diff
changeset
|
558 print '\nNo rows Selected.\n' |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
559 elif self.rc == 1: |
|
47
46b7b31dc395
feedback should print, not stdout
catherine.devlin@gmail.com
parents:
46
diff
changeset
|
560 print '\n1 row selected.\n' |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
561 if self.autobind: |
|
42
05c20d6bcec4
picking up from hiatus
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
41
diff
changeset
|
562 self.binds.update(dict(zip([''.join(l for l in d[0] if l.isalnum()) for d in self.desc], self.rows[0]))) |
|
05c20d6bcec4
picking up from hiatus
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
41
diff
changeset
|
563 if len(self.desc) == 1: |
|
05c20d6bcec4
picking up from hiatus
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
41
diff
changeset
|
564 self.binds['_'] = self.rows[0][0] |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
565 elif self.rc < self.maxfetch: |
|
47
46b7b31dc395
feedback should print, not stdout
catherine.devlin@gmail.com
parents:
46
diff
changeset
|
566 print '\n%d rows selected.\n' % self.rc |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
567 else: |
|
47
46b7b31dc395
feedback should print, not stdout
catherine.devlin@gmail.com
parents:
46
diff
changeset
|
568 print '\nSelected Max Num rows (%d)' % self.rc |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
569 except Exception, e: |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
570 print e |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
571 import traceback |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
572 traceback.print_exc(file=sys.stdout) |
| 0 | 573 self.sqlBuffer.append(self.query) |
| 574 | |
| 82 | 575 @options([make_option('-f', '--full', action='store_true', help='get dependent objects as well')]) |
| 81 | 576 def do_pull(self, arg, opts): |
| 577 """Displays source code.""" | |
| 80 | 578 |
|
6
f9ec5c20beb4
working on resolving more objects
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
5
diff
changeset
|
579 object_type, owner, object_name = self.resolve(arg.strip(self.terminator).upper()) |
|
79
01d578f4e6e7
prevent crash when pull unresolved
catherine@DellZilla.myhome.westell.com
parents:
78
diff
changeset
|
580 if not object_type: |
|
01d578f4e6e7
prevent crash when pull unresolved
catherine@DellZilla.myhome.westell.com
parents:
78
diff
changeset
|
581 return |
|
46
ad41f6a577f7
think terminators are working now
catherine.devlin@gmail.com
parents:
45
diff
changeset
|
582 self.stdout.write("%s %s.%s\n" % (object_type, owner, object_name)) |
| 63 | 583 self.stdout.write(str(self.curs.callfunc('DBMS_METADATA.GET_DDL', cx_Oracle.CLOB, |
| 80 | 584 [object_type, object_name, owner]))) |
| 81 | 585 if opts.full: |
|
40
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
586 for dependent_type in ('OBJECT_GRANT', 'CONSTRAINT', 'TRIGGER'): |
|
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
587 try: |
| 63 | 588 self.stdout.write(str(self.curs.callfunc('DBMS_METADATA.GET_DEPENDENT_DDL', cx_Oracle.CLOB, |
| 80 | 589 [dependent_type, object_name, owner]))) |
|
40
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
590 except cx_Oracle.DatabaseError: |
|
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
591 pass |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
592 |
| 82 | 593 @options([make_option('-i', '--insensitive', action='store_true', help='case-insensitive search')]) |
| 81 | 594 def do_find(self, arg, opts): |
| 595 """Finds argument in source code.""" | |
| 80 | 596 |
| 81 | 597 if opts.insensitive: |
|
40
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
598 searchfor = "LOWER(text)" |
|
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
599 arg = arg.lower() |
|
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
600 else: |
|
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
601 searchfor = "text" |
|
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
602 self.do_select("* from all_source where %s like '%%%s%%'" % (searchfor, arg)) |
| 80 | 603 |
| 83 | 604 @options([make_option('-a','--all',action='store_true', |
| 605 help='Describe all objects (not just my own)')]) | |
| 606 def do_describe(self, arg, opts): | |
| 0 | 607 "emulates SQL*Plus's DESCRIBE" |
| 83 | 608 |
| 609 if opts.all: | |
| 610 which_view = (', owner', 'all') | |
| 611 else: | |
| 612 which_view = ('', 'user') | |
| 80 | 613 |
| 77 | 614 if not arg: |
| 83 | 615 self.do_select("""object_name, object_type%s FROM %s_objects WHERE object_type IN ('TABLE','VIEW','INDEX') ORDER BY object_name""" % which_view) |
| 77 | 616 return |
| 0 | 617 object_type, owner, object_name = self.resolve(arg.strip(self.terminator).upper()) |
| 77 | 618 if not object_type: |
| 87 | 619 self.do_select("""object_name, object_type%s FROM %s_objects |
| 620 WHERE object_type IN ('TABLE','VIEW','INDEX') | |
| 621 AND object_name LIKE '%%%s%%' | |
| 622 ORDER BY object_name""" % | |
| 623 (which_view[0], which_view[1], arg.upper()) ) | |
| 77 | 624 return |
|
46
ad41f6a577f7
think terminators are working now
catherine.devlin@gmail.com
parents:
45
diff
changeset
|
625 self.stdout.write("%s %s.%s\n" % (object_type, owner, object_name)) |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
626 descQ = descQueries.get(object_type) |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
627 if descQ: |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
628 for q in descQ: |
|
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
629 self.do_select(q,bindVarsIn={'object_name':object_name, 'owner':owner}) |
| 0 | 630 elif object_type == 'PACKAGE': |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
631 self.curs.execute(descQueries['PackageObjects'][0], {'package_name':object_name, 'owner':owner}) |
|
40
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
632 packageContents = self.curs.fetchall() |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
633 for (packageObj_name,) in packageContents: |
|
111
289b0a472b65
print results in describe
catherine@Elli.myhome.westell.com
parents:
109
diff
changeset
|
634 self.stdout.write(packageObj_name + '\n') |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
635 self.do_select(descQueries['PackageObjArgs'][0],bindVarsIn={'package_name':object_name, 'owner':owner, 'object_name':packageObj_name}) |
| 0 | 636 do_desc = do_describe |
| 80 | 637 |
| 638 def do_deps(self, arg): | |
| 639 object_type, owner, object_name = self.resolve(arg.strip(self.terminator).upper()) | |
| 640 if object_type == 'PACKAGE BODY': | |
| 641 q = "and (type != 'PACKAGE BODY' or name != :object_name)'" | |
| 642 object_type = 'PACKAGE' | |
| 643 else: | |
| 644 q = "" | |
| 645 q = """ name, | |
| 646 type | |
| 647 from user_dependencies | |
| 648 where | |
| 649 referenced_name like :object_name | |
| 650 and referenced_type like :object_type | |
| 651 and referenced_owner like :owner | |
| 652 %s""" % (q) | |
| 653 self.do_select(q, {'object_name':object_name, 'object_type':object_type, 'owner':owner}) | |
| 654 | |
| 0 | 655 def do_comments(self, arg): |
| 656 'Prints comments on a table and its columns.' | |
| 657 object_type, owner, object_name = self.resolve(arg.strip(self.terminator).upper()) | |
| 658 if object_type: | |
| 659 self.curs.execute(queries['tabComments'],{'table_name':object_name, 'owner':owner}) | |
|
46
ad41f6a577f7
think terminators are working now
catherine.devlin@gmail.com
parents:
45
diff
changeset
|
660 self.stdout.write("%s %s.%s: %s\n" % (object_type, owner, object_name, self.curs.fetchone()[0])) |
|
5
65ae6cec71c6
expanded desc good so far
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
4
diff
changeset
|
661 self.do_select(queries['colComments'],bindVarsIn={'owner':owner, 'object_name': object_name}) |
| 0 | 662 |
| 663 def resolve(self, identifier): | |
| 664 """Checks (my objects).name, (my synonyms).name, (public synonyms).name | |
| 665 to resolve a database object's name. """ | |
| 666 parts = identifier.split('.') | |
| 667 try: | |
| 668 if len(parts) == 2: | |
| 669 owner, object_name = parts | |
| 670 self.curs.execute('SELECT object_type FROM all_objects WHERE owner = :owner AND object_name = :object_name', | |
| 671 {'owner': owner, 'object_name': object_name}) | |
| 672 object_type = self.curs.fetchone()[0] | |
| 673 elif len(parts) == 1: | |
| 674 object_name = parts[0] | |
| 675 self.curs.execute(queries['resolve'], {'objName':object_name}) | |
| 676 object_type, object_name, owner = self.curs.fetchone() | |
| 677 except TypeError: | |
| 678 print 'Could not resolve object %s.' % identifier | |
| 679 object_type, owner, object_name = '', '', '' | |
| 680 return object_type, owner, object_name | |
| 681 | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
682 def do_resolve(self, arg): |
|
46
ad41f6a577f7
think terminators are working now
catherine.devlin@gmail.com
parents:
45
diff
changeset
|
683 self.stdout.write(self.resolve(arg)+'\n') |
| 80 | 684 |
| 0 | 685 def spoolstop(self): |
| 686 if self.spoolFile: | |
| 88 | 687 self.stdout = self.stdoutBeforeSpool |
| 0 | 688 print 'Finished spooling to ', self.spoolFile.name |
| 689 self.spoolFile.close() | |
| 690 self.spoolFile = None | |
| 80 | 691 |
| 0 | 692 def do_spool(self, arg): |
| 693 """spool [filename] - begins redirecting output to FILENAME.""" | |
| 694 self.spoolstop() | |
| 695 arg = arg.strip() | |
| 696 if not arg: | |
| 697 arg = 'output.lst' | |
| 698 if arg.lower() != 'off': | |
| 699 if '.' not in arg: | |
| 700 arg = '%s.lst' % arg | |
| 701 print 'Sending output to %s (until SPOOL OFF received)' % (arg) | |
| 702 self.spoolFile = open(arg, 'w') | |
| 88 | 703 self.stdout = self.spoolFile |
| 80 | 704 |
| 0 | 705 def do_write(self, args): |
| 88 | 706 print 'Use (query) > outfilename instead.' |
| 707 return | |
| 80 | 708 |
| 0 | 709 def do_compare(self, args): |
| 710 """COMPARE query1 TO query2 - uses external tool to display differences. | |
| 80 | 711 |
| 58 | 712 Sorting is recommended to avoid false hits. |
| 713 Will attempt to use a graphical diff/merge tool like kdiff3, meld, or Araxis Merge, | |
| 714 if they are installed.""" | |
| 0 | 715 fnames = [] |
| 716 args2 = args.split(' to ') | |
| 88 | 717 if len(args2) < 2: |
| 718 print self.do_compare.__doc__ | |
| 719 return | |
| 0 | 720 for n in range(len(args2)): |
| 721 query = args2[n] | |
| 722 fnames.append('compare%s.txt' % n) | |
| 723 if query.rstrip()[-1] != self.terminator: | |
| 724 query = '%s%s' % (query, self.terminator) | |
| 88 | 725 self.onecmd_plus_hooks('%s > %s' % (query, fnames[n])) |
| 0 | 726 diffMergeSearcher.invoke(fnames[0], fnames[1]) |
| 727 | |
| 728 bufferPosPattern = re.compile('\d+') | |
| 729 rangeIndicators = ('-',':') | |
|
16
2776755a3a7e
beginning separation of cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
15
diff
changeset
|
730 |
| 0 | 731 def do_psql(self, arg): |
| 732 '''Shortcut commands emulating psql's backslash commands. | |
| 80 | 733 |
| 0 | 734 \c connect |
| 735 \d desc | |
| 736 \e edit | |
| 737 \g run | |
| 738 \h help | |
| 92 | 739 \i load |
| 0 | 740 \o spool |
| 741 \p list | |
| 77 | 742 \q quit |
| 0 | 743 \w save |
| 61 | 744 \db _dir_tablespaces |
| 745 \dd comments | |
| 746 \dn _dir_schemas | |
| 747 \dt _dir_tables | |
| 748 \dv _dir_views | |
| 749 \di _dir_indexes | |
| 0 | 750 \? help psql''' |
| 751 commands = {} | |
| 752 for c in self.do_psql.__doc__.splitlines()[2:]: | |
| 753 (abbrev, command) = c.split(None, 1) | |
| 754 commands[abbrev[1:]] = command | |
| 755 words = arg.split(None,1) | |
| 80 | 756 try: |
| 757 abbrev = words[0] | |
| 758 except IndexError: | |
| 759 return | |
| 0 | 760 try: |
| 761 args = words[1] | |
| 762 except IndexError: | |
| 763 args = '' | |
| 764 try: | |
| 77 | 765 return self.onecmd('%s %s' % (commands[abbrev], args)) |
| 0 | 766 except KeyError: |
| 61 | 767 print 'psql command \%s not yet supported.' % abbrev |
| 77 | 768 |
| 83 | 769 @options([make_option('-a','--all',action='store_true', |
| 770 help='Describe all objects (not just my own)')]) | |
| 84 | 771 def do__dir_tables(self, arg, opts): |
| 83 | 772 if opts.all: |
| 773 which_view = (', owner', 'all') | |
| 774 else: | |
| 775 which_view = ('', 'user') | |
| 776 self.do_select("""table_name, 'TABLE' as type%s FROM %s_tables WHERE table_name LIKE '%%%s%%'""" % | |
| 777 (which_view[0], which_view[1], arg.upper())) | |
| 80 | 778 |
| 83 | 779 @options([make_option('-a','--all',action='store_true', |
| 780 help='Describe all objects (not just my own)')]) | |
| 84 | 781 def do__dir_views(self, arg, opts): |
| 83 | 782 if opts.all: |
| 783 which_view = (', owner', 'all') | |
| 784 else: | |
| 785 which_view = ('', 'user') | |
| 786 self.do_select("""view_name, 'VIEW' as type%s FROM %s_views WHERE view_name LIKE '%%%s%%'""" % | |
| 787 (which_view[0], which_view[1], arg.upper())) | |
| 61 | 788 |
| 83 | 789 @options([make_option('-a','--all',action='store_true', |
| 790 help='Describe all objects (not just my own)')]) | |
| 84 | 791 def do__dir_indexes(self, arg, opts): |
| 83 | 792 if opts.all: |
| 793 which_view = (', owner', 'all') | |
| 794 else: | |
| 795 which_view = ('', 'user') | |
| 796 self.do_select("""index_name, index_type%s FROM %s_indexes WHERE index_name LIKE '%%%s%%' OR table_name LIKE '%%%s%%'""" % | |
| 797 (which_view[0], which_view[1], arg.upper(), arg.upper())) | |
| 61 | 798 |
| 799 def do__dir_tablespaces(self, arg): | |
| 800 self.do_select("""tablespace_name, file_name from dba_data_files""") | |
| 801 | |
| 802 def do__dir_schemas(self, arg): | |
| 803 self.do_select("""owner, count(*) AS objects FROM all_objects GROUP BY owner ORDER BY owner""") | |
| 80 | 804 |
| 62 | 805 def do_head(self, arg): |
| 806 nrows = 10 | |
| 807 args = arg.split() | |
| 808 if len(args) > 1: | |
| 809 for a in args: | |
| 810 if a[0] == '-': | |
| 811 try: | |
| 812 nrows = int(a[1:]) | |
| 813 args.remove(a) | |
| 814 except: | |
| 815 pass | |
| 816 arg = ' '.join(args) | |
| 817 self.do_select('* from %s;%d' % (arg, nrows)) | |
| 80 | 818 |
| 0 | 819 def do_print(self, arg): |
| 820 'print VARNAME: Show current value of bind variable VARNAME.' | |
| 821 if arg: | |
| 822 if arg[0] == ':': | |
| 823 arg = arg[1:] | |
| 824 try: | |
| 95 | 825 self.stdout.write(str(self.binds[arg])+'\n') |
| 0 | 826 except KeyError: |
|
46
ad41f6a577f7
think terminators are working now
catherine.devlin@gmail.com
parents:
45
diff
changeset
|
827 self.stdout.write('No bind variable %s\n' % arg) |
| 0 | 828 else: |
| 829 for (var, val) in self.binds.items(): | |
| 830 print ':%s = %s' % (var, val) | |
| 80 | 831 |
| 0 | 832 def do_setbind(self, arg): |
| 833 args = arg.split(None, 2) | |
| 72 | 834 if len(args) < 2: |
| 835 self.do_print(arg) | |
| 0 | 836 elif len(args) > 2 and args[1] in ('=',':='): |
| 837 var, val = args[0], args[2] | |
| 838 if val[0] == val[-1] == "'" and len(val) > 1: | |
| 80 | 839 self.binds[var] = val[1:-1] |
| 840 return | |
| 74 | 841 try: |
| 80 | 842 self.binds[var] = int(val) |
| 843 return | |
| 74 | 844 except ValueError: |
| 845 try: | |
| 80 | 846 self.binds[var] = float(val) |
| 847 return | |
| 74 | 848 except ValueError: |
| 80 | 849 try: |
| 850 self.binds[var] = self.curs.callfunc(val, []) | |
| 851 return | |
| 852 except: | |
| 853 pass | |
| 91 | 854 # TODO: when setting |
| 80 | 855 |
| 856 print 'Could not parse ', args | |
| 857 | |
| 0 | 858 def do_exec(self, arg): |
| 859 if arg[0] == ':': | |
| 860 self.do_setbind(arg[1:]) | |
| 861 else: | |
| 71 | 862 try: |
| 863 self.curs.execute('begin\n%s;end;' % arg) | |
| 864 except Exception, e: | |
| 865 print e | |
| 866 ''' | |
| 867 exec :x := 'box' | |
| 868 exec :y := sysdate | |
| 869 ''' | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
870 |
| 96 | 871 '''Works: |
| 872 exec myproc() | |
| 873 begin | |
| 874 myproc(); | |
| 875 end; | |
| 876 | |
| 877 Fails: | |
| 878 select n into :n from test; | |
| 879 :d := sysdate | |
| 880 :n := myfunc()''' | |
| 881 | |
| 882 | |
| 883 | |
| 70 | 884 def anon_plsql(self, line1): |
| 885 lines = [line1] | |
| 886 while True: | |
| 887 line = self.pseudo_raw_input(self.continuationPrompt) | |
| 888 if line.strip() == '/': | |
| 889 try: | |
| 890 self.curs.execute('\n'.join(lines)) | |
| 891 except Exception, e: | |
| 892 print e | |
| 893 return | |
| 894 lines.append(line) | |
| 80 | 895 |
| 70 | 896 def do_begin(self, arg): |
| 897 self.anon_plsql('begin ' + arg) | |
| 898 | |
| 899 def do_declare(self, arg): | |
| 900 self.anon_plsql('declare ' + arg) | |
| 80 | 901 |
| 87 | 902 #def do_create(self, arg): |
| 903 # self.anon_plsql('create ' + arg) | |
| 78 | 904 |
| 82 | 905 @options([make_option('-l', '--long', action='store_true', help='long descriptions')]) |
| 81 | 906 def do_ls(self, arg, opts): |
| 78 | 907 where = '' |
| 908 if arg: | |
| 909 where = """\nWHERE object_type || '/' || object_name | |
| 80 | 910 LIKE '%%%s%%'""" % (arg.upper().replace('*','%')) |
| 78 | 911 else: |
| 912 where = '' | |
| 913 result = [] | |
| 914 statement = '''SELECT object_type, object_name, | |
| 80 | 915 status, last_ddl_time |
| 916 FROM user_objects %s | |
| 917 ORDER BY object_type, object_name''' % (where) | |
| 78 | 918 self.curs.execute(statement) |
| 919 for (object_type, object_name, status, last_ddl_time) in self.curs.fetchall(): | |
| 81 | 920 if opts.long: |
| 78 | 921 result.append('%s\t%s\t%s/%s' % (status, last_ddl_time, object_type, object_name)) |
| 922 else: | |
| 923 result.append('%s/%s' % (object_type, object_name)) | |
| 924 self.stdout.write('\n'.join(result) + '\n') | |
| 80 | 925 |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
926 def do_cat(self, arg): |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
927 targets = arg.split() |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
928 for target in targets: |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
929 self.do_select('* from %s' % target) |
| 80 | 930 |
| 86 | 931 @options([make_option('-i', '--ignore-case', dest='ignorecase', action='store_true', help='Case-insensitive search')]) |
| 932 def do_grep(self, arg, opts): | |
| 933 """grep PATTERN TABLE - search for term in any of TABLE's fields""" | |
| 934 | |
| 935 targets = [] | |
| 936 for target in arg.split(): | |
| 937 if '*' in target: | |
| 938 self.curs.execute("SELECT owner, table_name FROM all_tables WHERE table_name LIKE '%s'" % | |
| 939 (target.upper().replace('*','%'))) | |
| 940 for row in self.curs: | |
| 941 targets.append('%s.%s' % row) | |
| 942 else: | |
| 943 targets.append(target) | |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
944 pattern = targets.pop(0) |
|
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
945 for target in targets: |
| 86 | 946 print target |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
947 target = target.rstrip(';') |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
948 sql = [] |
|
7
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
949 try: |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
950 self.curs.execute('select * from %s where 1=0' % target) |
| 86 | 951 if opts.ignorecase: |
| 952 sql = ' or '.join("LOWER(%s) LIKE '%%%s%%'" % (d[0], pattern.lower()) for d in self.curs.description) | |
| 953 else: | |
| 954 sql = ' or '.join("%s LIKE '%%%s%%'" % (d[0], pattern) for d in self.curs.description) | |
|
7
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
955 sql = '* FROM %s WHERE %s' % (target, sql) |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
956 self.do_select(sql) |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
957 except Exception, e: |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
958 print e |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
959 import traceback |
|
d44784122203
more history tweaks
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
6
diff
changeset
|
960 traceback.print_exc(file=sys.stdout) |
|
4
23c3a58d7804
about to strip out tselect
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
3
diff
changeset
|
961 |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
962 def do_refs(self, arg): |
|
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
963 object_type, owner, object_name = self.resolve(arg.strip(self.terminator).upper()) |
|
40
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
964 if object_type == 'TABLE': |
|
1fb9f7dee7d8
tearing out cmd2
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
39
diff
changeset
|
965 self.do_select(queries['refs'],bindVarsIn={'object_name':object_name, 'owner':owner}) |
|
10
2ef0e2608123
reworking pull
devlinjs@FA7CZA6N1254998.wrightpatterson.afmc.ds.af.mil
parents:
8
diff
changeset
|
966 |
| 0 | 967 def _test(): |
| 968 import doctest | |
| 969 doctest.testmod() | |
| 80 | 970 |
| 0 | 971 if __name__ == "__main__": |
| 972 "Silent return implies that all unit tests succeeded. Use -v to see details." | |
| 109 | 973 _test() |