261 lines
8.2 KiB
Python
Executable File
261 lines
8.2 KiB
Python
Executable File
#!/bin/python3
|
|
|
|
|
|
from imgsort.ueberzug import UeberzugLayer
|
|
from imgsort.configs import read_config, write_config, select_config, create_config
|
|
|
|
import curses as c
|
|
from os import path, getcwd, listdir, mkdir, makedirs, rename
|
|
from sys import argv
|
|
|
|
settings = {
|
|
"q": "quit",
|
|
"s": "skip",
|
|
"u": "undo",
|
|
}
|
|
|
|
# Size settings
|
|
FOOTER_LEFT = 0
|
|
FOOTER_HEIGHT = 2
|
|
|
|
SIDEBAR_WIDTH = 40
|
|
|
|
CURSOR_X = 0
|
|
CURSOR_Y = 2
|
|
|
|
KEYS_BEGIN = 5
|
|
|
|
class Sorter:
|
|
def __init__(self, wdir, config):
|
|
self.wd = wdir
|
|
|
|
self.images = [] # old paths
|
|
self.images_new = [] # new paths
|
|
self.image_iter = 0
|
|
self.image = ""
|
|
|
|
self.keys = config
|
|
|
|
self.settings = settings
|
|
|
|
self.validate_dirs()
|
|
|
|
# info about last action
|
|
self.last_dir = ""
|
|
|
|
# curses
|
|
self.window = c.initscr()
|
|
# c.start_color()
|
|
# c.use_default_colors()
|
|
self.win_y, self.win_x = self.window.getmaxyx()
|
|
|
|
self.message = "" # displayed in footer
|
|
self.pressed_key = ""
|
|
|
|
c.echo()
|
|
|
|
# ueberzug
|
|
|
|
self._ueberzug = UeberzugLayer(pid_file="/tmp/ueberzu-imgsort.pid")
|
|
self._img_x = SIDEBAR_WIDTH + 1
|
|
self._img_y = 2
|
|
self._img_width = self.win_x - SIDEBAR_WIDTH - 1
|
|
self._img_height = self.win_y - FOOTER_HEIGHT - 2
|
|
self._img_identifier = "imgsort_preview"
|
|
|
|
# version
|
|
self.version = "1.2++"
|
|
|
|
def validate_dirs(self):
|
|
"""
|
|
Create the directories that dont exist.
|
|
"""
|
|
for d in self.keys.values():
|
|
if not path.isdir(d):
|
|
print(f"Directory '{d}' does not exist.")
|
|
decision = input(f"Create directory '{path.abspath(d)}'? y/n: ")
|
|
if (decision == "y"):
|
|
makedirs(d)
|
|
else:
|
|
print("Exiting - can not use non-existing directory.")
|
|
exit(1)
|
|
|
|
|
|
def get_images(self):
|
|
"""
|
|
Put all image-paths from wd in images dictionary.
|
|
"""
|
|
self.images.clear()
|
|
self.images_new.clear()
|
|
|
|
for name in listdir(self.wd):
|
|
name = path.normpath(self.wd + "/" + name)
|
|
if (path.isfile(name)):
|
|
self.images.append(name)
|
|
|
|
self.images.sort()
|
|
self.images_new = self.images.copy()
|
|
# print(self.images)
|
|
|
|
def display_image(self):
|
|
self._ueberzug.display_image(self.image, x=self._img_x, y=self._img_y, max_width=self._img_width, max_height=self._img_height, identifier=self._img_identifier)
|
|
self.window.addnstr(0, SIDEBAR_WIDTH + 1, self.image, self.win_x - SIDEBAR_WIDTH - 1)
|
|
|
|
def sort(self):
|
|
"""
|
|
Loop until all images are processed
|
|
"""
|
|
while (self.image_iter < len(self.images)):
|
|
self.image = self.images[self.image_iter]
|
|
|
|
self.print_window()
|
|
self.display_image()
|
|
|
|
self.pressed_key = self.window.getkey() # wait until user presses something
|
|
|
|
# check for quit, skip or undo
|
|
if self.pressed_key in self.settings:
|
|
if self.settings[self.pressed_key] == "quit":
|
|
self.quit(f"Key '{self.pressed_key}' pressed. Canceling image sorting")
|
|
elif self.settings[self.pressed_key] == "skip":
|
|
self.image_iter += 1
|
|
self.message = "Skipped image"
|
|
continue
|
|
elif settings[self.pressed_key] == "undo":
|
|
if self.image_iter > 0:
|
|
self.image_iter -= 1 # using last image
|
|
rename(self.images_new[self.image_iter], self.images[self.image_iter])
|
|
self.images_new[self.image_iter] = self.images[self.image_iter]
|
|
self.message = "Undone last action"
|
|
continue
|
|
else:
|
|
self.message = "Nothing to undo!"
|
|
continue
|
|
|
|
# move to folder
|
|
elif self.pressed_key in self.keys:
|
|
new_filepath = self.move_file(self.image, self.keys[self.pressed_key])
|
|
if new_filepath: # is string when successful
|
|
self.images_new[self.image_iter] = new_filepath
|
|
self.message = f"Moved image to {self.keys[self.pressed_key]}"
|
|
else:
|
|
self.message = f"ERROR: Failed to move '{self.image}' to '{keys[self.pressed_key]}'."
|
|
self.image_iter += 1
|
|
|
|
self.quit("All done!")
|
|
|
|
def print_window(self):
|
|
"""
|
|
Draw lines and text
|
|
"""
|
|
self.window.erase()
|
|
|
|
# lines
|
|
self.window.hline(self.win_y - FOOTER_HEIGHT, FOOTER_LEFT, '=', self.win_x)
|
|
self.window.vline(0, SIDEBAR_WIDTH, '|', self.win_y - FOOTER_HEIGHT + 1)
|
|
|
|
# version
|
|
x = self.win_x - len(self.version) - 1
|
|
self.window.addstr(self.win_y - 1, x, self.version)
|
|
|
|
# wd
|
|
wdstring = f"Sorting {self.wd} - {len(self.images)} files - {len(self.images) - self.image_iter} remaining."
|
|
self.window.addnstr(self.win_y - 1, FOOTER_LEFT, wdstring, self.win_x)
|
|
|
|
# message
|
|
self.window.addstr(self.win_y - FOOTER_HEIGHT, SIDEBAR_WIDTH + 2, self.message)
|
|
self.message = ""
|
|
|
|
# progress
|
|
progstring = f"File {self.image_iter + 1}/{len(self.images)}"
|
|
x = self.win_x - len(progstring) - 1
|
|
self.window.addstr(self.win_y - FOOTER_HEIGHT, x, progstring)
|
|
|
|
# print all key : action pairs
|
|
i = 0
|
|
self.window.hline(KEYS_BEGIN + i, 0, '-', SIDEBAR_WIDTH)
|
|
i += 1
|
|
self.window.addnstr(KEYS_BEGIN + i, 0, "Key: Action", SIDEBAR_WIDTH)
|
|
i += 1
|
|
self.window.hline(KEYS_BEGIN + i, 0, '-', SIDEBAR_WIDTH)
|
|
i += 1
|
|
for k, v in self.settings.items():
|
|
if i >= self.win_y - KEYS_BEGIN - FOOTER_HEIGHT: # dont write into footer
|
|
break
|
|
if k == self.pressed_key:
|
|
self.window.addnstr(KEYS_BEGIN + i, 0, f" {k}: {v}", SIDEBAR_WIDTH, c.A_STANDOUT)
|
|
else:
|
|
self.window.addnstr(KEYS_BEGIN + i, 0, f" {k}: {v}", SIDEBAR_WIDTH)
|
|
i += 1
|
|
self.window.hline(KEYS_BEGIN + i, 0, '-', SIDEBAR_WIDTH)
|
|
i += 1
|
|
|
|
# print all key : directory pairs
|
|
self.window.addnstr(KEYS_BEGIN + i, 0, "Key: Directory", SIDEBAR_WIDTH)
|
|
i += 1
|
|
self.window.hline(KEYS_BEGIN + i, 0, '-', SIDEBAR_WIDTH)
|
|
i += 1
|
|
for k, v in self.keys.items():
|
|
if i >= self.win_y - KEYS_BEGIN - FOOTER_HEIGHT: # dont write into footer
|
|
break
|
|
if k == self.pressed_key:
|
|
self.window.addnstr(KEYS_BEGIN + i, 0, f" {k}: {v}", SIDEBAR_WIDTH, c.A_STANDOUT)
|
|
else:
|
|
self.window.addnstr(KEYS_BEGIN + i, 0, f" {k}: {v}", SIDEBAR_WIDTH)
|
|
i += 1
|
|
|
|
self.window.move(CURSOR_Y, CURSOR_X)
|
|
|
|
def move_file(self, file, dest):
|
|
# if not path.isdir(dest):
|
|
# makedirs(dest)
|
|
if not path.isfile(file): return False
|
|
if not path.isdir(dest): return False
|
|
|
|
new_path = path.normpath(dest + '/' + path.split(file)[1])
|
|
|
|
rename(file, new_path)
|
|
return new_path
|
|
|
|
def quit(self, message = ""):
|
|
self.window.clear()
|
|
self.window.refresh()
|
|
c.endwin()
|
|
print(message)
|
|
print("Quitting " + self.version)
|
|
exit(0)
|
|
|
|
|
|
|
|
def main():
|
|
# set working directory
|
|
print("""
|
|
===================================================================================================
|
|
Image Sorter
|
|
===================================================================================================
|
|
""")
|
|
if len(argv) > 1:
|
|
wd = path.abspath(argv[1])
|
|
else:
|
|
wd = getcwd();
|
|
|
|
config_name = select_config()
|
|
if type(config_name) == str:
|
|
config = read_config(config_name)
|
|
else:
|
|
config = create_config()
|
|
|
|
if not config:
|
|
print("Error reading the config:")
|
|
print(" Config Name:", config_name)
|
|
print(" Config:", config)
|
|
exit(1)
|
|
|
|
sorter = Sorter(wd, config)
|
|
sorter.get_images()
|
|
sorter.sort()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|