Update sizer.py
Now handle portrait mode
This commit is contained in:
parent
4df6b42d08
commit
d0b512080e
37
sizer.py
37
sizer.py
@ -18,35 +18,48 @@ def file_move(file, dest):
|
||||
os.rename(file, dest)
|
||||
|
||||
|
||||
def main():
|
||||
def main():
|
||||
#For landscape
|
||||
folder4k = "Wallpapers_4K"
|
||||
folder2k = "Wallpapers_2k"
|
||||
folderhd = "Wallpapers_hd"
|
||||
foldersd = "Wallpapers_sd"
|
||||
folderuw = "UltrawideWallpapers"
|
||||
|
||||
#For portrait
|
||||
vert4k = "Vertical_4k"
|
||||
vert2k = "Vertical_2k"
|
||||
verthd = "Vertical_hd"
|
||||
vertuw = "Vertical_ultrawide"
|
||||
|
||||
res4k = (4096, 2160)
|
||||
reslow4k = (3840, 2160)
|
||||
res2k = (2560, 1440)
|
||||
reshd = (1920, 1080)
|
||||
resuw = (3440, 1440)
|
||||
|
||||
#VARS
|
||||
orig_folder = "VerticalWallpapers"
|
||||
|
||||
print("Size begins !")
|
||||
|
||||
for folder in [folder4k, folder2k, folderhd, foldersd, folderuw]:
|
||||
for folder in [folder4k, folder2k, folderhd, foldersd, folderuw, vert2k, vert4k, verthd, vertuw]:
|
||||
os.makedirs(join(".", folder), exist_ok=True)
|
||||
print(f"Folder {folder} created if needed")
|
||||
|
||||
|
||||
files = pics_list("./Wallpapers")
|
||||
|
||||
files = pics_list(orig_folder)
|
||||
|
||||
print(f"About to run on {len(files)} pictures...")
|
||||
cont = input("Press enter to start, Ctrl+C to cancel")
|
||||
|
||||
for f in files:
|
||||
f_path = join("wallpapers", f)
|
||||
f_path = join(orig_folder, f)
|
||||
size = pic_size_find(f_path)
|
||||
rsize = size[::-1]
|
||||
print(f"{f} => {size}")
|
||||
#Check landscape first
|
||||
if size == res4k or size == reslow4k:
|
||||
file_move(f_path, join(folder4k, f))
|
||||
dest_folder = folder4k
|
||||
@ -59,6 +72,22 @@ def main():
|
||||
elif size == resuw:
|
||||
file_move(f_path, join(folderuw, f))
|
||||
dest_folder = folderuw
|
||||
|
||||
#Check portrait by inverting sizes
|
||||
elif rsize == res4k or rsize == reslow4k:
|
||||
file_move(f_path, join(vert4k, f))
|
||||
dest_folder = vert4k
|
||||
elif rsize == res2k[::-1]:
|
||||
file_move(f_path, join(vert2k, f))
|
||||
dest_folder = vert2k
|
||||
elif rsize == reshd[::-1]:
|
||||
file_move(f_path, join(verthd, f))
|
||||
dest_folder = verthd
|
||||
elif rsize == resuw[::-1]:
|
||||
file_move(f_path, join(vertuw, f))
|
||||
dest_folder = vertuw
|
||||
|
||||
#...Or put in "SD" folder
|
||||
else:
|
||||
file_move(f_path, join(foldersd, f))
|
||||
dest_folder = foldersd
|
||||
|
Loading…
x
Reference in New Issue
Block a user