annotate sqlpyPlus.py @ 14:1300a997d319

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