#!python3

import os
from PIL import Image

#獲取文件名列表
imgList = os.listdir("./")

##函數(shù)：獲取文件的后綴長(zhǎng)度(例如 jpg 和 jpeg 的長(zhǎng)度分別為 3 和 4 )
#def typeNameLengthCount(name):
#?? ?count = 0
#?? ?for i in range(1, len(name)):
#?? ??? ?if(name[-i] == '.'):
#?? ??? ??? ?break
#?? ??? ?else:
#?? ??? ??? ?count += 1
#?? ?return count
#修改：上面這是原來(lái)的代碼，后來(lái)想起os庫(kù)有現(xiàn)成的分離文件名和后綴名的函數(shù)，于是此自定義函數(shù)棄用

#檢查 ./SourceImage 目錄是否存在，不存在則創(chuàng)建它
if (not (os.path.exists("./SourceImage"))):
?? ?os.mkdir("./SourceImage")

#開(kāi)始遍歷文件
for name in imgList:
?? ?try:
?? ??? ?img = Image.open(name) #嘗試打開(kāi)圖片，如果不是可以打開(kāi)的類(lèi)型，則忽略（因?yàn)樾枰雎源a文件本身及后面創(chuàng)建的SourceImage文件夾）
?? ??? ?
?? ??? ?#type_length = typeNameLengthCount(name)
?? ??? ?#file_type = name[-type_length:] #得到后綴的長(zhǎng)度
?? ??? ?#修改：用庫(kù)方法替代自定義方法：
?? ??? ?file_name, file_type = os.path.splitext(name)
?? ??? ?
?? ??? ?if file_type == 'jpg' or file_type == 'jpeg': #如果已經(jīng)是 .jpg 或 .jpeg 類(lèi)型，則跳過(guò)
?? ??? ??? ?continue
?? ??? ?else:
?? ??? ?
?? ??? ??? ?#img.save("%s.jpg"%(name[0:(-type_length-1)]), 'jpeg')
?? ??? ??? ?#修改：改用下面這行代碼：
?? ??? ??? ?img.save("%s.jpg"%(file_name), 'jpeg')
?? ??? ??? ?
?? ??? ??? ?os.rename(name, "./SourceImage/%s"%name) #移動(dòng)原圖片，此處用shutil.move( , ) 方法同樣可行, 而且這個(gè)方法可以實(shí)現(xiàn)文件在磁盤(pán)間的移動(dòng)
?? ?except IOError:
?? ??? ?print("one file was ignored")