glob/path_glob.py

from glob import glob

from towhee.operator.base import PyOperator


class PathGlob(PyOperator):
    """
    Wrapper of glob.glob
    """
    def __init__(self, pathname, *, recursive=False):
        self._pattern = pathname
        self._recursive = recursive

    def __call__(self):
        return glob(self._pattern, recursive=self._recursive)