-
Notifications
You must be signed in to change notification settings - Fork 16
/
sqx
executable file
·151 lines (131 loc) · 5.93 KB
/
sqx
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
#!/bin/env python
##################################################################################################
# Name: sqx (SQL Execute) #
# Author: Randy Johnson #
# Description: Takes a sql script file from the command line and executes it in the database. #
# If no connect string provided ORACLE_SID is used and authentication defaults to #
# sys / as sysdba. #
# #
# History: #
# #
# Date Ver. Who Change Description #
# ---------- ---- ---------------- ------------------------------------------------------------- #
# 07/06/2012 1.00 Randy Johnson Initial release. #
# 09/19/2012 2.00 Randy Johnson Cleanup, minor cosmetic changes, removed OptionParser. #
# 07/17/2015 3.00 Randy Johnson Updated for Python 2.4-3.4 compatibility. Added prompts for #
# username, password, tnsname. Added -f option. #
##################################################################################################
# --------------------------------------
# ---- Import Python Modules -----------
# --------------------------------------
#from datetime import datetime
from optparse import OptionParser
from os import environ
from os.path import basename
from os.path import isfile
from sys import argv
from sys import exit
from sys import version_info
from signal import SIGPIPE
from signal import SIG_DFL
from signal import signal
from Oracle import RunSqlplus
from Oracle import SetOracleEnv
from Oracle import ParseConnectString
# --------------------------------------
# ---- Main Program --------------------
# --------------------------------------
if (__name__ == '__main__'):
Cmd = basename(argv[0]).split('.')[0]
CmdDesc = 'SQL Execute'
Version = '3.00'
VersionDate = 'Tue Sep 15 21:02:11 CDT 2015'
DevState = 'Production'
Banner = CmdDesc + ': Release ' + Version + ' ' + DevState + '. Last updated: ' + VersionDate
Sql = ''
SqlHeader = '/***** ' + CmdDesc.upper() + ' *****/'
ErrChk = False
InStr = ''
ConnStr = ''
# For handling termination in stdout pipe; ex: when you run: oerrdump | head
signal(SIGPIPE, SIG_DFL)
Usage = Cmd + " [connect_string] [-f script] [-s] [-v] [-h]"
Usage += "\n\nExample:"
Usage += "\n " + Cmd
Usage += "\n " + Cmd + " -f myscript.sql"
Usage += "\n " + Cmd + " scott -f myscript.sql"
Usage += "\n " + Cmd + " scott@MYDB -f myscript.sql"
Usage += "\n " + Cmd + " scott/tiger -f myscript.sql"
Usage += "\n " + Cmd + " scott/tiger@MYDB -f myscript.sql"
Usage += "\n " + Cmd + " scott/tiger@dbhost.com:1521/MYDB -f myscript.sql"
ArgParser = OptionParser(usage=Usage)
ArgParser.add_option('-f', dest='InputFile', default='', type=str, help="SQL script to execute.")
ArgParser.add_option('--s', dest='Show', action='store_true', default=False, help="print SQL query.")
ArgParser.add_option('--v', dest='ShowVer', action='store_true', default=False, help="print version info.")
# Parse command line arguments
Options, args = ArgParser.parse_args()
argc = len(argv)
InputFile = Options.InputFile
Show = Options.Show
ShowVer = Options.ShowVer
if (ShowVer):
print('\n%s' % Banner)
exit()
if (not InputFile):
if (version_info[0] >= 3):
InputFile = input('\nEnter script: ')
else:
InputFile = raw_input('\nEnter script: ')
if (not InputFile):
print('\nScript file is required.')
print("\n" + Usage)
exit(1)
if (not isfile(InputFile)):
if (isfile(InputFile + '.sql')):
InputFile += '.sql'
else:
if (InputFile):
print('\nScript not found: %s\n' % InputFile)
else:
print('\nScript not found.\n')
print(Usage)
exit(1)
hSql = open(InputFile, 'r')
try:
hSql = open(InputFile, 'r')
Sql = hSql.read()
except:
print('\n>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<')
print('Cannot open script for read: %s' % InputFile)
print('\n>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<')
exit(1)
Sql = Sql.strip()
if(Show):
print('-----------cut-----------cut-----------cut-----------cut-----------cut-----------')
print(Sql)
print('-----------cut-----------cut-----------cut-----------cut-----------cut-----------')
exit()
# Check/setup the Oracle environment
if (not('ORACLE_SID' in list(environ.keys()))):
print('ORACLE_SID is required.')
exit(1)
else:
# Set the ORACLE_HOME just in case it isn't set already.
if (not('ORACLE_HOME' in list(environ.keys()))):
(OracleSid, OracleHome) = SetOracleEnv(environ['ORACLE_SID'])
# Parse the connect string if any, prompt for username, password if needed.
if (len(args) > 0 and Show == False):
InStr = args[0]
ConnStr = ParseConnectString(InStr)
# Execute the report
if (ConnStr != ''):
(Stdout) = RunSqlplus(Sql, ErrChk, ConnStr)
else:
(Stdout) = RunSqlplus(Sql, ErrChk)
# Print the report
if (Stdout != ''):
print('\n%s' % Stdout)
exit(0)
# --------------------------------------
# ---- End Main Program ----------------
# --------------------------------------