poetry2nix/generate.py

#!/usr/bin/env nix-shell
#! nix-shell -i python3 -p python3 poetry

# Run code generation used by poetry2nix

from poetry.packages.utils.utils import SUPPORTED_EXTENSIONS
import json

EXT_FILE = "extensions.json"

if __name__ == "__main__":
    with open(EXT_FILE, "w") as f:
        ext = set(ext.lstrip(".") for ext in SUPPORTED_EXTENSIONS)
        ext.add("egg")
        f.write(json.dumps(sorted(ext), indent=2) + "\n")