samba-mirror/source4/scripting/bin/samba-tool

#!/usr/bin/env python

# Unix SMB/CIFS implementation.
# Copyright (C) Amitay Isaacs <amitay@gmail.com> 211
#
# 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/>.
#

import sys

# Find right direction when running from source tree
sys.path.insert(0, "bin/python")

from samba import netcmd
from samba.netcmd import Command, CommandError


class MainCommand(Command):
    """Main class for samba tool commands"""

    commands = {}

    def _run(self, myname, command=None, *args):
        if command in self.commands:
            return self.commands[command]._run(command, *args)
        print "Usage: samba-tool <command> [options]"
        print "Available commands:"
        cmds = self.commands.keys()
        cmds.sort()
        for cmd in cmds:
            print "    %-16s - %s" % (cmd, self.commands[cmd].description)
        if command in [None]:
            raise CommandError("You must specify a command")
        if command in ['help', '-h', '--help']:
            print "For more help on a specific command, please type: samba-tool <command> (-h|--help)"
            return 0
        raise CommandError("No such command '%s'" % command)

    def usage(self, myname, command=None, *args):
        if command is None or not command in self.commands:
            print "Usage: samba-tool %s (%s) [options]" % (myname,
                    " | ".join(self.commands.keys()))
        else:
            return self.commands[command].usage(*args)


class cmd_sambatool(MainCommand):
    """Samba Tool Commands"""
    commands = netcmd.commands


if __name__ == '__main__':
    cmd = cmd_sambatool()

    command = None
    args = ()

    if len(sys.argv) > 1:
        command = sys.argv[1]
        if len(sys.argv) > 2:
            args = sys.argv[2:]

    try:
        retval = cmd._run("samba-tool", command, *args)
        sys.exit(retval)
    except Exception, e:
        cmd.show_command_error(e)