#!/usr/bin/env python3
"""
Generate a JSON state file with N items in each array.
Output goes to stdout.

Usage: ./generate_state.py <n>
"""

import json
import sys

def main():
    if len(sys.argv) != 2:
        print("Usage: generate_state.py <n>", file=sys.stderr)
        sys.exit(1)

    n = int(sys.argv[1])

    state = {
        "colors": [f"color_{i}" for i in range(1, n + 1)],
        "numbers": [f"number_{i}" for i in range(1, n + 1)],
        "animals": [f"animal_{i}" for i in range(1, n + 1)],
    }

    print(json.dumps(state))

if __name__ == "__main__":
    main()