Fix(DB): Fix column encoding

[xsetbg.git] / print-filename.py

diff --git a/print-filename.py b/print-filename.py
index 63a1481079468b4179fe6bc79494830ecdbc6a9f..e9f217b75b46920f4561b72ce29c0ac6dfcd0cf8 100755 (executable)
--- a/print-filename.py
+++ b/print-filename.py
@@ -1,86 +1,61 @@
-#! /usr/local/bin/python -O
-"""This file is a part of XSetBg.
-Author: Oleg BroytMann <phd@phd.pp.ru>
-Copyright (C) 2004-2006 PhiloSoft Design
+#! /usr/bin/env python3
+"""print(background filename)
+
+print(the filename of the current or previous background image.)
+
+This file is a part of XSetBg.
+
 """
 
-__version__ = "1.3.0"
-__author__ = "Oleg BroytMann <phd@phd.pp.ru>"
-__copyright__ = "Copyright (C) 2004-2006 PhiloSoft Design"
-__revision__ = "$Id$"[5:-2]
-__date__ = "$Date$"[7:-2]
+import sys
+from xsetbg_db import xsetbg_db
 
+if not xsetbg_db:
+    sys.exit("Error: no database found")
 
-import sys, os, shelve
 
 def usage(code=0):
-   sys.stderr.write("Usage: %s [-0|--null] [-o|--old] [-w|--width] [width]\n" % sys.argv[0])
-   sys.exit(code)
+    sys.stderr.write("Usage: %s [index]\n" % sys.argv[0])
+    sys.exit(code)
 
 
 def get_args():
-   from getopt import getopt, GetoptError
-
-   try:
-      options, arguments = getopt(sys.argv[1:], "0ow:",
-         ["null", "old", "width="])
-   except GetoptError:
-      usage(1)
-
-   print0 = False
-   old = False
-   width = None
-
-   for option, value in options:
-      if option in ("-h", "--help"):
-         usage()
-      elif option in ("-0", "--null"):
-         print0 = True
-      elif option in ("-o", "--old"):
-         old = True
-      elif option in ("-w", "--width"):
-         width = int(value)
-      else:
-         usage(2)
-
-   if arguments:
-      if width is not None:
-         usage(3)
-      elif len(arguments) > 1:
-         usage(4)
-      else:
-         width = int(arguments[0])
-
-   return print0, old, width
-
-print0, old, width = get_args()
-
-
-xsetbg_dir = os.path.join(os.environ["HOME"], "lib", "xsetbg")
-os.chdir(xsetbg_dir)
-
-global_db_name = "xsetbg.db"
-filename_key = "filename"
-old_filename_key = "old_filename"
-
-global_db = shelve.open(global_db_name, flag='r')
-if old:
-   key = old_filename_key
-else:
-   key = filename_key
-filename = global_db[key]
-global_db.close()
-
-if width is not None:
-   lines = []
-   while filename:
-      lines.append(filename[:width])
-      filename = filename[width:]
-   filename = "\n".join(lines)
-
-sys.stdout.write(filename)
-
-if print0:
-   sys.stdout.write('\0')
-else:
-   print
+    from getopt import getopt, GetoptError
+
+    try:
+        options, arguments = getopt(sys.argv[1:], "he:",
+                                    ["help", "output-encoding="])
+    except GetoptError:
+        usage(1)
+
+    index = 0
+    output_encoding = None
+
+    for option, value in options:
+        if option in ("-h", "--help"):
+            usage()
+        elif option in ("-e", "--output-encoding"):
+            output_encoding = value
+        else:
+            usage(2)
+
+    if arguments:
+        if len(arguments) == 1:
+            try:
+                index = int(arguments[0])
+            except ValueError:
+                usage(3)
+        else:
+            usage(4)
+
+    if output_encoding is None:
+        from m_lib.defenc import default_encoding
+        output_encoding = default_encoding
+    return index, output_encoding
+
+
+index, output_encoding = get_args()
+filename = xsetbg_db.select('last_shown IS NOT NULL',
+                            orderBy='-last_shown')[index].full_name
+
+sys.stdout.buffer.write(filename.encode(output_encoding) + b'\n')