diff options
Diffstat (limited to 'bin/ansible')
-rwxr-xr-x | bin/ansible | 25 |
1 files changed, 11 insertions, 14 deletions
diff --git a/bin/ansible b/bin/ansible index 085a8e0f4e..7420c2cc30 100755 --- a/bin/ansible +++ b/bin/ansible @@ -61,22 +61,19 @@ if __name__ == '__main__': try: display = Display() - if me == 'ansible-playbook': - from ansible.cli.playbook import PlaybookCLI as mycli - elif me == 'ansible': - from ansible.cli.adhoc import AdHocCLI as mycli - elif me == 'ansible-pull': - from ansible.cli.pull import PullCLI as mycli - elif me == 'ansible-doc': - from ansible.cli.doc import DocCLI as mycli - elif me == 'ansible-vault': - from ansible.cli.vault import VaultCLI as mycli - elif me == 'ansible-galaxy': - from ansible.cli.galaxy import GalaxyCLI as mycli - + sub = None try: + if me.find('-') != -1: + target = me.split('-') + if len(target) > 1: + sub = target[1] + myclass = "%sCLI" % sub.capitalize() + mycli = getattr(__import__("ansible.cli.%s" % sub, fromlist=[myclass]), myclass) + elif me == 'ansible': + from ansible.cli.adhoc import AdHocCLI as mycli + cli = mycli(sys.argv, display=display) - except NameError: + except (ImportError,NameError): raise AnsibleError("Ansible sub-program not implemented: %s" % me) cli.parse() |