restructure gcd

files/gcd-findmatch.py

@@ -12,19 +12,28 @@ key_list="1234567890qwertyuiop"
 
 class getch:
     def get(self):
+        import termios, tty
         fd = sys.stdin.fileno()
         old_settings = termios.tcgetattr(fd)
         try:
             tty.setraw(sys.stdin.fileno())
             ch = sys.stdin.read(1)
         finally:
-            termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+            termios.tcsetattr(
+                fd,
+                termios.TCSADRAIN,
+                old_settings
+            )
         return ch
 
 def printerr(s):
     sys.stderr.write(s + "\n")
     sys.stderr.flush()
 
+def errexit(s):
+    printerr(s)
+    sys.exit(1)
+
 def colorize_matches(directories, repl):
     return [repl.sub("%s\g<1>%s"%(Green, Reset), d) for d in directories]
 
@@ -33,18 +42,29 @@ def search_close_match(directories, key):
     match = []
     key = key.lower()
     for name in directories:
-        s=dl.SequenceMatcher(None, name.lower(), key)
-        s_ratio=s.ratio()
-        if s_ratio > 0:
-            match.append(( s_ratio, name ))
+        s = dl.SequenceMatcher(
+            None,
+            name.lower(),
+            key
+        )
+        if s.ratio() > 0:
+            match.append(( s.ratio(), name ))
     match.sort(key = lambda x: x[0], reverse = True)
     best_matches=[x[1] for x in match[0:10]]
     return best_matches
 
+def get_dirs(pat_dir):
+
+    # list sub-folders in the folder
+    current_folders = [d for d in os.listdir(pat_dir) if os.path.isdir(os.path.join(pat_dir, d))]
+    current_folders = [d for d in current_folders if not d.startswith(".")]
+    current_folders.sort()
+    return current_folders
+
+def main():
     # Print help
     if len(sys.argv) == 1 or sys.argv[-1] == "-h":
-    printerr("Guess cd: Find first match in folder, interactive if multiple matches\n  Arguments: [dir/]pattern_to_match_folders")
-    sys.exit(0)
+        errexit("Guess cd: Find first match in folder, interactive if multiple matches\n  Arguments: [dir/]pattern_to_match_folders")
 
     pattern = ".".join(sys.argv[1:])
     pat_base = os.path.basename(pattern)
@@ -54,18 +74,14 @@ if pat_dir=="":
         pat_dir = "."
 
     if not os.path.exists(pat_dir):
-    printerr("%s: %sNo such folder%s"%(pat_dir,Red,Reset))
-    sys.exit(1)
+        errexit("%s: %sNo such folder%s"%(pat_dir, Red, Reset))
 
     # list sub-folders in the folder
-current_folders=[d for d in os.listdir(pat_dir) if os.path.isdir(os.path.join(pat_dir,d))]
-current_folders=[d for d in current_folders if not d.startswith(".")]
-current_folders.sort()
+    current_folders = get_dirs(pat_dir)
 
     # no sub-folders
     if len(current_folders) == 0:
-    printerr("There are no subfolders in %s"%(pat_dir,))
-    sys.exit(0)
+        errexit("There are no subfolders in %s"%( pat_dir,))
 
     # find matches
     match_case = re.compile(".*" + pat_base + ".*")
@@ -95,15 +111,18 @@ if len(matching_folders)==1:
     # Many matches, ask the user for folder index
     for i,m in enumerate(matches[0:len(key_list)]):
         printerr(key_list[i] + ": " + m)
+
     if len(matches) > len(key_list):
         printerr("Skipping the rest...")
-import termios,tty
-ch=getch()
-key_in=ch.get()
+
+    key_in = getch().get()
     try:
         key_index = key_list.index(key_in)
         key_match = matching_folders[key_index]
     except (ValueError, IndexError) as err:
-    printerr("%s'%s' Not in the list%s"%(Red,key_in,Reset))
-    sys.exit(1)
+        errexit("%s'%s' Not in the list%s"%(Red, key_in, Reset))
+
     print(os.path.join(pat_dir, key_match))
+
+if __name__ == "__main__":
+    main()