Class FilesCompleter

testing/test_argcomplete.py:39–69

File completer class, optionally takes a list of allowed extensions.

class FilesCompleter:
    """File completer class, optionally takes a list of allowed extensions."""

    def __init__(self, allowednames=(), directories=True):
        # Fix if someone passes in a string instead of a list
        if type(allowednames) is str:
            allowednames = [allowednames]

        self.allowednames = [x.lstrip("*").lstrip(".") for x in allowednames]
        self.directories = directories

    def __call__(self, prefix, **kwargs):
        completion = []
        if self.allowednames:
            if self.directories:
                files = _wrapcall(["bash", "-c", f"compgen -A directory -- '{prefix}'"])
                completion += [f + "/" for f in files]
            for x in self.allowednames:
                completion += _wrapcall(
                    ["bash", "-c", f"compgen -A file -X '!*.{x}' -- '{prefix}'"]
                )
        else:
            completion += _wrapcall(["bash", "-c", f"compgen -A file -- '{prefix}'"])

            anticomp = _wrapcall(["bash", "-c", f"compgen -A directory -- '{prefix}'"])

            completion = list(set(completion) - set(anticomp))

            if self.directories:
                completion += [f + "/" for f in anticomp]
        return completion


class TestArgComplete: