1
0
mirror of https://github.com/wting/autojump synced 2024-09-28 22:10:45 +00:00

remove some trailing whitespace, change root test to check for UID instead of whoami

This commit is contained in:
William Ting 2012-01-13 11:05:43 -10:00
parent 0f0ba550f9
commit 04357595e2
2 changed files with 11 additions and 11 deletions

View File

@ -93,7 +93,7 @@ def save(path_dict, dic_file):
# using rename to overwrite files # using rename to overwrite files
shutil.move(temp.name, dic_file) shutil.move(temp.name, dic_file)
try: #backup file try: #backup file
import time import time
if (not os.path.exists(dic_file+".bak") or if (not os.path.exists(dic_file+".bak") or
time.time()-os.path.getmtime(dic_file+".bak")>86400): time.time()-os.path.getmtime(dic_file+".bak")>86400):
shutil.copy(dic_file, dic_file+".bak") shutil.copy(dic_file, dic_file+".bak")
@ -147,14 +147,14 @@ def open_dic(dic_file, error_recovery=False):
def forget(path_dict, dic_file): def forget(path_dict, dic_file):
"""Gradually forget about directories. Only call """Gradually forget about directories. Only call
from the actual jump since it can take time""" from the actual jump since it can take time"""
keyweight = sum(path_dict.values()) keyweight = sum(path_dict.values())
if keyweight > MAX_KEYWEIGHT: if keyweight > MAX_KEYWEIGHT:
for k in path_dict.keys(): for k in path_dict.keys():
path_dict[k] *= 0.9 * MAX_KEYWEIGHT / keyweight path_dict[k] *= 0.9 * MAX_KEYWEIGHT / keyweight
save(path_dict, dic_file) save(path_dict, dic_file)
def clean_dict(sorted_dirs, path_dict): def clean_dict(sorted_dirs, path_dict):
"""Limits the sized of the path_dict to MAX_STORED_PATHS. """Limits the sized of the path_dict to MAX_STORED_PATHS.
Returns True if keys were deleted""" Returns True if keys were deleted"""
if len(sorted_dirs) > MAX_STORED_PATHS: if len(sorted_dirs) > MAX_STORED_PATHS:
#remove 25 more than needed, to avoid doing it every time #remove 25 more than needed, to avoid doing it every time
@ -183,7 +183,7 @@ def match(path, pattern, ignore_case=False, only_end=False):
return (does_match, eaten_path) return (does_match, eaten_path)
def find_matches(dirs, patterns, result_list, ignore_case, max_matches, current_dir): def find_matches(dirs, patterns, result_list, ignore_case, max_matches, current_dir):
"""Find max_matches paths that match the pattern, """Find max_matches paths that match the pattern,
and add them to the result_list""" and add them to the result_list"""
for path, count in dirs: for path, count in dirs:
# Don't jump to where we alread are # Don't jump to where we alread are
@ -222,7 +222,7 @@ def shell_utility():
if ('-a', '') in optlist: 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[-1] != os.path.expanduser("~")): if(args[-1] != os.path.expanduser("~")):
dicadd(path_dict, decode(args[-1])) dicadd(path_dict, decode(args[-1]))
save(path_dict, dic_file) save(path_dict, dic_file)
elif ('--stat', '') in optlist: elif ('--stat', '') in optlist:
@ -241,7 +241,7 @@ def shell_utility():
import re import re
completion = False completion = False
#userchoice is i if the pattern is __pattern__i, otherwise -1 #userchoice is i if the pattern is __pattern__i, otherwise -1
userchoice = -1 userchoice = -1
results = [] results = []
if ('--completion', '') in optlist: if ('--completion', '') in optlist:
completion = True completion = True
@ -286,10 +286,10 @@ def shell_utility():
find_matches(dirs, patterns, results, False, max_matches, current_dir) find_matches(dirs, patterns, results, False, max_matches, current_dir)
# If not found, try ignoring case. # If not found, try ignoring case.
# On completion always show all results # On completion always show all results
if completion or not results: if completion or not results:
find_matches(dirs, patterns, results, find_matches(dirs, patterns, results,
ignore_case=True, ignore_case=True,
max_matches=max_matches, current_dir=current_dir) max_matches=max_matches, current_dir=current_dir)
# Keep the database to a reasonable size # Keep the database to a reasonable size
if not completion and clean_dict(dirs, path_dict): if not completion and clean_dict(dirs, path_dict):
save(path_dict, dic_file) save(path_dict, dic_file)
@ -298,7 +298,7 @@ def shell_utility():
else: quotes = "" else: quotes = ""
if userchoice != -1: if userchoice != -1:
if len(results) > userchoice-1 : if len(results) > userchoice-1 :
output(unico("%s%s%s") % (quotes,results[userchoice-1],quotes)) output(unico("%s%s%s") % (quotes,results[userchoice-1],quotes))
elif len(results) > 1 and completion: elif len(results) > 1 and completion:
output("\n".join(("%s%s%d%s%s" % (patterns[-1], output("\n".join(("%s%s%d%s%s" % (patterns[-1],

View File

@ -57,7 +57,7 @@ while true; do
esac esac
done done
if [[ $(whoami) != "root" ]] && ! ${local}; then if [[ ${UID} != 0 ]] && ! ${local}; then
echo "Please rerun as root or use the --local option." echo "Please rerun as root or use the --local option."
exit 1 exit 1
fi fi