134 lines
3.6 KiB
Python
Executable File
134 lines
3.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sys, os, re
|
|
import difflib as dl
|
|
|
|
# Define color codes
|
|
Green = "\033[32;1m"
|
|
Red = "\033[31;1m"
|
|
Reset = "\033[0m"
|
|
# Available shortcut keys
|
|
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)
|
|
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(r"%s\g<1>%s" % (Green, Reset), d) for d in directories]
|
|
|
|
|
|
def search_close_match(directories, key):
|
|
""" Return a list of Components whose name closely matches key """
|
|
match = []
|
|
key = key.lower()
|
|
for name in directories:
|
|
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":
|
|
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)
|
|
pat_dir = os.path.dirname(pattern)
|
|
|
|
if pat_dir == "":
|
|
pat_dir = "."
|
|
|
|
if not os.path.exists(pat_dir):
|
|
errexit("%s: %sNo such folder%s" % (pat_dir, Red, Reset))
|
|
|
|
# list sub-folders in the folder
|
|
current_folders = get_dirs(pat_dir)
|
|
|
|
# no sub-folders
|
|
if len(current_folders) == 0:
|
|
errexit("There are no subfolders in %s" % (pat_dir,))
|
|
|
|
# find matches
|
|
match_case = re.compile(".*" + pat_base + ".*")
|
|
repl_case = re.compile("(" + pat_base + ")")
|
|
|
|
matching_folders = [d for d in current_folders if match_case.match(d)]
|
|
matches = colorize_matches(matching_folders, repl_case)
|
|
|
|
# no matches, try case insensitive
|
|
if len(matching_folders) == 0:
|
|
match_nocase = re.compile(".*" + pat_base + ".*", re.I)
|
|
repl_nocase = re.compile("(" + pat_base + ")", re.I)
|
|
matching_folders = [d for d in current_folders if match_nocase.match(d)]
|
|
matches = colorize_matches(matching_folders, repl_nocase)
|
|
|
|
# No matches, try close match
|
|
if len(matching_folders) == 0:
|
|
matching_folders = search_close_match(current_folders, pat_base)
|
|
matches = [d for d in matching_folders]
|
|
|
|
# One match, print and return
|
|
if len(matching_folders) == 1:
|
|
printerr(matches[0])
|
|
print(os.path.join(pat_dir, matching_folders[0]))
|
|
sys.exit(0)
|
|
|
|
# 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...")
|
|
|
|
key_in = getch().get()
|
|
try:
|
|
key_index = key_list.index(key_in)
|
|
key_match = matching_folders[key_index]
|
|
except (ValueError, IndexError) as err:
|
|
errexit("%s'%s' Not in the list%s" % (Red, key_in, Reset))
|
|
|
|
print(os.path.join(pat_dir, key_match))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|