星期四, 9月 23, 2010

自動分類pdf與shutil

這script主要是分類pdf用的,從謎之網站下載的 pdf 檔有規則,所以就可以寫code來自動分類。寫script時發現有shutil這個很方便的模組,可以用來複製、搬移檔案、目錄等,相當的方便,原本還以為要自己硬幹咧~


#!/usr/bin/env python
# -*- coding: utf-8 -*-

def main():
import glob
import os
import shutil
files = glob.glob( "*.pdf" )
for file_name in files:
parts = file_name.split('.')
index = -2
while not parts[ index ].isdigit():
index = index + 1
dir_name = " ".join( parts[ :index ] )
if not os.path.exists( dir_name ):
os.mkdir( dir_name )
try:
print( "Move '%s' to '%s'." %
( file_name, os.path.join( dir_name, file_name ) ) )
shutil.move( file_name,
os.path.join( dir_name, file_name ) )
except shutil.Error, e:
print( e )

if __name__ == "__main__":
main()

沒有留言: