blob: d20a56c364a265f5097f6f5345bd12117d543f08 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
#! python3
# Chapter 17 Practice Indetifying Photo Folders on the Hard Drive
# Searches your hard drive for any (sub)folders that contain image files
# that are .png or .jpg that are at least 500 pixels in size or larger
# and prints their absolute path to the screen.
import os
from PIL import Image
for foldername, subfolders, filenames in os.walk('c:\\'):
numPhotoFiles = 0
numNonPhotoFiles = 0
for filename in filenames:
# Check if extension is not .png or .jpg
if not filename.endswith('.png') or filename.endswith('.jpg'):
numNonPhotoFiles += 1
continue
# Open image file using pillow. Skip file if error occurs.
try: im = Image.open(os.path.join(foldername, filename))
except: continue
width, height = im.size
# Check if width and height are larger than 500
if width >= 500 and height >= 500:
# Image is large enough to be considered a photo.
numPhotoFiles += 1
else:
# Image is too small to be a photo.
numNonPhotoFiles += 1
# If more than half of files were photos,
# print the absolute path of the folder.
if numPhotoFiles > numNonPhotoFiles:
print(foldername)
|
