mirror of
https://github.com/samba-team/samba.git
synced 2024-12-28 07:21:54 +03:00
211c9a5f85
Signed-off-by: Joe Guo <joeg@catalyst.net.nz> Reviewed-by: Andrew Bartlett <abartlet@samba.org> Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
231 lines
7.6 KiB
Python
231 lines
7.6 KiB
Python
# sites management
|
|
#
|
|
# Copyright Matthieu Patou <mat@matws.net> 2011
|
|
#
|
|
# 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/>.
|
|
#
|
|
|
|
from __future__ import print_function
|
|
from samba import sites, subnets
|
|
from samba.samdb import SamDB
|
|
import samba.getopt as options
|
|
from samba.auth import system_session
|
|
from samba.netcmd import (
|
|
Command,
|
|
CommandError,
|
|
SuperCommand,
|
|
Option,
|
|
)
|
|
|
|
|
|
class cmd_sites_create(Command):
|
|
"""Create a new site."""
|
|
|
|
synopsis = "%prog <site> [options]"
|
|
|
|
takes_args = ["sitename"]
|
|
|
|
takes_optiongroups = {
|
|
"sambaopts": options.SambaOptions,
|
|
"versionopts": options.VersionOptions,
|
|
"credopts": options.CredentialsOptions,
|
|
}
|
|
|
|
takes_options = [
|
|
Option("-H", "--URL", help="LDB URL for database or target server",
|
|
type=str, metavar="URL", dest="H"),
|
|
]
|
|
|
|
def run(self, sitename, H=None, sambaopts=None, credopts=None,
|
|
versionopts=None):
|
|
lp = sambaopts.get_loadparm()
|
|
creds = credopts.get_credentials(lp, fallback_machine=True)
|
|
samdb = SamDB(url=H, session_info=system_session(),
|
|
credentials=creds, lp=lp)
|
|
|
|
samdb.transaction_start()
|
|
try:
|
|
sites.create_site(samdb, samdb.get_config_basedn(), sitename)
|
|
samdb.transaction_commit()
|
|
except sites.SiteAlreadyExistsException as e:
|
|
samdb.transaction_cancel()
|
|
raise CommandError("Error while creating site %s, error: %s" %
|
|
(sitename, str(e)))
|
|
|
|
self.outf.write("Site %s created !\n" % sitename)
|
|
|
|
|
|
class cmd_sites_delete(Command):
|
|
"""Delete an existing site."""
|
|
|
|
synopsis = "%prog <site> [options]"
|
|
|
|
takes_args = ["sitename"]
|
|
|
|
takes_optiongroups = {
|
|
"sambaopts": options.SambaOptions,
|
|
"versionopts": options.VersionOptions,
|
|
"credopts": options.CredentialsOptions,
|
|
}
|
|
|
|
takes_options = [
|
|
Option("-H", "--URL", help="LDB URL for database or target server",
|
|
type=str, metavar="URL", dest="H"),
|
|
]
|
|
|
|
def run(self, sitename, H=None, sambaopts=None, credopts=None,
|
|
versionopts=None):
|
|
lp = sambaopts.get_loadparm()
|
|
creds = credopts.get_credentials(lp, fallback_machine=True)
|
|
samdb = SamDB(url=H, session_info=system_session(),
|
|
credentials=creds, lp=lp)
|
|
|
|
samdb.transaction_start()
|
|
try:
|
|
sites.delete_site(samdb, samdb.get_config_basedn(), sitename)
|
|
samdb.transaction_commit()
|
|
except sites.SiteException as e:
|
|
samdb.transaction_cancel()
|
|
raise CommandError(
|
|
"Error while removing site %s, error: %s" % (sitename, str(e)))
|
|
|
|
self.outf.write("Site %s removed!\n" % sitename)
|
|
|
|
|
|
class cmd_sites_subnet_create(Command):
|
|
"""Create a new subnet."""
|
|
synopsis = "%prog <subnet> <site-of-subnet> [options]"
|
|
takes_args = ["subnetname", "site_of_subnet"]
|
|
|
|
takes_optiongroups = {
|
|
"sambaopts": options.SambaOptions,
|
|
"versionopts": options.VersionOptions,
|
|
"credopts": options.CredentialsOptions,
|
|
}
|
|
|
|
takes_options = [
|
|
Option("-H", "--URL", help="LDB URL for database or target server",
|
|
type=str, metavar="URL", dest="H"),
|
|
]
|
|
|
|
def run(self, subnetname, site_of_subnet, H=None, sambaopts=None,
|
|
credopts=None, versionopts=None):
|
|
lp = sambaopts.get_loadparm()
|
|
creds = credopts.get_credentials(lp)
|
|
samdb = SamDB(url=H, session_info=system_session(),
|
|
credentials=creds, lp=lp)
|
|
|
|
samdb.transaction_start()
|
|
try:
|
|
subnets.create_subnet(samdb, samdb.get_config_basedn(), subnetname,
|
|
site_of_subnet)
|
|
samdb.transaction_commit()
|
|
except subnets.SubnetException as e:
|
|
samdb.transaction_cancel()
|
|
raise CommandError("Error while creating subnet %s: %s" %
|
|
(subnetname, e))
|
|
|
|
self.outf.write("Subnet %s created !\n" % subnetname)
|
|
|
|
|
|
class cmd_sites_subnet_delete(Command):
|
|
"""Delete an existing subnet."""
|
|
|
|
synopsis = "%prog <subnet> [options]"
|
|
|
|
takes_args = ["subnetname"]
|
|
|
|
takes_optiongroups = {
|
|
"sambaopts": options.SambaOptions,
|
|
"versionopts": options.VersionOptions,
|
|
"credopts": options.CredentialsOptions,
|
|
}
|
|
|
|
takes_options = [
|
|
Option("-H", "--URL", help="LDB URL for database or target server",
|
|
type=str, metavar="URL", dest="H"),
|
|
]
|
|
|
|
def run(self, subnetname, H=None, sambaopts=None, credopts=None,
|
|
versionopts=None):
|
|
lp = sambaopts.get_loadparm()
|
|
creds = credopts.get_credentials(lp)
|
|
samdb = SamDB(url=H, session_info=system_session(),
|
|
credentials=creds, lp=lp)
|
|
|
|
samdb.transaction_start()
|
|
try:
|
|
subnets.delete_subnet(samdb, samdb.get_config_basedn(), subnetname)
|
|
samdb.transaction_commit()
|
|
except subnets.SubnetException as e:
|
|
samdb.transaction_cancel()
|
|
raise CommandError("Error while removing subnet %s, error: %s" %
|
|
(subnetname, e))
|
|
|
|
self.outf.write("Subnet %s removed!\n" % subnetname)
|
|
|
|
|
|
class cmd_sites_subnet_set_site(Command):
|
|
"""Assign a subnet to a site."""
|
|
synopsis = "%prog <subnet> <site-of-subnet> [options]"
|
|
takes_args = ["subnetname", "site_of_subnet"]
|
|
|
|
takes_optiongroups = {
|
|
"sambaopts": options.SambaOptions,
|
|
"versionopts": options.VersionOptions,
|
|
"credopts": options.CredentialsOptions,
|
|
}
|
|
|
|
takes_options = [
|
|
Option("-H", "--URL", help="LDB URL for database or target server",
|
|
type=str, metavar="URL", dest="H"),
|
|
]
|
|
|
|
def run(self, subnetname, site_of_subnet, H=None, sambaopts=None,
|
|
credopts=None, versionopts=None):
|
|
lp = sambaopts.get_loadparm()
|
|
creds = credopts.get_credentials(lp)
|
|
samdb = SamDB(url=H, session_info=system_session(),
|
|
credentials=creds, lp=lp)
|
|
|
|
samdb.transaction_start()
|
|
try:
|
|
subnets.set_subnet_site(samdb, samdb.get_config_basedn(),
|
|
subnetname, site_of_subnet)
|
|
samdb.transaction_commit()
|
|
except subnets.SubnetException as e:
|
|
samdb.transaction_cancel()
|
|
raise CommandError("Error assigning subnet %s to site %s: %s" %
|
|
(subnetname, site_of_subnet, e))
|
|
|
|
print(("Subnet %s shifted to site %s" %
|
|
(subnetname, site_of_subnet)), file=self.outf)
|
|
|
|
|
|
class cmd_sites_subnet(SuperCommand):
|
|
"""Subnet management subcommands."""
|
|
subcommands = {
|
|
"create": cmd_sites_subnet_create(),
|
|
"remove": cmd_sites_subnet_delete(),
|
|
"set-site": cmd_sites_subnet_set_site(),
|
|
}
|
|
|
|
|
|
class cmd_sites(SuperCommand):
|
|
"""Sites management."""
|
|
subcommands = {}
|
|
subcommands["create"] = cmd_sites_create()
|
|
subcommands["remove"] = cmd_sites_delete()
|
|
subcommands["subnet"] = cmd_sites_subnet()
|