51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
|
import argparse
|
||
|
|
import fnmatch
|
||
|
|
import os
|
||
|
|
import re
|
||
|
|
import sys
|
||
|
|
|
||
|
|
from voussoirkit import safeprint
|
||
|
|
from voussoirkit import spinal
|
||
|
|
|
||
|
|
def search(terms, match_any=False, do_regex=False, do_glob=False):
|
||
|
|
search_terms = [term.lower() for term in terms]
|
||
|
|
|
||
|
|
def term_matches(text, term):
|
||
|
|
return (
|
||
|
|
(term in text) or
|
||
|
|
(do_regex and re.search(term, text)) or
|
||
|
|
(do_glob and fnmatch.fnmatch(text, term))
|
||
|
|
)
|
||
|
|
|
||
|
|
anyall = any if match_any else all
|
||
|
|
|
||
|
|
generator = spinal.walk_generator(depth_first=False, yield_directories=True)
|
||
|
|
for filepath in generator:
|
||
|
|
basename = filepath.basename.lower()
|
||
|
|
if anyall(term_matches(basename, term) for term in search_terms):
|
||
|
|
safeprint.safeprint(filepath.absolute_path)
|
||
|
|
|
||
|
|
|
||
|
|
def search_argparse(args):
|
||
|
|
return search(
|
||
|
|
terms=args.search_terms,
|
||
|
|
do_glob=args.do_glob,
|
||
|
|
do_regex=args.do_regex,
|
||
|
|
match_any=args.match_any,
|
||
|
|
)
|
||
|
|
|
||
|
|
def main(argv):
|
||
|
|
parser = argparse.ArgumentParser()
|
||
|
|
|
||
|
|
parser.add_argument('search_terms', nargs='+', default=None)
|
||
|
|
parser.add_argument('--any', dest='match_any', action='store_true')
|
||
|
|
parser.add_argument('--regex', dest='do_regex', action='store_true')
|
||
|
|
parser.add_argument('--glob', dest='do_glob', action='store_true')
|
||
|
|
parser.set_defaults(func=search_argparse)
|
||
|
|
|
||
|
|
args = parser.parse_args(argv)
|
||
|
|
args.func(args)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main(sys.argv[1:])
|