Python調(diào)用SQLPlus來操作和解析Oracle數(shù)據(jù)庫的方法

這篇文章主要介紹了Python調(diào)用SQLPlus來操作和解析Oracle數(shù)據(jù)庫的方法,這樣用SQL*Plus方式來分析Oracle中的數(shù)據(jù)就變得十分方便,需要的朋友可以參考下

先來看一個(gè)簡單的利用python調(diào)用sqlplus來輸出結(jié)果的例子：    
import os
import sys
from subprocess import Popen, PIPE
 
sql = """
set linesize 400
col owner for a10
col object_name for a30
 
select owner, object_name
 from dba_objects
 where rownum<=10;
"""
 
proc = Popen(["sqlplus", "-S", "/", "as", "sysdba"], stdout=PIPE, stdin=PIPE, stderr=PIPE)
proc.stdin.write(sql)
(out, err) = proc.communicate()
 
if proc.returncode != 0:
  print err
  sys.exit(proc.returncode)
else:
  print out

用Python查詢Oracle，當(dāng)然最好用cx_Oracle庫，但有時(shí)候受到種種限制，不能安裝Python第三方庫，就得利用現(xiàn)有資源，硬著頭皮上了。

用Python調(diào)用SqlPlus查詢Oracle，首先要知道SqlPlus返回結(jié)果是什么樣的：
    
（這是空行）
Number    Name    Address
------------ ----------- ------------------
1001     張三     南京路
1002     李四     上海路

第1行是空行，第2行是字段名稱，第3行都是橫杠，有空格隔開，第4行開始是查詢到的結(jié)果。

在查詢結(jié)果規(guī)整的情況下，根據(jù)第3行可以很清晰的看到結(jié)構(gòu)，用Python解析起來也比較方便。但是，如果一張表字段特別多，記錄數(shù)也相當(dāng)多，那么默認(rèn)情況下調(diào)用SqlPlus查詢出的結(jié)果會(huì)比較亂，這就需要在調(diào)用查詢之前做一些設(shè)定，比如：    
set linesize 32767
set pagesize 9999
set term off verify off feedback off tab off
set numwidth 40

這樣的調(diào)用查詢結(jié)果就比較規(guī)整了。接下來就是用強(qiáng)大的Python來解析查詢結(jié)果。

這里封裝了一個(gè)函數(shù)，可以根據(jù)傳入的SQL語句查詢并解析結(jié)果，將每行結(jié)果存到列表中，列表中的每個(gè)元素是一個(gè)字段名稱與值的映射。    
#!/usr/bin/python
#coding=UTF-8
 
'''
@author: 雙子座@開源中國
@summary: 通過SqlPlus查詢Oracles數(shù)據(jù)庫
'''
 
import os;
 
os.environ['NLS_LANG'] = 'AMERICAN_AMERICA.AL32UTF8'
gStrConnection = 'username/password@10.123.5.123:1521/ora11g'
 
#解析SqlPlus的查詢結(jié)果，返回列表
def parseQueryResult(listQueryResult):
  listResult = []
  #如果少于4行，說明查詢結(jié)果為空
  if len(listQueryResult) < 4:
    return listResult
  #第0行是空行，第1行可以獲取字段名稱，第2行可獲取SQLPlus原始結(jié)果中每列寬度，第3行開始是真正輸出
  # 1 解析第2行，取得每列寬度，放在列表中
  listStrTmp = listQueryResult[2].split(' ')
  listIntWidth = []
  for oneStr in listStrTmp:
    listIntWidth.append(len(oneStr))
  # 2 解析第1行，取得字段名稱放在列表中
  listStrFieldName = []
  iLastIndex = 0
  lineFieldNames = listQueryResult[1]
  for iWidth in listIntWidth:
    #截取[iLastIndex, iLastIndex+iWidth)之間的字符串
    strFieldName = lineFieldNames[iLastIndex:iLastIndex + iWidth]
    strFieldName = strFieldName.strip() #去除兩端空白符
    listStrFieldName.append(strFieldName)
    iLastIndex = iLastIndex + iWidth + 1
  # 3 第3行開始，解析結(jié)果，并建立映射，存儲到列表中
  for i in range(3, len(listQueryResult)):
    oneLiseResult = unicode(listQueryResult[i], 'UTF-8')
    fieldMap = {}
    iLastIndex = 0
    for j in range(len(listIntWidth)):
      strFieldValue = oneLiseResult[iLastIndex:iLastIndex + listIntWidth[j]]
      strFieldValue = strFieldValue.strip()
      fieldMap[listStrFieldName[j]] = strFieldValue
      iLastIndex = iLastIndex + listIntWidth[j] + 1
    listResult.append(fieldMap)
  return listResult
 
def QueryBySqlPlus(sqlCommand):
  global gStrConnection
  #構(gòu)造查詢命令
  strCommand = 'sqlplus -S %s <<!\n' % gStrConnection
  strCommand = strCommand + 'set linesize 32767\n'
  strCommand = strCommand + 'set pagesize 9999\n'
  strCommand = strCommand + 'set term off verify off feedback off tab off \n'
  strCommand = strCommand + 'set numwidth 40\n'
  strCommand = strCommand + sqlCommand + '\n'
  #調(diào)用系統(tǒng)命令收集結(jié)果
  result = os.popen(strCommand)
  list = []
  for line in result:
    list.append(line)
  return parseQueryResult(list)
其中os.environ['NLS_LANG']的值來自    
select userenv['language'] from dual;
在調(diào)用的時(shí)候，只要類似：    
listResult = QueryBySqlPlus('select * from studentinfo')
然后就可以用循環(huán)打印出結(jié)果了。