csv、excel導入oracle

hcsv導入oracle

    
    
    
            

     
     
     
             

     
     
     
             
#coding:gbk
     
     
     
             

     
     
     
             
import csv 
     
     
     
             
import cx_Oracle
     
     
     
             

     
     
     
             
fields = []
     
     
     
             
data = []
     
     
     
             
table_name = 'ygl_test3'
     
     
     
             
file_name = '清單.csv'
     
     
     
             

     
     
     
             
with open(file_name, 'rb') as f:
     
     
     
             
 reader = csv.reader(f)
     
     
     
             
 contents = [i for i in reader]
     
     
     
             

     
     
     
             
title = contents[0]
     
     
     
             
data = contents[1:]
     
     
     
             

     
     
     
             

     
     
     
             
conn = cx_Oracle.connect('user/passwd@cd')
     
     
     
             
cursor = conn.cursor()
     
     
     
             

     
     
     
             
#生成create table語句中字段名字符串
     
     
     
             
for i in title:
     
     
     
             
 fields.append(i+' varchar2(200)')
     
     
     
             
fields_str = ', '.join(fields)
     
     
     
             
sql = 'create table %s (%s)' % (table_name, fields_str)
     
     
     
             

     
     
     
             
print sql
     
     
     
             
#根據excel列名創建數據表的字段
     
     
     
             
cursor.execute(sql)
     
     
     
             

     
     
     
             
#生成executemany語句中的:1,:2字段參數格式
     
     
     
             
a = [':%s' %i for i in range(len(title)+1)]
     
     
     
             
value= ','.join(a[1:])
     
     
     
             
sql = 'insert into %s values(%s)' %(table_name, value)
     
     
     
             

     
     
     
             
print sql
     
     
     
             
#把所有字段插入數據值
     
     
     
             
cursor.prepare(sql)
     
     
     
             
cursor.executemany(None, data)
     
     
     
             

     
     
     
             
cursor.close()
     
     
     
             
conn.commit()
     
     
     
             
conn.close()
    
    
    
            

升級版：

    
    
    
            

     
     
     
             
#coding:gbk
     
     
     
             

     
     
     
             
import cx_Oracle
     
     
     
             
import csv
     
     
     
             
import xlrd
     
     
     
             

     
     
     
             
class ImportOracle(object):
     
     
     
             
 #工廠模式初始化導入函數inoracle，用作讀取文件數據
     
     
     
             
 def inoracle(self):
     
     
     
             
 pass
     
     
     
             
 #oracle 連接和寫入函數
     
     
     
             
 def ConnOracle(self):
     
     
     
             
 conn = cx_Oracle.connect('user/passwd@cd')
     
     
     
             
 cursor = conn.cursor()
     
     
     
             

     
     
     
             

     
     
     
             
 #以數字開頭的字段加個字符a
     
     
     
             

     
     
     
             
 self.title = [i if i[0].isdigit() == False else 'a'+i for i in self.title ]
     
     
     
             
 fields = [i+' varchar2(200)' for i in self.title]
     
     
     
             
 fields_str = ', '.join(fields)
     
     
     
             
 sql = 'create table %s (%s)' % (self.table_name, fields_str)
     
     
     
             
 print sql
     
     
     
             
 cursor.execute(sql)
     
     
     
             

     
     
     
             
 a = [':%s' %i for i in range(len(self.title)+1)]
     
     
     
             
 value= ','.join(a[1:])
     
     
     
             
 sql = 'insert into %s values(%s)' %(self.table_name, value)
     
     
     
             
 print sql
     
     
     
             

     
     
     
             
 cursor.prepare(sql)
     
     
     
             
 cursor.executemany(None, self.data)
     
     
     
             

     
     
     
             
 cursor.close()
     
     
     
             
 conn.commit()
     
     
     
             
 conn.close()
     
     
     
             

     
     
     
             

     
     
     
             
#ImportOracle的子類，用作導入csv文件的類
     
     
     
             
class ImportOracleCsv(ImportOracle):
     
     
     
             
 #重構父類的inoracle函數，用作從csv讀取數據，返回標題和內容
     
     
     
             
 def inoracle(self):
     
     
     
             
 with open(self.filename, 'rb') as f:
     
     
     
             
 reader = csv.reader(f)
     
     
     
             
 contents = [i for i in reader]
     
     
     
             

     
     
     
             
 title = contents[0]
     
     
     
             
 data = contents[1:]
     
     
     
             

     
     
     
             
 return (title, data)
     
     
     
             
#ImportOracle的子類，用作導入excel文件的類
     
     
     
             
class ImportOracleExcel(ImportOracle):
     
     
     
             
 #重構父類的inoracle函數，用作從EXCEL文件讀取數據，返回標題和內容
     
     
     
             
 def inoracle(self):
     
     
     
             
 wb = xlrd.open_workbook(self.filename)
     
     
     
             
 sheet1 = wb.sheet_by_index(0)
     
     
     
             

     
     
     
             
 title = sheet1.row_values(0)
     
     
     
             
 data = [sheet1.row_values(row) for row in range(1, sheet1.nrows)]
     
     
     
             
 return (title, data)
     
     
     
             
#ImportOracle的子類，在文件類型錯的情況返回
     
     
     
             
class ImportError(ImportOracle):
     
     
     
             
 def inoracle(self):
     
     
     
             
 print 'Undefine file type'
     
     
     
             
 return 0 
     
     
     
             

     
     
     
             
#工廠選擇類，csv文件返回ImportOracleCsv類，EXCEL文件返回ImportOracleExcel類
     
     
     
             
class ChooseFactory(object):
     
     
     
             
 choose = {}
     
     
     
             
 choose['csv'] = ImportOracleCsv()
     
     
     
             
 choose['xlsx'] = ImportOracleExcel()
     
     
     
             
 choose['xls'] = ImportOracleExcel()
     
     
     
             

     
     
     
             
 def choosefile(self, ch):
     
     
     
             
 if ch in self.choose:
     
     
     
             
 op = self.choose[ch]
     
     
     
             
 else:
     
     
     
             
 op = ImportError()
     
     
     
             

     
     
     
             
 return op
     
     
     
             

     
     
     
             
if __name__ =="__main__":
     
     
     
             
 #定義文件名和數據庫表名
     
     
     
             
 file_name = '清單.xlsx'
     
     
     
             
 table_name= 'ygl_test'
     
     
     
             

     
     
     
             
 #獲取文件類型
     
     
     
             
 op = file_name.split('.')[-1]
     
     
     
             

     
     
     
             
 factory = ChooseFactory()
     
     
     
             
 #選擇工廠類處理，cal被賦值為處理的具體類
     
     
     
             
 cal = factory.choosefile(op)
     
     
     
             
 #設置類屬性(文件名)
     
     
     
             
 cal.filename = file_name
     
     
     
             
 #根據函數返回值設置類屬性(標題，內容)
     
     
     
             
 (cal.title, cal.data) = cal.inoracle()
     
     
     
             
 #設置類屬性(表名)
     
     
     
             
 cal.table_name = table_name
     
     
     
             
 #調用類的導入數據庫函數
     
     
     
             
 cal.ConnOracle()
    
    
    
            

來自為知筆記(Wiz)

附件列表