Subversion Repositories radiohdl

#!/usr/bin/env python3
###############################################################################
#
# Copyright (C) 2018
# ASTRON (Netherlands Institute for Radio Astronomy) <http://www.astron.nl/>
# P.O.Box 2, 7990 AA Dwingeloo, The Netherlands
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# $Id$
#
###############################################################################
import sys
import os
from os.path import expandvars, dirname, basename
from argparse import ArgumentParser
from configfile import ConfigFile
 
if __name__ == '__main__':
    # setup parser and parse the arguments.
    argparser = ArgumentParser(description="Options and arguments for constructing an 'export' command for shell based on the content of a hdltool config file.")
    argparser.add_argument('configfile', help="Filename like 'hdl_buildset_<boardtype>.cfg'")
    argparser.add_argument('keynames', help="Name(s) of the key(s) to show the value of. Use comma to seperate multiple keys.")
    argparser.add_argument('--varnames', help="Name(s) of the environment variable(s) the keys are mapped to. Default the keynames in capitals are used as environment variable names.")
    argparser.add_argument('--is-group', help="The keynames refer to groups of environment variables that must be set.",
                           action="store_true")
    argparser.add_argument('--optional', help="The keynames are optional. When them do not exist not error is generated",
                           action="store_true")
    args = argparser.parse_args()
 
    # resolve full name of configfile and force it to be explicit absolute or relative.
    full_configfile_name = expandvars(args.configfile)
    if full_configfile_name[0] != '/':
        full_configfile_name = "./" + full_configfile_name
    # read the file
    cfg_info = ConfigFile(full_configfile_name)
    cfg_info.resolve_key_references()
 
    # parse the keys if they are no group references
    if not args.is_group:
        # setup key- and variable- names
        keys = args.keynames.split(',')
        if args.varnames:
            env_vars = args.varnames.split(',')
            if len(keys) != len(env_vars):
                argparser.error("Number of variable names must match the number of keys.")
        else:
            env_vars = []
            for key in keys:
                env_vars.append(key.upper())
 
        # finally construct an export command for the key value pairs.
        for idx, key in enumerate(keys):
            print("export {}='{}'\n".format(env_vars[idx],
                                            os.path.expandvars(cfg_info.get_value(key, must_exist=not(args.optional)))))
        sys.exit(0)
 
    # Each key contains key-value pairs that must be exported in stead of a value
    if args.varnames:
        argparser.error("The option --varnames can not be used in combination with the option --is-group.")
 
    # print("args.keynames=%s" % str(args.keynames))
    keys = args.keynames.split(',')
    for key in keys:
        kv_pairs = cfg_info.get_value(key, must_exist=not(args.optional))
        # currently 'by definition' the value we got has the format: <key> <value> [<key> <value> [...]]
        # check we have an even number of items
        # print("KV_pairs={}".format(kv_pairs))
        if kv_pairs is None:
            continue
        items = kv_pairs.split()
        if len(items) % 2:
            argparser.error("Key '{}' should contain an even number of items ({}).".format(key, items))
        for idx in range(0, len(items)//2, 2):
            print("export {}='{}'\n".format(items[idx].upper(), os.path.expandvars(items[idx+1])))