yu/photo-classifier
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: remove ppt files and add skip folders

Browse Source
This commit is contained in:
bigfish 2023-06-27 14:04:37 +08:00
parent 6276c1fafc
commit f688825080
1 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/photo_classifier.py
View File

@ -21,12 +21,13 @@ from win32com.propsys import propsys, pscon
class Classifier(): class Classifier():
mode = 'prod' # 开发模式(dev)还是产品模式(prod) mode = 'prod' # 开发模式(dev)还是产品模式(prod)
IMAGE_EXTENTIONS = ['jpg', 'jpeg', 'bmp', 'png', 'tif', 'gif'] IMAGE_EXTENTIONS = ['jpg', 'jpeg', 'bmp', 'png', 'tif', 'gif']
VIDEO_EXTENTIONS = ['mp4', 'avi', 'rmvb', 'mkv', 'mov', 'ppt', 'amr', 'mpg'] VIDEO_EXTENTIONS = ['mp4', 'avi', 'rmvb', 'mkv', 'mov', 'amr', 'mpg']
TEST_TABLE = 'TEST_PHOTO' TEST_TABLE = 'TEST_PHOTO'
TABLE = 'PHOTO' TABLE = 'PHOTO'
PHOTO_NO_DATE_KEYS = ['EXIF ExifVersion'] PHOTO_NO_DATE_KEYS = ['EXIF ExifVersion']
PHOTO_DATE_KEYS = ['Image DateTime', 'EXIF DateTimeOriginal'] PHOTO_DATE_KEYS = ['Image DateTime', 'EXIF DateTimeOriginal']
PHOTO_EXIF_KEYS = PHOTO_NO_DATE_KEYS + PHOTO_DATE_KEYS PHOTO_EXIF_KEYS = PHOTO_NO_DATE_KEYS + PHOTO_DATE_KEYS
SKIP_FOLDERS = ['System Volume Information', '$RECYCLE.BIN', '.stfolder']
def __init__(self, input_folder, photo_output, video_output, image_output): def __init__(self, input_folder, photo_output, video_output, image_output):
self.input = input_folder self.input = input_folder
@ -76,6 +77,8 @@ class Classifier():
def delete_folders(self, folder): def delete_folders(self, folder):
for (root, dirs, files) in os.walk(folder): for (root, dirs, files) in os.walk(folder):
for dir in dirs: for dir in dirs:
if dir in self.SKIP_FOLDERS:
continue
abs_path = os.path.join(root, dir) abs_path = os.path.join(root, dir)
if os.path.isdir(abs_path): if os.path.isdir(abs_path):
if self.get_file_count(abs_path) == 0: if self.get_file_count(abs_path) == 0:
@ -114,6 +117,7 @@ class Classifier():
def process_file(self, root, file): def process_file(self, root, file):
file_path = os.path.join(root, file) file_path = os.path.join(root, file)
if self.is_image(file_path) or self.is_video(file_path):
md5 = self.get_md5(file_path) md5 = self.get_md5(file_path)
try: try:
self.validate(file_path, md5) self.validate(file_path, md5)
@ -124,6 +128,8 @@ class Classifier():
print('已处理 {}: {} --> {}'.format(self.processed_count, file, new_name)) print('已处理 {}: {} --> {}'.format(self.processed_count, file, new_name))
except Exception as e: except Exception as e:
print(str(e)) print(str(e))
else:
print('非图片或视频, 忽略文件: {}'.format(file_path))
def add_record(self, md5): def add_record(self, md5):
try: try:
@ -219,7 +225,7 @@ class Classifier():
cf = Classifier( cf = Classifier(
input_folder='D:/temp/相册', input_folder='D:/待分类照片视频',
# input_folder='z:/待分类照片视频/Picture', # input_folder='z:/待分类照片视频/Picture',
photo_output='D:/总仓库-照片视频/总照片备份', photo_output='D:/总仓库-照片视频/总照片备份',
video_output='D:/总仓库-照片视频/总视频备份', video_output='D:/总仓库-照片视频/总视频备份',