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