根据博友的PySQLPLUS脚本改写,增加Readline模块,解决SQLPLUS命令行内难以编辑和回退的问题。
PySQLPLUS源:[url]http://blog.csdn.net/yzsind/article/details/6693160[/url]
PySQLPLUS源:[url]http://blog.csdn.net/yzsind/article/details/6693160[/url]
#!/bin/env python2.7
# encode : utf-8
import cx_Oracle
import os
import sys
import readline
os.environ['NLS_LANG'] = ".AL32UTF8";
histfile = os.path.join(os.path.expanduser("~"), ".pyhist")
try:
readline.read_history_file(histfile)
except IOError:
pass
import atexit
atexit.register(readline.write_history_file, histfile)
fetchsize = 50;
print("------------Welcome To Python Sqlplus ----------------------");
print("| Version : 0.1");
print("| Author : MKing");
print("| Blog : http://blog.csdn.net/yzsind");
print("| Sina weibo : http://weibo.com/yzsind");
print("| Release Date: 2011-08-08");
print("| Login Example1:username/password@tnsname");
print("| Login Example2:username/password@host:port/dbname");
print("| Input exit to Quit");
print("-----------------------------------------------------------");
print("");
def getConnect(loginstr):
try:
conn = cx_Oracle.connect(loginstr);
print("Database version:", conn.version);
print("Connected.");
return conn;
except cx_Oracle.InterfaceError as exc:
error, = exc.args
print(exc);
except cx_Oracle.DatabaseError as exc:
error, = exc.args
print(error.message);
return None;
def getcolformatstr(coldef):
if coldef[1] == cx_Oracle.NUMBER:
formatstr = '%12s';
else:
if coldef[2] <= 32:
formatstr = '%-' + str(coldef[2]) + 's';
else:
formatstr = '%-32s';
return formatstr
def main(loginstr):
for i in range(3):
try:
if (loginstr == None or len(loginstr) == 0):
loginstr = raw_input("login>").strip();
continue;
conn = getConnect(loginstr);
if conn == None:
print("Connection failure!");
loginstr = None;
continue;
else:
break;
except KeyboardInterrupt:
print("^C");
continue;
except EOFError:
print("Bye");
return;
promptstr = conn.dsn;
executor(conn, promptstr);
def executor(conn, promptstr):
while True:
sqlstr = "";
try:
sqlstrline = raw_input(promptstr + ">").strip();
if sqlstrline == "" :
continue;
elif sqlstrline.lower() in ["exit", "exit;"]:
print("...bye...");
exit();
elif sqlstrline[0:7].lower() == "connect" :
conn = getConnect(sqlstrline[8:]);
elif sqlstrline.lower() in ["disconnect", "disconnect;"] :
conn.close();
print("Connection closed.");
elif sqlstrline[0:4].lower() == "host" :
os.system(sqlstrline[4:])
else:
sqlstr = sqlstr + sqlstrline + '\n';
while sqlstrline[-1] != ";" :
sqlstrline = raw_input().strip();
sqlstr = sqlstr + sqlstrline + '\n';
sqlstr = sqlstr[0:len(sqlstr) - 2]
try:
cursor = conn.cursor();
cursor.execute(sqlstr);
if sqlstr[0:6].lower() == "select" :
cols = []
for col in cursor.description:
print(getcolformatstr(col) % (col[0])),
print('');
for col in cursor.description:
if col[1] == cx_Oracle.NUMBER:
print('-' * 12), ;
else:
if col[2] <= 32:
print('-' * col[2]), ;
else:
print('-' * 32), ;
print('');
recs = cursor.fetchmany(fetchsize);
while len(recs) > 0:
for row in recs:
for i in range(len(row)):
if row[i] != None:
print(getcolformatstr(cursor.description[i]) % row[i]), ;
else:
print(getcolformatstr(cursor.description[i]) % ''), ;
print('')
recs = cursor.fetchmany(fetchsize);
print(str(cursor.rowcount) + " rows selected.");
elif sqlstr[0:6].lower() == "insert" :
print(str(cursor.rowcount) + " rows inserted.");
elif sqlstr[0:6].lower() == "update" :
print(str(cursor.rowcount) + " rows updated.");
elif sqlstr[0:6].lower() == "delete" :
print(str(cursor.rowcount) + " rows deleted.");
elif sqlstr[0:5].lower() == "merge" :
print(str(cursor.rowcount) + " rows merged.");
elif sqlstr[0:6].lower() == "commit" :
print("Commit complete.");
elif sqlstr[0:6].lower() == "rollback" :
print("Rollback complete.");
else :
print("sql execute complete.");
except cx_Oracle.InterfaceError as exc:
error, = exc.args
print(exc);
except cx_Oracle.DatabaseError as exc:
error, = exc.args
print(error.message);
except KeyboardInterrupt:
print("^C");
continue;
except EOFError:
print("Bye");
return;
#########################################################################
if __name__ == '__main__':
args = sys.argv[1:];
loginstr = None;
if (len(args) > 0):
loginstr = args[0];
main(loginstr);