import os import os.path import sys from argparse import ArgumentParser from pkg_resources import iter_entry_points from requests.exceptions import ConnectionError from requests.exceptions import HTTPError from requests.exceptions import RequestException from synergy.client import keystone_v3 __author__ = "Lisa Zangrando" __email__ = "lisa.zangrando[AT]pd.infn.it" __copyright__ = """Copyright (c) 2015 INFN - INDIGO-DataCloud All Rights Reserved Licensed under the Apache License, Version 2.0; 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.""" COMMANDS_ENTRY_POINT = "synergy.commands" # used to discover Synergy commands def main(): try: parser = ArgumentParser(prog="synergy", epilog="Command-line interface to the" " OpenStack Synergy API.") # Global arguments parser.add_argument("--version", action="version", version="v1.5.3") parser.add_argument("--debug", default=False, action="store_true", help="print debugging output") parser.add_argument("--os-username", metavar="", default=os.environ.get("OS_USERNAME"), help="defaults to env[OS_USERNAME]") parser.add_argument("--os-password", metavar="", default=os.environ.get("OS_PASSWORD"), help="defaults to env[OS_PASSWORD]") parser.add_argument("--os-user-domain-id", metavar="", default=os.environ.get("OS_USER_DOMAIN_ID"), help="defaults to env[OS_USER_DOMAIN_ID]") parser.add_argument("--os-user-domain-name", metavar="", default=os.environ.get("OS_USER_DOMAIN_NAME"), help="defaults to env[OS_USER_DOMAIN_NAME]") parser.add_argument("--os-project-name", metavar="", default=os.environ.get("OS_PROJECT_NAME"), help="defaults to env[OS_PROJECT_NAME]") parser.add_argument("--os-project-id", metavar="", default=os.environ.get("OS_PROJECT_ID"), help="defaults to env[OS_PROJECT_ID]") parser.add_argument("--os-project-domain-id", metavar="", default=os.environ.get("OS_PROJECT_DOMAIN_ID"), help="defaults to env[OS_PROJECT_DOMAIN_ID]") parser.add_argument("--os-project-domain-name", metavar="", default=os.environ.get("OS_PROJECT_DOMAIN_NAME"), help="defaults to env[OS_PROJECT_DOMAIN_NAME]") parser.add_argument("--os-auth-url", metavar="", default=os.environ.get("OS_AUTH_URL"), help="defaults to env[OS_AUTH_URL]") parser.add_argument("--bypass-url", metavar="", dest="bypass_url", help="use this API endpoint instead of the " "Service Catalog") parser.add_argument("--os-cacert", metavar="", default=os.environ.get("OS_CACERT", None), help="Specify a CA bundle file to use in verifying" " a TLS (https) server certificate. Defaults " "to env[OS_CACERT]") subparser = parser.add_subparsers(help="commands", dest="command_name") commands = {} for entry in iter_entry_points(COMMANDS_ENTRY_POINT): command_class = entry.load() command = command_class() commands[entry.name] = command for command_name in sorted(commands.keys()): commands[command_name].configureParser(subparser) args = parser.parse_args(sys.argv[1:]) os_username = args.os_username os_password = args.os_password os_user_domain_id = args.os_user_domain_id os_user_domain_name = args.os_user_domain_name os_project_name = args.os_project_name os_project_domain_id = args.os_project_domain_id os_project_domain_name = args.os_project_domain_name os_auth_url = args.os_auth_url os_cacert = args.os_cacert bypass_url = args.bypass_url command_name = args.command_name token = None if not os_username: raise ValueError("'os-username' not defined!") if not os_password: raise ValueError("'os-password' not defined!") if not os_project_name: raise ValueError("'os-project-name' not defined!") if not os_auth_url: raise ValueError("'os-auth-url' not defined!") if not os_user_domain_name: os_user_domain_name = "default" if not os_project_domain_name: os_project_domain_name = "default" client = keystone_v3.KeystoneClient( auth_url=os_auth_url, username=os_username, password=os_password, ca_cert=os_cacert, user_domain_id=os_user_domain_id, user_domain_name=os_user_domain_name, project_name=os_project_name, project_domain_id=os_project_domain_id, project_domain_name=os_project_domain_name) token = client.authenticate() if bypass_url: synergy_url = bypass_url else: synergy_endpoint = client.getEndpoint("synergy") synergy_url = synergy_endpoint["url"] if command_name not in commands: print("Command %r not found!" % command_name) commands[command_name].setToken(token) commands[command_name].execute(synergy_url, args) except KeyboardInterrupt: print("Shutting down synergyclient") sys.exit(1) except ConnectionError as ex: print("Failed to establish a new connection to %s" % synergy_url) sys.exit(1) except HTTPError as ex: if ex.response._content: print("%s" % ex.response._content) else: print("%s" % ex.message) sys.exit(1) except RequestException as ex: print("%s" % ex.response._content) sys.exit(1) except Exception as ex: print(ex) sys.exit(1) if __name__ == "__main__": main()