.fvwmrc: Refactor many SetVolumeAndBrowser functions into one with parameters

[dotfiles.git] / bin / recode-filenames-recursive.py

diff --git a/bin/recode-filenames-recursive.py b/bin/recode-filenames-recursive.py
index 9afd6c7a361dd9cc24c13edf58a96b43aba8c2e3..5afeed12d9a818619cb4d61b2dc776d0ce02a375 100755 (executable)
--- a/bin/recode-filenames-recursive.py
+++ b/bin/recode-filenames-recursive.py
@@ -1,39 +1,39 @@
-#! /usr/bin/env python
+#! /home/phd/.local/bin/python3
 
-import sys, os
+import os
 
-if len(sys.argv) == 3:
-   start_dir = '.'
-elif len(sys.argv) == 4:
-   start_dir = sys.argv[3]
-else:
-   sys.exit("Usage: %s src_enc dst_enc [start_dir]" % sys.argv[0])
+from recode_filenames import parse_args, build_recode
 
+from_encoding, to_encoding, dirnames = parse_args(default='.')
+_recode = build_recode(from_encoding, to_encoding)
 
-def _p(plist, dirname, names):
-   plist.append((dirname, names))
 
+def _onerror(exc):
+    raise exc
 
-plist = []
-os.path.walk(start_dir, _p, plist)
-plist.reverse()
-
-
-from recode_filenames import _recode
 
 save_dir = os.getcwd()
-for dirname, names in plist:
-   os.chdir(dirname)
-   for filename in names:
-      # if not exists - it was renamed already
-      if os.path.exists(filename) and \
-            os.path.isfile(filename):
-         newname = _recode(filename)
-         if newname <> filename:
-            os.rename(filename, newname)
-   os.chdir('..')
-   dirname = os.path.basename(dirname)
-   newname = _recode(dirname)
-   if newname <> dirname:
-      os.rename(dirname, newname)
-   os.chdir(save_dir)
+for start_dir in dirnames:
+    for dirname, _subdirs, fnames in list(
+        os.walk(start_dir, topdown=False, onerror=_onerror)
+    ):
+        if dirname != '.':
+            os.chdir(dirname)
+        for filename in fnames:
+            # if not exists - it was renamed already
+            if os.path.exists(filename) and \
+                    os.path.isfile(filename):
+                newname = _recode(filename)
+                if not isinstance(newname, str):
+                    newname = newname.decode()
+                if newname != filename:
+                    os.rename(filename, newname)
+        if dirname != '.':
+            os.chdir('..')
+            dirname = os.path.basename(dirname)
+            newname = _recode(dirname)
+            if not isinstance(newname, str):
+                newname = newname.decode()
+            if newname != dirname:
+                os.rename(dirname, newname)
+        os.chdir(save_dir)