110 lines
3.2 KiB
Python
Executable File
110 lines
3.2 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
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):
|
|
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 colorize_matches(directories,repl):
|
|
return [repl.sub("%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)
|
|
s_ratio=s.ratio()
|
|
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
|
|
|
|
# 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)
|
|
|
|
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):
|
|
printerr("%s: %sNo such folder%s"%(pat_dir,Red,Reset))
|
|
sys.exit(1)
|
|
|
|
# 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()
|
|
|
|
# no sub-folders
|
|
if len(current_folders)==0:
|
|
printerr("There are no subfolders in %s"%(pat_dir,))
|
|
sys.exit(0)
|
|
|
|
# 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...")
|
|
import termios,tty
|
|
ch=getch()
|
|
key_in=ch.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)
|
|
print(os.path.join(pat_dir,key_match))
|