#!/usr/bin/env python
#
# Copyright 2011-2015 Splunk, Inc.
#
# 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.

"""A simple command line interface for the Splunk REST APIs."""

import sys, os
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
from xml.etree import ElementTree

import splunklib.binding as binding

try:
    import utils
except ImportError:
    raise Exception("Add the SDK repository to your PYTHONPATH to run the examples "
                    "(e.g., export PYTHONPATH=~/splunk-sdk-python.")

# Invoke the url using the given opts parameters
def invoke(path, **kwargs):
    method = kwargs.get("method", "GET")
    return binding.connect(**kwargs).request(path, method=method)

def print_response(response):
    if response.status != 200:
        print "%d %s" % (response.status, response.reason)
        return
    body = response.body.read()
    try:
        root = ElementTree.XML(body)
        print ElementTree.tostring(root)
    except Exception:
        print body

def main():
    opts = utils.parse(sys.argv[1:], {}, ".splunkrc")
    for arg in opts.args: 
        print_response(invoke(arg, **opts.kwargs))

if __name__ == "__main__":
    main()

