main.py

import argparse
import json
import logging
import os

from config.config import Configuration
from config.logging_configuration import LoggingConfiguration
from resource_assignment.resource_assignment_problem import ResourceAllocationProblem
from dragon_agent.dragon_agent import SDONode


def parse_arguments():

    # need to modify global configuration
    global SDO_NAME
    global SERVICE_BUNDLE
    global CONF_FILE
    global LOG_LEVEL
    global LOG_FILE

    # define arguments
    parser = argparse.ArgumentParser()
    parser.add_argument(
        'sdo_name',
        metavar='sdo-name',
        type=str,
        help='Name of the agent.',
    )
    parser.add_argument(
        'service',
        type=str,
        nargs='+',
        help='List of services.',
    )
    parser.add_argument(
        '-l',
        '--log-level',
        nargs='?',
        type=str,
        default='INFO',
        help='The logging level.'
    )
    parser.add_argument(
        '-o',
        '--log-on-file',
        action='store_true',
        help='If not given, log will be redirected on stdout.'
    )
    parser.add_argument(
        '-f',
        '--log-file',
        nargs='?',
        default=None,
        help='The log file name. If not given, log will be redirected on stdout.'
    )
    parser.add_argument(
        '-d',
        '--conf_file',
        nargs='?',
        default='config/default-config.ini',
        help='Configuration file.'
    )
    parser.add_argument(
        '-c',
        '--centralized',
        action='store_true',
        help='The log file name. If not given, log will be redirected on stdout.'
    )

    # parse arguments
    args = parser.parse_args()

    SDO_NAME = args.sdo_name
    SERVICE_BUNDLE = args.service
    CONF_FILE = args.conf_file
    LOG_LEVEL = args.log_level
    LOG_FILE = args.log_file
    if LOG_FILE is None and args.log_on_file:
        LOG_FILE = SDO_NAME + ".log"


if __name__ == "__main__":

    parse_arguments()
    configuration = Configuration(CONF_FILE)
    LoggingConfiguration(LOG_LEVEL, LOG_FILE).configure_log()

    rap = ResourceAllocationProblem()
    with open(configuration.RAP_INSTANCE) as rap_file:
        rap.parse_dict(json.loads(rap_file.read()))
    logging.info(rap)

    # SDO node
    sdo_node = SDONode(SDO_NAME, rap, SERVICE_BUNDLE)

    # Start scheduling
    strong, placement, rates = sdo_node.start_distributed_scheduling()

    placement_filename = configuration.RESULTS_FOLDER + "/placement_" + SDO_NAME + ".json"
    os.makedirs(os.path.dirname(placement_filename), exist_ok=True)
    with open(placement_filename, "w") as f:
        f.write(json.dumps(placement, indent=4))

    rates_filename = configuration.RESULTS_FOLDER + "/rates_" + SDO_NAME + ".json"
    with open(rates_filename, "w") as f:
        f.write(json.dumps(list(rates.items()), indent=4))

    utility_filename = configuration.RESULTS_FOLDER + "/utility_" + SDO_NAME + ".json"
    with open(utility_filename, "w") as f:
        f.write(str(sdo_node.sdo_bidder.private_utility))

    exit(sdo_node.sdo_bidder.private_utility)