Python模拟实现Oracle的sqlplus工具

团队计划开发数据库服务平台，需要用到一些服务器的脚本开发，为了了解python，给自己定了一个模拟sqlplus的小需求，然后去实现。个人体会：python开发快捷，集成很多常用的公共包，对常用数据结构使用很方便，最大的缺点是版本较多，新版本不向前兼容，对AIX及HP-UNIX不太常用的OS也缺少直接支持。
以下是工具演示：

以下是源代码：以下代码用的python版本是python2.7 http://www.python.org/
需要安装cx_Oracle开发包（python访问ORACLE用的） http://cx-oracle.sourceforge.net/

import cx_Oracle
import os
import sys
os.environ["NLS_LANG"] ="AMERICAN_AMERICA.ZHS16GBK";
connectresult=0;
promptstr="";
fetchsize=50;
#conn = cx_Oracle.connect（"yzs/yzs@mydb"）;
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）:
global connectresult
global promptstr
try:
connectresult=0;
promptstr="";
conn= cx_Oracle.connect（loginstr）;
promptstr=conn.username+"@"+conn.dsn;
print（"Database version:",conn.version）;
print（"Connected."）;
connectresult=1;
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）;
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
#########################################################################
while 1:
try:
loginstr=raw_input（"login>"）.strip（）;
if loginstr=="" :
continue;
elif loginstr in ["exit","exit;"]:
print（"...bye..."）;
exit（）;
conn = getConnect（loginstr）;
if connectresult==1:
break;
except KeyboardInterrupt:
print（"^C"）;
continue;
while 1:
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+" ";
while sqlstrline[-1]！=";" :
sqlstrline=raw_input（）.strip（）;
sqlstr=sqlstr+sqlstrline+" ";
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;

特别说明：纯属个人学习python的代码，未考虑扩展性与性能，未经过专业测试，不建议拿到实际工作中当SQLPLUS用。如何让Oracle索引不起作用（不可见）Linux下卸载Oracle数据库相关资讯 Oracle教程

Oracle中纯数字的varchar2类型和（07/29/2015 07:20:43）
Oracle教程：Oracle中查看DBLink密（07/29/2015 07:16:55）
[Oracle] SQL*Loader 详细使用教程（08/11/2013 21:30:36）

Oracle教程：Oracle中kill死锁进程（07/29/2015 07:18:28）
Oracle教程：ORA-25153 临时表空间（07/29/2015 07:13:37）
Oracle教程之管理安全和资源（04/08/2013 11:39:32）

本文评论查看全部评论（0）