快捷導(dǎo)航

復(fù)制粘貼功能的Python程序

更新時(shí)間：2008年04月04日 19:19:03 作者：

接下來(lái)，由于我覺(jué)得手動(dòng)復(fù)制粘貼這29000條插入語(yǔ)句實(shí)在是太痛苦了，所以打算用Python來(lái)完成這項(xiàng)工作。這是我第一次自己動(dòng)手寫(xiě)Python代碼，感覺(jué)還挺順利的。

今天因?yàn)榻oBeauBeau提供的抽獎(jiǎng)號(hào)碼做SQL文件，一開(kāi)始收到ZIP文件解開(kāi)壓縮之后被嚇到了——29個(gè)CSV文件，每個(gè)文件保存了1000個(gè)獎(jiǎng)券ID和號(hào)碼-_-!

照上次一樣，打開(kāi)每個(gè)CSV文件做先做單獨(dú)的SQL文件，每個(gè)SQL中有1000條插入語(yǔ)句，隨后將29個(gè)文件的所有SQL語(yǔ)句都復(fù)制粘貼到同一個(gè)總的SQL文件中。

CSV文件中的結(jié)構(gòu)是“ID,NUMBER”的結(jié)構(gòu)，其中ID是7位數(shù)字，NUMBER是11位數(shù)字。這樣用正則式來(lái)進(jìn)行捕捉的時(shí)候就比較方便了，在Eclipse的查找/替換功能中所使用的正則式就是“(\d{7}),(\d{11})”，進(jìn)行替換的文本內(nèi)容就是“INSERT INTO cards VALUES ('$1','$2',now());”。使用這種方法對(duì)29個(gè)CSV文件中的內(nèi)容進(jìn)行替換。

所有代碼如下：

復(fù)制代碼代碼如下:

import sys, os  
def readFile(filename):  
    file=open(filename, "r")  
    s=file.read().strip()  
    file.close()  
    return s  

def writeFile(filename, files):  
    content=[]  
    for f in files:  
        print "reading file ' %s ' " % f  
        s=readFile(f)  
        print "read file ' %s ' completed" % f  
        content.append(s)  
    print "writing file ' %s ' " % filename  
    file=open(filename, "w")  
    file.write("\n/*-----This is a seperating line.-----*/\n".join(content))  
    file.close()  
    print "write file ' %s ' completed" % filename  


filters=['.txt']  
fullpath=os.getcwd();  

print "opening directory: ' %s ' " % fullpath  

sys.path.append(fullpath)  
files = os.listdir(fullpath)  
files =[f for f in files if os.path.splitext(f)[1].lower() in filters]  
writeFile("beaunet_be_card.sql", files) 