這script主要是分類pdf用的,從謎之網站下載的 pdf 檔有規則,所以就可以寫code來自動分類。寫script時發現有shutil這個很方便的模組,可以用來複製、搬移檔案、目錄等,相當的方便,原本還以為要自己硬幹咧~
[python]#!/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()
[/python]