最近遇到一个需求,就是要把SQLite数据中没有存储的文件名的文件删除掉,想来想去还是决定用python。所以也就花了一天半的时间学习了下,随手写了个小例子。
直接上代码
要用到的头文件包
#coding=utf-8
#!/usr/bin/python
#!/usr/bin/env python
import os
import shutil
import sqlite3
定义记录变量
#记录所文件数
sumCount=0;
#记录留存文件数
count=0;
#记录删除文件数
delCount=0;
#定义存储遍历所有文件数组
delList = []
#文件存储路径
delDir = "/Users/liudengtan/Desktop/testFile/"
#获取路径下所有文件
delList = os.listdir(delDir)
#打开连接数据库
conn = sqlite3.connect('images.db')
print "开始处理...";
把文件目录下的所有文件与数据库的存的对比,如果文件在数据中,就存留,否则删除文件
#遍历
for f in delList:
#获取到文件路径
filePath = os.path.join(delDir, f)
if os.path.isfile(filePath):
sumCount=sumCount+1
#将文件全路径中存储路径替换,只留文件名
fileName=filePath.replace(delDir,'')
#数据库查看当前文件名是否存在
cursor = conn.execute("SELECT image FROM '表名' where image=(?)",(fileName))
res = cursor.fetchall()
#条件判断>0文件存在
if len(res) > 0:
count = count + 1;
else:#文件不存在将其删除
if os.path.isfile(delDir + fileName):
#删除文件操作
os.remove(delDir + fileName)
print delDir + fileName + " 删除!"
delCount = delCount + 1;
#关闭数据库
conn.close()
print "处里结束:";
print "所有文件总数 : ",sumCount;
print "删除文件数 : ",delCount;