diff --git a/imgsort/sorter.py b/imgsort/sorter.py index 05cac5a..dcdd956 100755 --- a/imgsort/sorter.py +++ b/imgsort/sorter.py @@ -1,17 +1,27 @@ #!/bin/python3 -from imgsort.ueberzug import UeberzugLayer -from imgsort.configs import read_config, write_config, select_config, create_config - import curses as c +import argparse +import os from os import path, getcwd, listdir, mkdir, makedirs, rename -from sys import argv +import subprocess + +if __name__ == "__main__": # make relative imports work as described here: https://peps.python.org/pep-0366/#proposed-change + if __package__ is None: + __package__ = "imgsort" + import sys + filepath = path.realpath(path.abspath(__file__)) + sys.path.insert(0, path.dirname(path.dirname(filepath))) + +from .ueberzug import UeberzugLayer +from .configs import read_config, write_config, select_config, create_config settings = { "q": "quit", "s": "skip", "u": "undo", + "o": "open" } # Size settings @@ -25,6 +35,8 @@ CURSOR_Y = 2 KEYS_BEGIN = 5 +version = "1.2" + class Sorter: def __init__(self, wdir, config): self.wd = wdir @@ -63,9 +75,6 @@ class Sorter: self._img_height = self.win_y - FOOTER_HEIGHT - 2 self._img_identifier = "imgsort_preview" - # version - self.version = "Image Sorter 1.1" - def validate_dirs(self): """ Create the directories that dont exist. @@ -131,6 +140,12 @@ class Sorter: else: self.message = "Nothing to undo!" continue + elif settings[self.pressed_key] == "open": + try: + subprocess.run(['xdg-open', self.image]) + self.message = "Opening with xdg-open" + except Exception as e: + print(f"open: Error: {e}") # move to folder elif self.pressed_key in self.keys: @@ -139,7 +154,7 @@ class Sorter: 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.message = f"ERROR: Failed to move '{self.image}' to '{self.keys[self.pressed_key]}'." self.image_iter += 1 self.quit("All done!") @@ -155,8 +170,9 @@ class Sorter: 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) + version_str = f"imgsort {version}" + x = self.win_x - len(version_str) - 1 + self.window.addstr(self.win_y - 1, x, version_str) # wd wdstring = f"Sorting {self.wd} - {len(self.images)} files - {len(self.images) - self.image_iter} remaining." @@ -217,37 +233,59 @@ class Sorter: rename(file, new_path) return new_path - def quit(self, message = ""): + def quit(self, message = ""): self.window.clear() self.window.refresh() c.endwin() print(message) - print("Quitting " + self.version) + print(f"Quitting imgsort {version}") exit(0) def main(): # set working directory - print(""" -=================================================================================================== -Image Sorter -=================================================================================================== -""") - if len(argv) > 1: - wd = path.abspath(argv[1]) - else: - wd = getcwd(); + print(f"""=================================================================================================== +imgsort {version} - Image Sorter +===================================================================================================""") + # set directories + config_dir = os.path.join(os.environ.get("XDG_CONFIG_HOME", os.path.expanduser("~/.config")), "regina") + # check if environment variables are set and use them if they are + if 'IMGSOSRT_CONFIG_DIR' in os.environ: config_dir = os.environ['IMGSORT_CONFIG_DIR'] - config_name = select_config() - if type(config_name) == str: - config = read_config(config_name) + parser = argparse.ArgumentParser(prog="imgsort") + parser.add_argument("--config", "-c", action="store", help="name or path of config file", metavar="config file") + parser.add_argument("--dir", "-d", action="store", help="working directory", metavar="working directory") + args = parser.parse_args() + + # working directory + if args.dir: + if not path.isdir(args.dir): + parser.error(f"invalid working directory: {args.wdir}") + wd = path.abspath(args.dir) + os.chdir(wd) + else: + wd = getcwd() + + # configuration + if args.config: + config_path = args.config_file + # search locally + if not path.isfile(config_path): + config_path = path.join(config_dir, args.config) + if not path.isfile(config_path): + parser.error(f"invalid configuration path/name:'{config_path}'/'{args.config}'") + else: + config_path = select_config() + print(config_path) + if config_path is not None: + config = read_config(config_path) else: config = create_config() if not config: print("Error reading the config:") - print(" Config Name:", config_name) + print(" Config Name:", config_path) print(" Config:", config) exit(1)