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

First version of argparse implemented

Browse Source
This commit is contained in:
William Ting 2012-04-07 04:14:19 -10:00
parent c91843c698
commit 9147d6cf71
2 changed files with 190 additions and 81 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

270
autojump
View File

@ -22,13 +22,14 @@ frequently used places."""
from __future__ import division, print_function from __future__ import division, print_function
import getopt import getopt
import argparse
from sys import argv, stderr, version_info, exit, getfilesystemencoding from sys import argv, stderr, version_info, exit, getfilesystemencoding
from tempfile import NamedTemporaryFile from tempfile import NamedTemporaryFile
from operator import itemgetter from operator import itemgetter
import os import os
import shutil import shutil
AUTOJUMP_VERSION = "release-v19" AUTOJUMP_VERSION = "release-v19-78-gc91843c-dirty"
MAX_KEYWEIGHT = 1000 MAX_KEYWEIGHT = 1000
MAX_STORED_PATHS = 600 MAX_STORED_PATHS = 600
COMPLETION_SEPARATOR = '__' COMPLETION_SEPARATOR = '__'
@ -218,104 +219,211 @@ def get_dic_file(filename="autojump.txt"):
def shell_utility(): def shell_utility():
"""Run this when autojump is called as a shell utility""" """Run this when autojump is called as a shell utility"""
try: parser = argparse.ArgumentParser(description='Automatically jump to directory passed as an argument.',
optlist, args = getopt.getopt(argv[1:], 'a', epilog="Please see autojump(1) man pages for full documentation.")
['stat', 'import', 'completion', 'bash', 'version', 'help']) parser.add_argument('directory', metavar='DIR', type=unicode, nargs='*', default=None,
except getopt.GetoptError as ex: help='directory to jump to')
print("Unknown command line argument: %s" % ex, file=stderr) parser.add_argument('-a', '--add', metavar='DIR', type=unicode,
exit(1) help='add path to database')
parser.add_argument('-b', '--bash', action="store_true", default=False,
help='')
parser.add_argument('--completion', action="store_true", default=False,
help='remember all entries')
parser.add_argument('--stat', action="store_true", default=False,
help='show database entries and their key weights')
parser.add_argument('--version', action="version", version="%(prog)s " + AUTOJUMP_VERSION,
help='show version information and exit')
args = parser.parse_args()
dic_file = get_dic_file() dic_file = get_dic_file()
path_dict = open_dic(dic_file) path_dict = open_dic(dic_file)
if ('-a', '') in optlist:
# The home dir can be reached quickly by "cd" # The home dir can be reached quickly by "cd" and may interfere with other directories
# and may interfere with other directories if (args.add):
if(args[-1] != os.path.expanduser("~")): if(args.add != os.path.expanduser("~")):
dicadd(path_dict, decode(args[-1])) dicadd(path_dict, decode(args.add))
save(path_dict, dic_file) save(path_dict, dic_file)
elif ('--stat', '') in optlist: return True
if (args.stat):
paths = list(path_dict.items()) paths = list(path_dict.items())
paths.sort(key=itemgetter(1)) paths.sort(key=itemgetter(1))
for path, count in paths[-100:]: for path, count in paths[-100:]:
output(unico("%.1f:\t%s") % (count, path)) output(unico("%.1f:\t%s") % (count, path))
print("Total key weight: %d. Number of stored paths: %d" % print("Total key weight: %d. Number of stored paths: %d" %
(sum(path_dict.values()), len(paths))) (sum(path_dict.values()), len(paths)))
elif ('--version', '') in optlist: return True
print("autojump %s" % AUTOJUMP_VERSION)
elif ('--help', '') in optlist: import re
print("usage: j <dirspec>") #userchoice is i if the pattern is __pattern__i, otherwise -1
print("where dirspec is a few characters of the directory you want to jump to.") userchoice = -1
results = []
#default: gradually forget about old directories
if (not args.completion): forget(path_dict, dic_file)
if (args.directory == None):
patterns = [unico("")]
else: else:
import re patterns = [decode(a) for a in args.directory]
completion = False
#userchoice is i if the pattern is __pattern__i, otherwise -1 # If the last pattern contains a full path, jump there
userchoice = -1 # The regexp is because we need to support stuff like
results = [] # "j wo jo__3__/home/joel/workspace/joel" for zsh
if ('--completion', '') in optlist: last_pattern_path = re.sub("(.*)"+COMPLETION_SEPARATOR, "", patterns[-1])
completion = True if (len(last_pattern_path)>0 and
last_pattern_path[0] == "/" and
os.path.exists(last_pattern_path)):
if not args.completion: output(last_pattern_path)
else:
#check for ongoing completion, and act accordingly
endmatch = re.search(COMPLETION_SEPARATOR+"([0-9]+)", patterns[-1])
if endmatch: #user has selected a completion
userchoice = int(endmatch.group(1))
patterns[-1] = re.sub(COMPLETION_SEPARATOR+"[0-9]+.*",
"", patterns[-1])
else: #user hasn't selected a completion, display the same choices again
endmatch = re.match("(.*)"+COMPLETION_SEPARATOR, patterns[-1])
if endmatch: patterns[-1] = endmatch.group(1)
dirs = list(path_dict.items())
dirs.sort(key=itemgetter(1), reverse=True)
if args.completion or userchoice != -1:
max_matches = 9
else: else:
forget(path_dict, dic_file) #gradually forget about old directories max_matches = 1
if not args: patterns = [unico("")]
else: patterns = [decode(a) for a in args]
# If the last pattern contains a full path, jump there # Don't jump to the current directory
# The regexp is because we need to support stuff like try:
# "j wo jo__3__/home/joel/workspace/joel" for zsh current_dir = decode(os.path.realpath(os.curdir))
last_pattern_path = re.sub("(.*)"+COMPLETION_SEPARATOR, "", patterns[-1]) #Sometimes the current path doesn't exist anymore.
if (len(last_pattern_path)>0 and #In that case, jump if possible.
last_pattern_path[0] == "/" and except OSError:
os.path.exists(last_pattern_path)): current_dir = None
if not completion: output(last_pattern_path) find_matches(dirs, patterns, results, False, max_matches, current_dir)
# If not found, try ignoring case.
# On completion always show all results
if args.completion or not results:
find_matches(dirs, patterns, results,
ignore_case=True,
max_matches=max_matches, current_dir=current_dir)
# Keep the database to a reasonable size
if not args.completion and clean_dict(dirs, path_dict):
save(path_dict, dic_file)
if args.completion and ('--bash', '') in optlist: quotes = "'"
else: quotes = ""
if userchoice != -1:
if len(results) > userchoice-1 :
output(unico("%s%s%s") % (quotes,results[userchoice-1],quotes))
elif len(results) > 1 and args.completion:
output("\n".join(("%s%s%d%s%s" % (patterns[-1],
COMPLETION_SEPARATOR, n+1, COMPLETION_SEPARATOR, r)
for n, r in enumerate(results[:8]))))
elif results: output(unico("%s%s%s")%(quotes,results[0],quotes))
else: else:
#check for ongoing completion, and act accordingly return False
endmatch = re.search(COMPLETION_SEPARATOR+"([0-9]+)", patterns[-1]) return True
if endmatch: #user has selected a completion
userchoice = int(endmatch.group(1))
patterns[-1] = re.sub(COMPLETION_SEPARATOR+"[0-9]+.*",
"", patterns[-1])
else: #user hasn't selected a completion, display the same choices again
endmatch = re.match("(.*)"+COMPLETION_SEPARATOR, patterns[-1])
if endmatch: patterns[-1] = endmatch.group(1)
dirs = list(path_dict.items()) #try:
dirs.sort(key=itemgetter(1), reverse=True) #optlist, args = getopt.getopt(argv[1:], 'a',
if completion or userchoice != -1: #['stat', 'import', 'completion', 'bash', 'version', 'help'])
max_matches = 9 #except getopt.GetoptError as ex:
else: #print("Unknown command line argument: %s" % ex, file=stderr)
max_matches = 1 #exit(1)
# Don't jump to the current directory #dic_file = get_dic_file()
try: #path_dict = open_dic(dic_file)
current_dir = decode(os.path.realpath(os.curdir)) #if ('-a', '') in optlist:
#Sometimes the current path doesn't exist anymore. ## The home dir can be reached quickly by "cd"
#In that case, jump if possible. ## and may interfere with other directories
except OSError: #if(args[-1] != os.path.expanduser("~")):
current_dir = None #dicadd(path_dict, decode(args[-1]))
find_matches(dirs, patterns, results, False, max_matches, current_dir) #save(path_dict, dic_file)
# If not found, try ignoring case. #elif ('--stat', '') in optlist:
# On completion always show all results #paths = list(path_dict.items())
if completion or not results: #paths.sort(key=itemgetter(1))
find_matches(dirs, patterns, results, #for path, count in paths[-100:]:
ignore_case=True, #output(unico("%.1f:\t%s") % (count, path))
max_matches=max_matches, current_dir=current_dir) #print("Total key weight: %d. Number of stored paths: %d" %
# Keep the database to a reasonable size #(sum(path_dict.values()), len(paths)))
if not completion and clean_dict(dirs, path_dict): #elif ('--version', '') in optlist:
save(path_dict, dic_file) #print("autojump %s" % AUTOJUMP_VERSION)
#elif ('--help', '') in optlist:
#print("usage: j <dirspec>")
#print("where dirspec is a few characters of the directory you want to jump to.")
#else:
#import re
#completion = False
##userchoice is i if the pattern is __pattern__i, otherwise -1
#userchoice = -1
#results = []
#if ('--completion', '') in optlist:
#completion = True
#else:
#forget(path_dict, dic_file) #gradually forget about old directories
#if not args: patterns = [unico("")]
#else: patterns = [decode(a) for a in args]
if completion and ('--bash', '') in optlist: quotes = "'" ## If the last pattern contains a full path, jump there
else: quotes = "" ## The regexp is because we need to support stuff like
## "j wo jo__3__/home/joel/workspace/joel" for zsh
#last_pattern_path = re.sub("(.*)"+COMPLETION_SEPARATOR, "", patterns[-1])
#if (len(last_pattern_path)>0 and
#last_pattern_path[0] == "/" and
#os.path.exists(last_pattern_path)):
#if not completion: output(last_pattern_path)
#else:
##check for ongoing completion, and act accordingly
#endmatch = re.search(COMPLETION_SEPARATOR+"([0-9]+)", patterns[-1])
#if endmatch: #user has selected a completion
#userchoice = int(endmatch.group(1))
#patterns[-1] = re.sub(COMPLETION_SEPARATOR+"[0-9]+.*",
#"", patterns[-1])
#else: #user hasn't selected a completion, display the same choices again
#endmatch = re.match("(.*)"+COMPLETION_SEPARATOR, patterns[-1])
#if endmatch: patterns[-1] = endmatch.group(1)
if userchoice != -1: #dirs = list(path_dict.items())
if len(results) > userchoice-1 : #dirs.sort(key=itemgetter(1), reverse=True)
output(unico("%s%s%s") % (quotes,results[userchoice-1],quotes)) #if completion or userchoice != -1:
elif len(results) > 1 and completion: #max_matches = 9
output("\n".join(("%s%s%d%s%s" % (patterns[-1], #else:
COMPLETION_SEPARATOR, n+1, COMPLETION_SEPARATOR, r) #max_matches = 1
for n, r in enumerate(results[:8]))))
elif results: output(unico("%s%s%s")%(quotes,results[0],quotes)) ## Don't jump to the current directory
else: #try:
return False #current_dir = decode(os.path.realpath(os.curdir))
return True ##Sometimes the current path doesn't exist anymore.
##In that case, jump if possible.
#except OSError:
#current_dir = None
#find_matches(dirs, patterns, results, False, max_matches, current_dir)
## If not found, try ignoring case.
## On completion always show all results
#if completion or not results:
#find_matches(dirs, patterns, results,
#ignore_case=True,
#max_matches=max_matches, current_dir=current_dir)
## Keep the database to a reasonable size
#if not completion and clean_dict(dirs, path_dict):
#save(path_dict, dic_file)
#if completion and ('--bash', '') in optlist: quotes = "'"
#else: quotes = ""
#if userchoice != -1:
#if len(results) > userchoice-1 :
#output(unico("%s%s%s") % (quotes,results[userchoice-1],quotes))
#elif len(results) > 1 and completion:
#output("\n".join(("%s%s%d%s%s" % (patterns[-1],
#COMPLETION_SEPARATOR, n+1, COMPLETION_SEPARATOR, r)
#for n, r in enumerate(results[:8]))))
#elif results: output(unico("%s%s%s")%(quotes,results[0],quotes))
#else:
#return False
#return True
if __name__ == "__main__": if __name__ == "__main__":

1
autojump.zsh
View File

@ -59,6 +59,7 @@ function j {
echo -e "\\033[31m${new_path}\\033[0m" echo -e "\\033[31m${new_path}\\033[0m"
cd "${new_path}" cd "${new_path}"
else else
echo "autojump returns: ${new_path}"
echo "autojump: directory '${@}' not found" echo "autojump: directory '${@}' not found"
echo "Try \`autojump --help\` for more information." echo "Try \`autojump --help\` for more information."
false false