main.py

import logging
import importlib.util
import pkgutil
from pathlib import Path


def get_available_solver(dir_name, baseclass):
    spec = importlib.util.find_spec(Path(dir_name).name)
    for (_, name, _) in pkgutil.iter_modules([dir_name]):
        try:
            importlib.import_module(f'.{name}', spec.name)
        except ModuleNotFoundError:
            pass
    _classes = {str(cls.__name__).lower(): cls for cls in baseclass.__subclasses__()}
    return [str(_cls).lower() for _cls in _classes], _classes


if __name__ == '__main__':
    import importlib
    import argparse

    cls_names, classes = get_available_solver("solver", importlib.import_module("solver.basesolver").BaseSolver)

    parser = argparse.ArgumentParser()

    # need to be
    parser.add_argument("input", help="input file")
    parser.add_argument("--output", help="output file")

    parser.add_argument('--debug', action='store_true', help='set debug level')
    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument("--solver", help="available solver", type=str, choices=cls_names, default="example")
    group.add_argument('--score', help="computes score and insights of given data set", action='store_true')

    args = parser.parse_args()

    if args.debug:
        logging.basicConfig(level=logging.DEBUG)
    else:
        logging.basicConfig(level=logging.INFO)
    LOGGER = logging.getLogger(__name__)

    if args.score:
        scoring = importlib.import_module("solver.scoring")
        score = scoring.compute_score(args.input, args.output)
        exit(0)

    # get the chosen solver
    Solver = classes[args.solver]
    # solver init with filepath
    solver = Solver(args.input)

    # solve the problem with given input
    success = solver.solve()
    if not success:
        raise RuntimeError("No solution found!")

    # maybe create a solution file
    if args.output:
        solver.write(args.output)