generate_packages.py

from itertools import product
from pathlib import Path
from zipfile import ZipFile

GRID_SIZE = 8
PACKAGE_DIR = Path("packages")
WHEEL_FILE_CONTENTS = """
Wheel-Version: 1.0
Generator: nqueens (1.0.0)
Root-Is-Purelib: true
Tag: py3-none-any
""".strip()


def to_name(column: int) -> str:
    return f"queen_{chr(ord('a') + column)}"


def to_version(row: int) -> str:
    return f"{row + 1}"


def exclude(column: int, row: int) -> str:
    return f"{to_name(column)} != {to_version(row)}"


def get_exclusions(column: int, row: int, grid_size: int):
    return (
        exclude(c, r)
        for c, r in product(range(grid_size), repeat=2)
        if (
            # column exclusion (row exclusion is implicit)
            (r == row and c != column)
            # diagonal exclusion
            or ((d_c := abs(c - column)) == (d_r := abs(r - row)) and (d_c or d_r))
        )
    )


def generate_package(column: int, row: int, grid_size: int):
    name = to_name(column)
    version = to_version(row)
    metadata = [
        f"Name: {name}",
        f"Version: {version}",
        "Metadata-Version: 2.2",
        *(f"Requires-Dist: {d}" for d in get_exclusions(column, row, grid_size)),
    ]
    filename = f"{name}-{version}-py3-none-any.whl"

    # Write the wheel
    with ZipFile(PACKAGE_DIR.joinpath(filename), "w") as writer:
        writer.writestr(f"{name}-{version}.dist-info/METADATA", "\n".join(metadata))
        writer.writestr(f"{name}-{version}.dist-info/WHEEL", WHEEL_FILE_CONTENTS)
        # Not checked anyway
        # record = f"{name}-{version}.dist-info/METADATA,,"
        # record += f"{name}-{version}.dist-info/WHEEL,,"
        # record += f"{name}-{version}.dist-info/RECORD,,"
        writer.writestr(f"{name}-{version}.dist-info/RECORD", "")


def main():
    PACKAGE_DIR.mkdir(exist_ok=True, parents=True)
    for column, row in product(range(GRID_SIZE), repeat=2):
        generate_package(column, row, GRID_SIZE)


if __name__ == "__main__":
    main()