rewrite_uparams.py

#!/usr/bin/env python3
# encoding: utf-8

import csv
import sys
import re
import pprint
import os


def main():
    """
    Rewrites uparams.py, thereby checking for duplicates and sorting the dictionary.
    Note:
        uparams.py will be overwritten, if you are not sure, if your changes to uparams.py are correct, you can use another output_file_name

    Usage:
        ./rewrite_uparams.py <output_file_name(optional)>
    """

    input_file_name = os.path.join("ursgal", "uparams.py")
    try:
        output_file_name = sys.argv[1]
    except:
        output_file_name = input_file_name

    print(
        """
        Testing params dict for duplicates
    """
    )

    params = set()
    duplicates = []
    with open(input_file_name, "r") as input_file:
        for line in input_file:
            if line.startswith("    '") and line[4] == "'":
                hit = re.match("    '.{1,}'", line)
                param = hit.group(0).strip()
                if param not in params:
                    params.add(param)
                else:
                    duplicates.append(param)
    if duplicates == []:
        print("no duplicates were found")
    else:
        print("remove duplicates for:", duplicates)
        sys.exit(1)

    print(
        """
        Rewriting params dict
    """
    )
    from ursgal.uparams import ursgal_params as urgsal_dict

    print(len(urgsal_dict.keys()), " parameters")
    output_file = open(output_file_name, "w")
    print("ursgal_params = {", file=output_file)
    for param in sorted(urgsal_dict.keys()):
        # print(param)
        # if param != 'force':
        #     continue
        # print(urgsal_dict[param])
        print("""    '{0}' : {1}""".format(param, "{"), file=output_file)
        for k, v in sorted(urgsal_dict[param].items()):
            # print(k, type(v))
            if k == "description":
                print(
                    """        '{0}' : """.format(k),
                    repr(v.strip()),
                    ",",
                    file=output_file,
                )
            elif type(v) == str:
                print("""        '{0}' : "{1}",""".format(k, v), file=output_file)
            elif type(v) == dict:
                print("""        '{0}' : {1}""".format(k, "{"), file=output_file)
                for k2, v2 in sorted(urgsal_dict[param][k].items()):
                    if type(v2) == dict:
                        print(
                            """            '{0}' : {1}""".format(k2, "{"),
                            file=output_file,
                        )
                        for k3, v3 in sorted(urgsal_dict[param][k][k2].items()):
                            if k3 in [True, False]:
                                if type(v3) != str:
                                    print(
                                        """                {0} : {1},""".format(k3, v3),
                                        file=output_file,
                                    )
                                else:
                                    print(
                                        """                {0} : '{1}',""".format(
                                            k3, v3
                                        ),
                                        file=output_file,
                                    )
                            elif type(v3) != str:
                                print(
                                    """                '{0}' : {1},""".format(k3, v3),
                                    file=output_file,
                                )
                            else:
                                print(
                                    """                '{0}' : '{1}',""".format(k3, v3),
                                    file=output_file,
                                )
                        print("            },", file=output_file)
                    elif type(v2) != str:
                        print(
                            """            '{0}' : {1},""".format(k2, v2),
                            file=output_file,
                        )
                    else:
                        print(
                            """            '{0}' : '{1}',""".format(k2, v2),
                            file=output_file,
                        )
                print("        },", file=output_file)
            elif type(v) == list:
                print("""        '{0}' : {1}""".format(k, "["), file=output_file)
                for elem in sorted(v):
                    print("""            '{0}',""".format(elem), file=output_file)
                print("        ],", file=output_file)
            else:
                print("""        '{0}' : {1},""".format(k, v), file=output_file)
        # sys.exit(1)
        # print('''        'trigger_rerun' : True,''', file=output_file)
        print("    },", file=output_file)
    print("}", file=output_file)
    output_file.close()


if __name__ == "__main__":
    main()