def main()

in scheme_adapters/polyfit_adapter/polyfit_adapter.py [0:0]

• 21 lines of code
• 1 McCabe index (conditional complexity)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("-dp", "--damoos_path", required=True, help="DAMOOS path")
    parser.add_argument("-pfn", "--pickle_file_name", required=False, help="Pickle file name")
    parser.add_argument("-lb", "--lazybox_path", required=True, help="Lazybox path")
    parser.add_argument("-jp", "--json_path", required=True, help="JSON file path")
    parser.add_argument("-dm", "--damos_path", required=True, help="DAMOS path")

    args = vars(parser.parse_args())
    damoos_path = args["damoos_path"]
    lazybox_path = args["lazybox_path"]
    json_path = args["json_path"]
    damos_path = args["damos_path"]

    try:
        file_name = args["pickle_file_name"]
        polyfit = Polyfit(damoos_path, lazybox_path, damos_path)
        polyfit.fit(file_name)
        return 0
    except:
        pass

    polyfit = Polyfit(damoos_path, lazybox_path, damos_path)
    polyfit.find_best_scheme(json_path)