Canonicalize content order in poetry.lock [package.extras]

lockfile.py

import tomlkit
import tomlkit.items
from pathlib import Path


def canonicalize_extras(entry):
    canonicalized = {}
    for key in sorted(entry.keys()):
        canonicalized[key] = sorted(entry[key])
    return canonicalized


def main():
    poetry_lock = Path(__file__).parent / "poetry.lock"
    parsed = tomlkit.parse(poetry_lock.read_text())

    for package in parsed["package"]:
        if "extras" in package:
            package["extras"] = canonicalize_extras(package["extras"])
            package["extras"].append(None, tomlkit.items.Whitespace("\n"))

    poetry_lock.write_text(tomlkit.dumps(parsed))


if __name__ == "__main__":
    main()