28 lines
585 B
Python
28 lines
585 B
Python
|
#!/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()
|