iiab/scripts/comp-vars.py

#!/usr/bin/env python
from __future__ import print_function

import argparse
import yaml

def main():
    args = parse_args()
    comp_var_file = args.var_file

    def_vars = read_yaml('/opt/iiab/iiab/vars/default_vars.yml')
    comp_vars = read_yaml(comp_var_file)

    list_extra_vars(comp_var_file, comp_vars, def_vars)
    list_changed(comp_var_file, comp_vars, def_vars)

    #min_vars = adm.read_yaml('/opt/iiab/iiab/vars/local_vars_min.yml')
    #med_vars = adm.read_yaml('/opt/iiab/iiab/vars/local_vars_medium.yml')
    #big_vars = adm.read_yaml('/opt/iiab/iiab/vars/local_vars_big.yml')

    #list_extra_vars('min_vars', min_vars, def_vars)
    #list_extra_vars('med_vars', med_vars, def_vars)
    #list_extra_vars('big_vars', big_vars, def_vars)

    #min_vars = remove_dups('min_vars', min_vars, def_vars)
    #med_vars = remove_dups('med_vars', med_vars, def_vars)
    #big_vars = remove_dups('big_vars', big_vars, def_vars)


def list_changed(file_name, var_dict, def_vars):
    # { k : v for k,v in d.items() if v} - copy only filtered to new and return (muy pythonic)
    print('Changed in ' + file_name)
    for var_name in var_dict:
        var_val = var_dict[var_name]
        if var_val != def_vars.get(var_name):
            print(var_name + ':', var_dict[var_name])

def remove_dups(dict_name, var_dict, def_vars):
    # { k : v for k,v in d.items() if v} - copy only filtered to new and return (muy pythonic)
    for var_name in var_dict.copy():
        var_val = var_dict[var_name]
        if var_val == def_vars.get(var_name):
            del var_dict[var_name]
    print(dict_name)
    for var_name in var_dict:
        print(var_name, var_dict[var_name])
    return var_dict

def list_extra_vars(dict_name, var_dict, def_vars):
    for var_name in var_dict:
        if def_vars.get(var_name, 'ReallyNone!@#$') == 'ReallyNone!@#$':
            print(dict_name, var_name, 'not in default')

# from adm_lib

def read_yaml(file_name, loader=yaml.SafeLoader):
    try:
        with open(file_name, 'r') as f:
            y = yaml.load(f, Loader=loader)
            return y
    except:
        raise

def parse_args():
    parser = argparse.ArgumentParser(description="Compare vars file to default_vars.")
    parser.add_argument("var_file", help="the full path to the local vars file you want to compare.")
    #parser.add_argument("-z", "--zim", type=str, help="zim to update (e.g. wikipedia_en_medicine_maxi). Leave blank for All.")

    return parser.parse_args()

# Now start the application
if __name__ == "__main__":

    # Run the main routine
    main()