|
|
@ -23,6 +23,7 @@ from __future__ import print_function |
|
|
|
|
|
|
|
from collections import namedtuple |
|
|
|
from functools import partial |
|
|
|
from itertools import chain |
|
|
|
# FIXME(ting|2013-12-17): fix imports for Python 3 compatability |
|
|
|
from itertools import ifilter |
|
|
|
from itertools import imap |
|
|
@ -84,7 +85,7 @@ def parse_environment(config): |
|
|
|
return config |
|
|
|
|
|
|
|
|
|
|
|
def eval_arguments(config): |
|
|
|
def parse_arguments(): |
|
|
|
"""Evaluate arguments and run appropriate logic, returning an error code.""" |
|
|
|
parser = ArgumentParser( |
|
|
|
description='Automatically jump to directory passed as an argument.', |
|
|
@ -119,49 +120,7 @@ def eval_arguments(config): |
|
|
|
'-v', '--version', action="version", version="%(prog)s " + |
|
|
|
VERSION, help='show version information') |
|
|
|
|
|
|
|
args = parser.parse_args() |
|
|
|
|
|
|
|
if args.add: |
|
|
|
add_path(config, args.add) |
|
|
|
return 0 |
|
|
|
|
|
|
|
# if args.complete: |
|
|
|
# config['match_cnt'] = 9 |
|
|
|
# config['ignore_case'] = True |
|
|
|
|
|
|
|
if args.decrease: |
|
|
|
try: |
|
|
|
print_entry(decrease_path(config, os.getcwdu(), args.decrease)) |
|
|
|
return 0 |
|
|
|
except OSError: |
|
|
|
print("Current directory no longer exists.", file=sys.stderr) |
|
|
|
return 1 |
|
|
|
|
|
|
|
if args.increase: |
|
|
|
try: |
|
|
|
print_entry(add_path(config, os.getcwdu(), args.increase)) |
|
|
|
return 0 |
|
|
|
except OSError: |
|
|
|
print("Current directory no longer exists.", file=sys.stderr) |
|
|
|
return 1 |
|
|
|
|
|
|
|
if args.purge: |
|
|
|
print("Purged %d entries." % purge_missing_paths(config)) |
|
|
|
return 0 |
|
|
|
|
|
|
|
if args.stat: |
|
|
|
print_stats(config) |
|
|
|
return 0 |
|
|
|
|
|
|
|
# default behavior, no optional arguments |
|
|
|
result = first(find_matches(config, args.directory)) |
|
|
|
if result: |
|
|
|
print(encode_local(result.path)) |
|
|
|
else: |
|
|
|
# always return something so the calling shell function has something |
|
|
|
# to `cd` to |
|
|
|
print(encode_local('.')) |
|
|
|
return 0 |
|
|
|
return parser.parse_args() |
|
|
|
|
|
|
|
|
|
|
|
def add_path(config, path, increment=10): |
|
|
@ -206,7 +165,7 @@ def detect_smartcase(needles): |
|
|
|
|
|
|
|
|
|
|
|
def find_matches(config, needles): |
|
|
|
"""Return an iterator to a matching entries.""" |
|
|
|
"""Return an iterator to matching entries.""" |
|
|
|
entriefy = lambda tup: Entry(*tup) |
|
|
|
not_cwd = lambda entry: entry.path != os.getcwdu() |
|
|
|
data = sorted( |
|
|
@ -221,10 +180,11 @@ def find_matches(config, needles): |
|
|
|
needles = map(sanitize, needles) |
|
|
|
ignore_case = detect_smartcase(needles) |
|
|
|
|
|
|
|
exact_matches = match_consecutive(needles, data, ignore_case) |
|
|
|
consecutive_matches = match_consecutive(needles, data, ignore_case) |
|
|
|
fuzzy_matches = match_fuzzy(needles, data, ignore_case) |
|
|
|
|
|
|
|
exists = lambda entry: os.path.exists(entry.path) |
|
|
|
return ifilter(exists, exact_matches) |
|
|
|
return ifilter(exists, chain(consecutive_matches, fuzzy_matches)) |
|
|
|
|
|
|
|
|
|
|
|
def match_consecutive(needles, haystack, ignore_case=False): |
|
|
@ -267,24 +227,10 @@ def match_consecutive(needles, haystack, ignore_case=False): |
|
|
|
return ifilter(found, haystack) |
|
|
|
|
|
|
|
|
|
|
|
def match_regex(needles, haystack, ignore_case=False): |
|
|
|
def match_fuzzy(needles, haystack, ignore_case=False): |
|
|
|
""" |
|
|
|
Performs an exact match by combining all arguments into a single regex |
|
|
|
expression and finding matches. |
|
|
|
|
|
|
|
For example: |
|
|
|
needles = ['qui', 'fox'] |
|
|
|
regex needle = r'.*qui.*fox.*' |
|
|
|
haystack = [ |
|
|
|
(path="foobar", 10.0), |
|
|
|
(path="The quick brown fox jumped over the lazy dog", 12.3)] |
|
|
|
|
|
|
|
result = [(path="The quick brown fox jumped over the lazy dog", 12.3)] |
|
|
|
""" |
|
|
|
regex_needle = '.*' + '.*'.join(needles) + '.*' |
|
|
|
regex_flags = re.IGNORECASE | re.UNICODE if ignore_case else re.UNICODE |
|
|
|
has_needle = lambda haystack: re.search(regex_needle, haystack.path, flags=regex_flags) |
|
|
|
return ifilter(has_needle, haystack) |
|
|
|
return [] |
|
|
|
|
|
|
|
|
|
|
|
def purge_missing_paths(config): |
|
|
@ -315,7 +261,42 @@ def print_stats(config): |
|
|
|
|
|
|
|
|
|
|
|
def main(): |
|
|
|
return eval_arguments(parse_environment(set_defaults())) |
|
|
|
config = parse_environment(set_defaults()) |
|
|
|
args = parse_arguments() |
|
|
|
|
|
|
|
if args.add: |
|
|
|
add_path(config, args.add) |
|
|
|
# elif args.complete: |
|
|
|
# config['match_cnt'] = 9 |
|
|
|
# config['ignore_case'] = True |
|
|
|
elif args.decrease: |
|
|
|
try: |
|
|
|
print_entry(decrease_path(config, os.getcwdu(), args.decrease)) |
|
|
|
except OSError: |
|
|
|
print("Current directory no longer exists.", file=sys.stderr) |
|
|
|
return 1 |
|
|
|
elif args.increase: |
|
|
|
try: |
|
|
|
print_entry(add_path(config, os.getcwdu(), args.increase)) |
|
|
|
except OSError: |
|
|
|
print("Current directory no longer exists.", file=sys.stderr) |
|
|
|
return 1 |
|
|
|
elif args.purge: |
|
|
|
print("Purged %d entries." % purge_missing_paths(config)) |
|
|
|
elif args.stat: |
|
|
|
print_stats(config) |
|
|
|
else: |
|
|
|
# default behavior, no optional arguments |
|
|
|
result = first(find_matches(load(config).iteritems(), args.directory)) |
|
|
|
|
|
|
|
if result: |
|
|
|
print(encode_local(result.path)) |
|
|
|
else: |
|
|
|
# always return something so the calling shell function has an |
|
|
|
# argument to `cd` to |
|
|
|
print(encode_local('.')) |
|
|
|
|
|
|
|
return 0 |
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|