#!/usr/bin/env python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
# not use this file except in compliance with the License. You may obtain
# a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# License for the specific language governing permissions and limitations
# under the License.

import argparse
import os
import yaml


def parse_opts():
    """Parse all of the conversion options."""

    parser = argparse.ArgumentParser(
        description="Convert a network V1 template to a V2 template."
    )
    parser.add_argument(
        "v1", metavar="<network_data.yaml>", help="Existing V1 Template."
    )

    return parser.parse_args()


def main():
    """Convert a network v1 template to the network v2 format.

    The V1 template will be converted to V2 format. The V1 template will be
    saved as a backup file before writing the V2 net-data format.
    """

    args = parse_opts()
    net_data_file = os.path.abspath(os.path.expanduser(args.v1))
    with open(net_data_file) as f:
        template_data = yaml.safe_load(f)

    new_template_data = list()
    for item in template_data:
        new_item = dict()
        item.pop("enabled", False)  # Drop unused var
        name = new_item["name"] = item.pop("name")
        name_lower = new_item["name_lower"] = item.pop(
            "name_lower", name.lower()
        )
        new_item["vip"] = item.pop("vip", False)
        new_item["mtu"] = item.pop("mtu", 1500)
        new_item["subnets"] = item.pop("subnets", dict())
        new_item["subnets"]["{}_subnet".format(name_lower)] = item
        new_template_data.append(new_item)

    os.rename(net_data_file, "{}.bak".format(net_data_file))
    try:
        # content is converted to yaml before opening the file.
        # This is done to ensure that we're not breaking any existing files
        # during the conversion process.
        dump_yaml = yaml.safe_dump(
            new_template_data, default_style=False, sort_keys=False
        )
    except Exception as e:
        print("Conversion could not be completed. Error:{}".format(str(e)))
    else:
        with open(net_data_file, "w") as f:
            f.write(dump_yaml)


if __name__ == "__main__":
    main()

Add tool to convert net-data from v1 to v2
Add tool to convert net-data from v1 to v2

This new tool will convert the old v1 net-data format to v2. This tool
will create a backup of the older net-data file so that its possible to
recover data should anything go wrong in the conversion. Upon completion
the tool will create the v2 file using the existing file format name.

Change-Id: I4b4d79b0f68287921fedb3ad7b29b50df5ae3f02
Signed-off-by: Kevin Carter <1d9a750a4fc7026fb02029431dc8a87d24487135@redhat.com>
