diff --git a/sizer.py b/sizer.py index cbbf21c..3b124cd 100755 --- a/sizer.py +++ b/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