get_script_depends.py

#!/usr/bin/env python3

"""Get dependencies from inline script metadata
"""

import re
import tomllib

REGEX = r'(?m)^# /// (?P<type>[a-zA-Z0-9-]+)$\s(?P<content>(^#(| .*)$\s)+)^# ///$'

def read(script: str) -> dict | None:
    name = 'script'
    matches = list(
        filter(lambda m: m.group('type') == name, re.finditer(REGEX, script))
    )
    if len(matches) > 1:
        raise ValueError(f'Multiple {name} blocks found')
    elif len(matches) == 1:
        content = ''.join(
            line[2:] if line.startswith('# ') else line[1:]
            for line in matches[0].group('content').splitlines(keepends=True)
        )
        return tomllib.loads(content)
    else:
        return None


if __name__ == "__main__":
    import sys
    fname = sys.argv[1]
    try:
        with open(fname, "r") as f:
            script = f.read()
            metadata = read(script)
            if metadata is not None:
                args = []
                for dep in metadata["dependencies"]:
                    args.append("--with")
                    args.append(dep)
                print(" ".join(args))
    except FileNotFoundError:
        pass