summaryrefslogtreecommitdiffstats
path: root/ipalib/cli.py
blob: e0ba11f80e4f3264b0d1c251afb48f59018f2927 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
# Authors:
#   Jason Gerard DeRose <jderose@redhat.com>
#
# Copyright (C) 2008  Red Hat
# see file 'COPYING' for use and warranty information
#
# 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; version 2 only
#
# 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, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA

"""
Functionality for Command Line Inteface.
"""

import sys
import re


def to_cli(name):
    """
    Takes a Python identifier and transforms it into form suitable for the
    Command Line Interface.
    """
    assert isinstance(name, str)
    return name.replace('_', '-')


def from_cli(cli_name):
    """
    Takes a string from the Command Line Interface and transforms it into a
    Python identifier.
    """
    assert isinstance(cli_name, basestring)
    return cli_name.replace('-', '_')


class CLI(object):
    def __init__(self, api):
        self.__api = api

    def __get_api(self):
        return self.__api
    api = property(__get_api)

    def print_commands(self):
        for cmd in self.api.cmd:
            print to_cli(cmd.name)

    def __contains__(self, key):
        return from_cli(key) in self.api.cmd

    def __getitem__(self, key):
        return self.api.cmd[from_cli(key)]

    def run(self):
        if len(sys.argv) < 2:
            self.print_commands()
            print 'Usage: ipa COMMAND [OPTIONS]'
            sys.exit(2)
        cmd = sys.argv[1]
        if cmd not in self:
            self.print_commands()
            print 'ipa: ERROR: unknown command %r' % cmd
            sys.exit(2)
        self.run_cmd(cmd)

    def run_cmd(self, cmd):
        print self[cmd]