#!/usr/bin/env python

# THIS FILE IS PART OF THE CYLC SUITE ENGINE.
# Copyright (C) 2008-2017 NIWA
#
# 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/>.
"""Suite Info API test, get_graph_raw."""

import json
import os
import sys

from cylc.network.suite_info_client import SuiteInfoClient


def main():
    kwargs = {
        'start_point_string': None,
        'stop_point_string': None,
        'group_nodes': None,
        'ungroup_nodes': None,
        'ungroup_recursive': False,
        'group_all': False,
        'ungroup_all': False}
    for item in sys.argv[1:]:
        key, value = item.split('=', 1)
        kwargs[key] = value
    client = SuiteInfoClient(os.environ['CYLC_SUITE_NAME'])
    print json.dumps(client.get_info('get_graph_raw', **kwargs), indent=4)


if __name__ == "__main__":
    main()
