U d@s:ddlZddlmZmZeeeefZGdddZdS)N)IterableUnionc@sveZdZdZdddeeedddZdd Zd d Zee d d dZ e deedddZ e deedddZ dS) GlobGroupaA set of patterns that candidate strings will be matched against. A candidate is composed of a list of segments separated by ``separator``, e.g. "foo.bar.baz". A pattern contains one or more segments. Segments can be: - A literal string (e.g. "foo"), which matches exactly. - A string containing a wildcard (e.g. "torch*", or "foo*baz*"). The wildcard matches any string, including the empty string. - A double wildcard ("**"). This matches against zero or more complete segments. Examples: ``torch.**``: matches ``torch`` and all its submodules, e.g. ``torch.nn`` and ``torch.nn.functional``. ``torch.*``: matches ``torch.nn`` or ``torch.functional``, but not ``torch.nn.functional``. ``torch*.**``: matches ``torch``, ``torchvision``, and all their submodules. A candidates will match the ``GlobGroup`` if it matches any of the ``include`` patterns and none of the ``exclude`` patterns. Args: include (Union[str, Iterable[str]]): A string or list of strings, each representing a pattern to be matched against. A candidate will match if it matches *any* include pattern exclude (Union[str, Iterable[str]]): A string or list of strings, each representing a pattern to be matched against. A candidate will be excluded from matching if it matches *any* exclude pattern. separator (str): A string that delimits segments in candidates and patterns. By default this is "." which corresponds to how modules are named in Python. Another common value for this is "/", which is the Unix path separator. .)exclude separator)includerrcCs:d|d|d|_t|||_t|||_||_dS)NzGlobGroup(include=z , exclude=))_dbgr _glob_listr rr)selfr rrrr7sz$GlobGroup.matches..c3s|]}| VqdSrrrrrrr7s)ranyr allr)r rrrrmatches5s $zGlobGroup.matcheselemsrcs.t|trt|gSfdd|DSdS)Ncsg|]}t|qSr)r _glob_to_re)rerrr @sz(GlobGroup._glob_list..) isinstancestrrr#r!rr%rr ;s zGlobGroup._glob_list)patternrcs4fdddfdd|D}t|S)Ncsdd|kr4|dkr*dtddStdn,tdddd|d DSdS) Nz**(z[^z]+)*z,** can only appear as an entire path segmentz]*css|]}t|VqdSr)reescape)rxrrrrMszAGlobGroup._glob_to_re..component_to_re..*)r+r, ValueErrorjoinsplit) componentr%rrcomponent_to_reFs z.GlobGroup._glob_to_re..component_to_rec3s|]}|VqdSrr)rc)r3rrrQsz(GlobGroup._glob_to_re..)r0r1r+compile)r)rresultr)r3rrr#Bs zGlobGroup._glob_to_reN)r)r)__name__ __module__ __qualname____doc__ GlobPatternr(rrrboolr staticmethodr r#rrrrrs  r)r+typingrrr(r<rrrrrs