omdev 0.0.0.dev99__py3-none-any.whl → 0.0.0.dev101__py3-none-any.whl

omdev/findmagic.py

@@ -1,91 +0,0 @@
-#!/usr/bin/env python3
-# @omlish-lite
-# @omlish-script
-import os.path
-import re
-import typing as ta
-
-
-def compile_magic_pat(m: str) -> re.Pattern:
-    return re.compile('^' + re.escape(m) + r'($|(\s.*))')
-
-
-def find_magic(
-        roots: ta.Sequence[str],
-        magics: ta.Sequence[str],
-        exts: ta.Sequence[str],
-        *,
-        py: bool = False,
-) -> ta.Iterator[str]:
-    if isinstance(roots, str):
-        raise TypeError(roots)
-    if isinstance(magics, str):
-        raise TypeError(magics)
-    if isinstance(exts, str):
-        raise TypeError(exts)
-
-    if not magics:
-        raise Exception('Must specify magics')
-    if not exts:
-        raise Exception('Must specify extensions')
-
-    pats = [compile_magic_pat(m) for m in magics]
-
-    for root in roots:
-        for dp, dns, fns in os.walk(root):  # noqa
-            for fn in fns:
-                if not any(fn.endswith(f'.{x}') for x in exts):
-                    continue
-
-                fp = os.path.join(dp, fn)
-                try:
-                    with open(fp) as f:
-                        src = f.read()
-                except UnicodeDecodeError:
-                    continue
-
-                if not any(
-                        any(pat.fullmatch(l) for pat in pats)
-                        for l in src.splitlines()
-                ):
-                    continue
-
-                if py:
-                    if fn == '__init__.py':
-                        out = dp.replace(os.sep, '.')
-                    elif fn.endswith('.py'):
-                        out = fp[:-3].replace(os.sep, '.')
-                    else:
-                        out = fp
-                else:
-                    out = fp
-                yield out
-
-
-# @omlish-manifest
-_CLI_MODULE = {'$omdev.cli.types.CliModule': {
-    'cmd_name': 'py/findmagic',
-    'mod_name': __name__,
-}}
-
-
-if __name__ == '__main__':
-    def _main(argv=None) -> None:
-        import argparse
-
-        arg_parser = argparse.ArgumentParser()
-        arg_parser.add_argument('--ext', '-x', dest='exts', action='append')
-        arg_parser.add_argument('--magic', '-m', dest='magics', action='append')
-        arg_parser.add_argument('--py', action='store_true')
-        arg_parser.add_argument('roots', nargs='*')
-        args = arg_parser.parse_args(argv)
-
-        for out in find_magic(
-            roots=args.roots,
-            magics=args.magics,
-            exts=args.exts,
-            py=args.py,
-        ):
-            print(out)
-
-    _main()