Compare revisions

services:

  node0:

    image: local-comm

    image: image-dfl-bad

    container_name: node0

    environment:

      - RANK=0

  node1:

    image: local-comm

    image: image-dfl-bad

    container_name: node1

    environment:

      - RANK=1

  node2:

    image: local-comm

    image: image-dfl-bad

    container_name: node2

    environment:

      - RANK=2

  node3:

    image: local-comm

    image: image-dfl-bad

    container_name: node3

    environment:

      - RANK=3

import sys

import argparse

image = 'local-comm'

#image = 'image-dfl-bad'

def generate_docker_compose(num_nodes):

def generate_docker_compose(image_name, num_nodes):

    docker_compose_template = """\

version: '3.8'

services:

      - BASE_PORT=8000

      - IP_ADDRESS=node{i}

    volumes:

      - ./neighbors/neighbors{file_suffix}.txt:/app/neighbors.txt

      - ./neighbors/neighbors{i}.txt:/app/neighbors.txt

    networks:

      - my_network

    docker_compose_content = docker_compose_template

    for i in range(0, num_nodes):

        file_suffix = i

        docker_compose_content += service_template.format(i=i, file_suffix=file_suffix,img=image)

    for i in range(0, int(num_nodes)):

        docker_compose_content += service_template.format(i=i,img=image_name)

    docker_compose_content += network_section

        file.write(docker_compose_content)

if __name__ == "__main__":

    if len(sys.argv) != 2:

        print("Usage: python generate_docker_compose.py <num_nodes>")

        sys.exit(1)

    num_nodes = int(sys.argv[1])

    generate_docker_compose(num_nodes)

    parser = argparse.ArgumentParser(description="Upload PDF files to an S3 bucket")

    parser.add_argument("--image_name", help="The name of the image")

    parser.add_argument("--nb_nodes", help="The number of nodes")

    args = parser.parse_args()

    # image_name= args.image_name

    # if len(sys.argv) != 2:

    #     print("Usage: python generate_docker_compose.py <num_nodes>")

    #     sys.exit(1)

    # num_nodes = int(sys.argv[1])

    generate_docker_compose(args.image_name, args.nb_nodes)