def test_getcreate_host():
interface.create_host(host="host10")
hostobj = interface.get_host("host10")
assert str(hostobj) == "host10"
assert interface.get_host(None) is None
assert_raises(interface.InexistantObjectError,
interface.get_host, "inexistant")
interface.create_pool("pool10", "10.10.0.0/24", ["cat10"])
interface.create_host(host="host10-2", pool=interface.get_pool("pool10"))
assert(interface.get_host("host10-2").address_set.all()[0].pool.name
== "pool10")
interface.create_host(host="host10-3",
pool=interface.get_pool(category="cat10"))
assert(interface.get_host("host10-3").address_set.all()[0].pool.name
== "pool10")
assert_raises(interface.DuplicateObjectError, interface.create_host,
"host10-3")
interface.create_host(host="host10-4", mac="mac10-4", category="cat10")
assert(interface.get_host("host10-4").category == "cat10")
assert(Address.objects.get(host__name="host10-4").pool.name == "pool10")
assert(interface.get_host("host10-4").address_set.all()[0].macaddr
== "mac10-4")
interface.create_host(host="host10-6",
pool=Pool.objects.get(name="pool10"), address="10.10.0.142")
assert Address.objects.get(host__name="host10-6").addr == "10.10.0.142"
interface.create_host(host="host10-5", pool=interface.get_pool("pool10"),
mac="mac10-5")
addrobj = interface.get_host("host10-5").address_set.all()[0]
assert addrobj.addr == "10.10.0.3" and addrobj.macaddr == "mac10-5"
interface.create_host(host="host10-15", pool=interface.get_pool("pool10"),
alias=["alias10-1", "alias10-2"])
hostobj = interface.get_host("host10-15")
assert(len(hostobj.alias_set.all()) == 2 and
hostobj.alias_set.all()[1].name == "alias10-2")
interface.create_pool("pool10-2", "172.16.102.0/24", ["cat10"])
interface.create_host(host="host10-25", address="172.16.102.234")
assert(Address.objects.get(host__name="host10-25").pool.name == "pool10-2")
interface.create_host(host="host10-30",
pool=interface.get_pool("pool10-2"), serial="serialnum",
inventory="inventorynum", duration=42)
hostobj = interface.get_host("host10-30")
assert(hostobj.serial == "serialnum"
and hostobj.inventory == "inventorynum" and
Address.objects.get(host__name="host10-30").duration
> Address.objects.get(host__name="host10-30").date + datetime.timedelta(days=41))
interface.create_host(host="host10-40", pool=interface.get_pool("pool10"), nodns=True)
hostobj = interface.get_host("host10-40")
assert hostobj.nodns
def test_modify():
interface.create_pool("pool30-1", "10.30.1.0/24")
assert interface.get_pool("pool30-1").category == ""
interface.modify(["pool30-1"], category=["cat30-1"])
assert interface.get_pool("pool30-1").category == "cat30-1"
interface.modify(["pool30-1"], newname="pool30-2")
assert(Pool.objects.filter(name="pool30-2")
and interface.get_pool("pool30-2").category == "cat30-1")
interface.create_host(host="host30-1", pool=interface.get_pool("pool30-2"),
mac="foobar")
assert Address.objects.get(host__name="host30-1").macaddr == "foobar"
interface.modify(host="host30-1", mac="mac30-1")
assert Address.objects.get(host__name="host30-1").macaddr == "mac30-1"
interface.modify(host="host30-1", newname="host30-2")
assert(Host.objects.filter(name="host30-2")
and Address.objects.get(host__name="host30-2").macaddr == "mac30-1")
hostobj = interface.get_host("host30-2")
assert len(hostobj.alias_set.all()) == 0
assert not hostobj.nodns
interface.modify(host="host30-2", alias=["alias30-1", "alias30-2"], nodns=True)
hostobj = interface.get_host("host30-2")
assert(len(hostobj.alias_set.all()) == 2
and hostobj.alias_set.all()[1].name == "alias30-2" and hostobj.nodns)
assert_raises(interface.MissingParameterError,
interface.modify, ["pool30-2"])
assert_raises(interface.MissingParameterError,
interface.modify, None, "host30-2")
assert_raises(interface.InexistantObjectError, interface.modify, None)
interface.create_host(host="host30-3", pool=interface.get_pool("pool30-2"),
address="10.30.1.142", mac="foobar")
interface.modify(host="host30-3", address="10.30.1.142", mac="barfoo")
assert Address.objects.get(host__name="host30-3").macaddr == "barfoo"
interface.create_host(host="host30-4")
interface.modify(host="host30-4", mac="imac")
assert Address.objects.get(host__name="host30-4").macaddr == "imac"
hostobj = interface.get_host("host30-4")
assert(not (hostobj.serial or hostobj.inventory or
Address.objects.get(host__name=hostobj.name).duration))
interface.modify(host=hostobj.name, serial="srlnm", inventory="invtnm")
hostobj = interface.get_host("host30-4")
assert hostobj.serial == "srlnm" and hostobj.inventory == "invtnm"
addrobj = Address.objects.get(addr="10.30.1.0")
assert not addrobj.comment
interface.modify(address="10.30.1.0", comment="The quick brown fox...")
addrobj = Address.objects.get(addr="10.30.1.0")
assert addrobj.comment == "The quick brown fox..."
def host_info(request, host_name):
"""Manipulate or show a host in the database."""
data = request_data(request)
try:
host = interface.get_host(host_name)
except interface.InexistantObjectError:
return error_404(request)
if request.method == "PUT":
category = data.get("category")
if category:
category = [category]
alias = data.get("alias")
if not alias:
alias = ""
alias = alias.replace(", ", ",")
if alias:
alias = alias.split(",")
mac = data.get("macaddr")
try:
if mac:
mac = mac.lower()
if not re.match("^[a-f0-9]{2}(:[a-f0-9]{2}){5}$", mac):
return error_view(
request, 400, _("Invalid MAC"),
_("The format of the given MAC address is invalid :"
" %(mac)s.") % {"mac": mac})
interface.modify(host=host_name,
mac=data.get("macaddr"),
newname=data.get("newname"),
category=category,
alias=alias,
serial=data.get("serial"),
inventory=data.get("inventory"),
nodns=((data.get("nodns") == "on") != host.nodns),
clearalias=data.get("clearalias") == "on")
if data.get("owner"):
interface.set_prop("owner", data.get("owner"), host=host)
except interface.MissingParameterError:
return error_view(
request, 400, _("Missing information"),
_("You must specify the new values to modify an object."))
#annomalie espace dans le name host
except interface.PropertyFormatError:
return error_view(
request, 400, _("Format invalid"),
_("You must specify a valid host name or alias without space, special character etc."
))
#anomalie9
except interface.DuplicateObjectError, e:
return error_view(request, 409, _("Host or alias already exists"),
e.args[0])
#fin anomalie9
except interface.InexistantObjectError:
return error_404(request)
def _list_hosts(args):
"""List all the hosts in the database."""
for host in args.host:
if host:
try:
hostobj = interface.get_host(host)
except interface.InexistantObjectError:
logging.error("Could not find a host name: " + host)
sys.exit(1)
# mac address(es)
hoststr = str(hostobj)
if models.Address.objects.filter(host=hostobj).exclude(macaddr=""):
hoststr = hoststr + ", mac: "
for addrobj in models.Address.objects.filter(
host=hostobj).exclude(macaddr=""):
hoststr = hoststr + addrobj.macaddr + ", "
hoststr = hoststr[:-2] # strip the last comma
print "Host " + hoststr
if hostobj.serial:
print("Serial number: " + hostobj.serial)
if hostobj.inventory:
print("Inventory number: " + hostobj.inventory)
if hostobj.category:
print("Category: " + hostobj.category)
if hostobj.alias_set.count() > 0:
print(
"Alias: " + ", ".join([
alias.name for alias in list(hostobj.alias_set.all())
]))
if hostobj.nodns:
print("NODNS")
for addr in models.Address.objects.filter(host=hostobj).exclude(
addr="").order_by("addr"):
if addr.pool:
print("Address " + str(addr) + " (pool: " +
addr.pool.name + ")")
else:
print "Address " + str(addr)
for prop in models.Property.objects.filter(host=hostobj):
print str(prop)
else:
for hostobj in models.Host.objects.all().order_by("name"):
msg = str(hostobj)
if hostobj.alias_set.all():
msg += " (" + ", ".join(
[str(alias)
for alias in hostobj.alias_set.all()]) + ")"
if hostobj.category:
msg += ", category: " + hostobj.category
for addr in hostobj.address_set.all():
msg += ", " + str(addr)
print(msg)
def _list_hosts(args):
"""List all the hosts in the database."""
for host in args.host:
if host:
try:
hostobj = interface.get_host(host)
except interface.InexistantObjectError:
logging.error("Could not find a host name: " + host)
sys.exit(1)
# mac address(es)
hoststr = str(hostobj)
if models.Address.objects.filter(host=hostobj).exclude(macaddr=""):
hoststr = hoststr + ", mac: "
for addrobj in models.Address.objects.filter(
host=hostobj).exclude(macaddr=""):
hoststr = hoststr + addrobj.macaddr + ", "
hoststr = hoststr[:-2] # strip the last comma
print "Host " + hoststr
if hostobj.serial:
print("Serial number: " + hostobj.serial)
if hostobj.inventory:
print("Inventory number: " + hostobj.inventory)
if hostobj.category:
print("Category: " + hostobj.category)
if hostobj.alias_set.count() > 0:
print("Alias: " + ", ".join(
[alias.name for alias in list(hostobj.alias_set.all())]))
if hostobj.nodns:
print("NODNS")
for addr in models.Address.objects.filter(host=hostobj).exclude(
addr="").order_by("addr"):
if addr.pool:
print ("Address " + str(addr) + " (pool: " +
addr.pool.name + ")")
else:
print "Address " + str(addr)
for prop in models.Property.objects.filter(host=hostobj):
print str(prop)
else:
for hostobj in models.Host.objects.all().order_by("name"):
msg = str(hostobj)
if hostobj.alias_set.all():
msg += " (" + ", ".join([str(alias) for alias
in hostobj.alias_set.all()]) + ")"
if hostobj.category:
msg += ", category: " + hostobj.category
for addr in hostobj.address_set.all():
msg += ", " + str(addr)
print(msg)
def host_info(request, host_name):
"""Manipulate or show a host in the database."""
data = request_data(request)
try:
host = interface.get_host(host_name)
except interface.InexistantObjectError:
return error_404(request)
if request.method == "PUT":
category = data.get("category")
if category:
category = [category]
alias = data.get("alias")
if not alias:
alias = ""
alias = alias.replace(", ", ",")
if alias:
alias = alias.split(",")
mac = data.get("macaddr")
try:
if mac:
mac = mac.lower()
if not re.match("^[a-f0-9]{2}(:[a-f0-9]{2}){5}$", mac):
return error_view(request, 400, _("Invalid MAC"),
_("The format of the given MAC address is invalid :"
" %(mac)s.") % {"mac": mac})
interface.modify(host=host_name,
mac=data.get("macaddr"),
newname=data.get("newname"),
category=category,
alias=alias,
serial=data.get("serial"),
inventory=data.get("inventory"),
nodns=((data.get("nodns") == "on") != host.nodns),
clearalias=data.get("clearalias") == "on")
if data.get("owner"):
interface.set_prop("owner", data.get("owner"), host=host)
except interface.MissingParameterError:
return error_view(request, 400, _("Missing information"),
_("You must specify the new values to modify an object."))
#annomalie espace dans le name host
except interface.PropertyFormatError:
return error_view(request, 400, _("Format invalid"),
_("You must specify a valid host name or alias without space, special character etc."))
#anomalie9
except interface.DuplicateObjectError, e:
return error_view(request, 409, _("Host or alias already exists"), e.args[0])
#fin anomalie9
except interface.InexistantObjectError:
return error_404(request)
def get(args):
"""Get new address from pool."""
pool = None
if args.pool_name:
try:
pool = interface.get_pool(args.pool_name[0])
except (interface.InexistantObjectError,
interface.DuplicateObjectError) as exc:
logging.error(str(exc))
sys.exit(1)
if args.host:
for host in args.host:
try:
hostobj = interface.get_host(host)
except interface.InexistantObjectError as exc:
logging.error(str(exc))
sys.exit(1)
addr = None
if not args.address:
args.address = [None]
try:
if args.category:
args.category = args.category[0]
addr = interface.allocate_address(pool, hostobj,
args.address[0], args.random,
args.category, args.duration)
del args.address[0]
except (interface.MissingParameterError,
models.FullPoolError) as exc:
logging.error(str(exc))
except (models.AddressNotInPoolError,
models.AddressNotAvailableError) as exc:
logging.warn(str(exc))
if addr:
print("Assigned " + str(addr) + " to host " + host)
else:
logging.error("Missing host to allocate address to.")
sys.exit(1)
def get(args):
"""Get new address from pool."""
pool = None
if args.pool_name:
try:
pool = interface.get_pool(args.pool_name[0])
except (interface.InexistantObjectError,
interface.DuplicateObjectError) as exc:
logging.error(str(exc))
sys.exit(1)
if args.host:
for host in args.host:
try:
hostobj = interface.get_host(host)
except interface.InexistantObjectError as exc:
logging.error(str(exc))
sys.exit(1)
addr = None
if not args.address:
args.address = [None]
try:
if args.category:
args.category = args.category[0]
addr = interface.allocate_address(pool, hostobj,
args.address[0], args.random, args.category, args.duration)
del args.address[0]
except (interface.MissingParameterError,
models.FullPoolError) as exc:
logging.error(str(exc))
except (models.AddressNotInPoolError,
models.AddressNotAvailableError) as exc:
logging.warn(str(exc))
if addr:
print("Assigned " + str(addr) + " to host " + host)
else:
logging.error("Missing host to allocate address to.")
sys.exit(1)