use argparse, add xdg-open

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
+        version_str = f"imgsort {version}"
-        self.window.addstr(self.win_y - 1, x, self.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("""
+    print(f"""===================================================================================================
-===================================================================================================
+imgsort {version} - Image Sorter
-Image Sorter
+===================================================================================================""")
-===================================================================================================
+    # set directories
-""")
+    config_dir = os.path.join(os.environ.get("XDG_CONFIG_HOME", os.path.expanduser("~/.config")), "regina")
-    if len(argv) > 1:
+    # check if environment variables are set and use them if they are
-        wd = path.abspath(argv[1])
+    if 'IMGSOSRT_CONFIG_DIR' in os.environ: config_dir = os.environ['IMGSORT_CONFIG_DIR']
-    else:
-        wd = getcwd();
 
-    config_name = select_config()
+    parser = argparse.ArgumentParser(prog="imgsort")
-    if type(config_name) == str:
+    parser.add_argument("--config", "-c",   action="store",         help="name or path of config file", metavar="config file")
-        config = read_config(config_name)
+    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)