Archives/wting_autojump
1
0
Fork 0
mirror of https://github.com/wting/autojump synced 2024-10-27 20:34:07 +00:00
Code Issues Releases Wiki Activity

some support for bash completion

Browse Source
This commit is contained in:
Joel Schaerer 2009-02-13 23:22:32 +01:00
parent 11dd2b4398
commit ff5c978079
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
autojump
View File

@ -21,6 +21,10 @@ function j { cd "$(autojump.py $1)"; }
max_keyweight=1000 max_keyweight=1000
def uniqadd(list,key):
if key not in list:
list.append(key)
def dicadd(dic,key,increment=1): def dicadd(dic,key,increment=1):
dic[key]=dic.get(key,0.)+increment dic[key]=dic.get(key,0.)+increment
@ -35,7 +39,7 @@ def match(path,pattern,path_dict,re_flags=0):
del path_dict[path] del path_dict[path]
return False return False
optlist, args = getopt.getopt(argv[1:], 'a',['stat','import']) optlist, args = getopt.getopt(argv[1:], 'a',['stat','import','completion'])
dic_file=os.path.expanduser("~/.autojump_py") dic_file=os.path.expanduser("~/.autojump_py")
try: try:
aj_file=open(dic_file) aj_file=open(dic_file)
@ -58,6 +62,10 @@ elif ('--import','') in optlist:
dicadd(path_dict,i[:-1]) dicadd(path_dict,i[:-1])
cPickle.dump(path_dict,open(dic_file,'w'),-1) cPickle.dump(path_dict,open(dic_file,'w'),-1)
else: else:
completion=False
if ('--completion','') in optlist:
completion=True
results=[]
keyweight=sum(path_dict.values()) #Gradually forget about old directories keyweight=sum(path_dict.values()) #Gradually forget about old directories
if keyweight>max_keyweight: if keyweight>max_keyweight:
for k in path_dict.keys(): for k in path_dict.keys():
@ -69,16 +77,22 @@ else:
found=False found=False
for path,count in dirs: for path,count in dirs:
if match(path," ".join(args),path_dict): #First look for case-matching path if match(path," ".join(args),path_dict): #First look for case-matching path
print path if not completion:
found=True print path
break found=True
break
else: uniqadd(results,path)
dirs=path_dict.items() #we need to recreate the list since the first iteration potentially deletes paths dirs=path_dict.items() #we need to recreate the list since the first iteration potentially deletes paths
dirs.sort(key=lambda e:e[1],reverse=True) dirs.sort(key=lambda e:e[1],reverse=True)
if not found: if not found:
for path,count in dirs: for path,count in dirs:
if match(path," ".join(args),path_dict,re.IGNORECASE): #Then try to ignore case if match(path," ".join(args),path_dict,re.IGNORECASE): #Then try to ignore case
print path if not completion:
break print path
break
else: uniqadd(results,path)
if completion:
for p in results: print p
cPickle.dump(path_dict,open(dic_file+".tmp",'w'),-1) cPickle.dump(path_dict,open(dic_file+".tmp",'w'),-1)
import shutil import shutil
shutil.copy(dic_file+".tmp",dic_file) #cPickle.dump doesn't seem to be atomic, so this is more secure shutil.copy(dic_file+".tmp",dic_file) #cPickle.dump doesn't seem to be atomic, so this is more secure