graphMaker.py

import sys
import random


def generate_symmetric_random_matrix(size):
    matrix = [[0 for _ in range(size)] for _ in range(size)]
    for i in range(size):
        for j in range(i + 1, size):
            value = random.randint(0, 1)
            matrix[i][j] = value
            matrix[j][i] = value
    return matrix


def print_usage():
    print(
        "Usage: python script_name.py <number_of_graphs> <default_size> [<size_of_graph_1> <size_of_graph_2> ...]"
    )


args = sys.argv[1:]


if len(args) < 2:
    print_usage()
    sys.exit(1)

num_graphs = int(args[0])
default_size = int(args[1])
sizes = [int(arg) for arg in args[2:]] + [default_size] * (num_graphs - len(args) + 2)

output_string = f"{num_graphs}\n"
for size in sizes[:num_graphs]:
    matrix = generate_symmetric_random_matrix(size)
    output_string += (
        f"{size}\n"
        + "\n".join(" ".join(str(cell) for cell in row) for row in matrix)
        + "\ninfo\n\n"
    )

filename = f"Graph{num_graphs}_{default_size}.txt"
with open(filename, "w") as file:
    file.write(output_string)

print(f"Output written to {filename}")