pathfinder-2-sqlite-MIRROR/data/yaml/cleanyaml.py

# THIS FILE SIMPLY LOADS THE YAML FILE INTO PYYAML AND THEN SPITS IT BACK OUT
# TO CLEAN UP AND ORDER ALL THE YAML

import yaml
import glob
import os



def main():

    # gets all files with a yaml extension in the directory
    yfiles = []
    for file in glob.glob("*.yaml"):
        yfiles.append(file)

    yfiles.sort()
    print(yfiles)

    for x in yfiles:
        print("Doing: {}".format(x))
        with open(x, 'r') as r:
            data = yaml.full_load(r)
            if x == "feats.yaml":
                for i in data['feat']:
                    # This is to clean out smart quotes that made it into the
                    # yaml file so it matches the requirements.yaml
                    if i['requirement'] != None:
                        # print("Before: {}".format(i['requirement']))
                        i['requirement'] = i['requirement'].replace('’', "'")
                        # print("After: {}".format(i['requirement']))
        final = yaml.safe_dump(data, allow_unicode=True)
        with open(x, 'w') as f:
            f.write(final)
        print("\tDone.")


if __name__ == "__main__":
    main()