def update_network(request, network_pk):
network = get_object_or_404(Network, pk=network_pk)
attrs = network.networkkeyvalue_set.all()
docs = get_dhcp_docstrings(NetworkKeyValue())
aa = get_dhcp_aa(NetworkKeyValue())
if request.method == 'POST':
form = NetworkForm(request.POST, instance=network)
try:
if not form.is_valid():
return render(
request, 'network/network_edit.html', {
'network': network,
'form': form,
'attrs': attrs,
'docs': docs,
'aa': json.dumps(aa)
})
else:
# Handle key value stuff.
kv = None
kv = get_attrs(request.POST)
update_attrs(kv, attrs, NetworkKeyValue, network, 'network')
network = form.save()
return redirect(network.get_update_url())
except ValidationError, e:
if form._errors is None:
form._errors = ErrorDict()
form._errors['__all__'] = ErrorList(e.messages)
if kv:
attrs = dict_to_kv(kv, NetworkKeyValue)
return render(
request, 'network/network_edit.html', {
'network': network,
'form': form,
'attrs': attrs,
'docs': docs,
'aa': json.dumps(aa)
})
def migrate_networks():
all = truth.models.Truth.objects.all()
print all
scopes = []
for thing in all:
if thing.name.find("-vlan") != -1:
scopes.append(thing)
vlan_number_re = re.compile("(\d+)$")
is_network_start = re.compile("dhcp.scope.start")
is_network_end = re.compile("dhcp.scope.end")
is_netmask = re.compile("dhcp.scope.netmask")
is_pool_start = re.compile("dhcp.pool.start")
is_pool_end = re.compile("dhcp.pool.end")
is_ntp_server = re.compile("dhcp.option.ntp_server.\d")
is_domain_name = re.compile("dhcp.option.domain_name.\d")
is_dns_servers = re.compile("dhcp.option.dns_server.\d")
is_pool_allow_booting = re.compile("dhcp.pool.allow_booting.\d")
is_pool_allow_bootp = re.compile("dhcp.pool.allow_bootp.\d")
for scope in scopes:
if str(scope).endswith("fake"):
print "Skipping " + str(scope)
continue
vlan_number = vlan_number_re.search(str(scope)).groups(1)[0]
network_start = None
network_end = None
netmask = None
pool_start = None
pool_end = None
ntp_servers = []
domain_name = None
dns_servers = []
allow_booting = None
allow_bootp = None
for kv in scope.keyvalue_set.all():
if is_network_start.match(kv.key):
network_start = kv.value
if is_network_end.match(kv.key):
network_end = kv.value
if is_netmask.match(kv.key):
netmask = kv.value
if is_pool_start.match(kv.key):
pool_start = kv.value
if is_pool_end.match(kv.key):
pool_end = kv.value
if is_ntp_server.match(kv.key):
ntp_servers.append(kv.value)
if is_domain_name.match(kv.key):
domain_name = kv.value
if is_dns_servers.match(kv.key):
dns_servers.append(kv.value)
if is_pool_allow_booting.match(kv.key):
allow_booting = bool(kv.value)
if is_pool_allow_bootp.match(kv.key):
allow_bootp = bool(kv.value)
vlan = Vlan.objects.filter(number=vlan_number)
try:
net = ipaddr.IPv4Network(network_start + '/' + netmask)
except ipaddr.AddressValueError, e:
if str(scope) == "phx1-vlan75":
net = ipaddr.IPv4Network("10.8.75.0/24")
else:
pdb.set_trace()
continue
network = Network.objects.filter(ip_lower=int(net.network),
ip_upper=0, prefixlen=net.prefixlen)
print "=" * 20 + " " + str(scope)
print "Expected Vlan: " + str(vlan)
print vlan_number
if not network:
print "Creating new Network: " + str(net)
network, _ = Network.objects.get_or_create(
network_str=str(net), ip_type='4')
parent = calc_parent(network)
if parent:
network.site = parent.site
else:
network = network[0]
print "Existing Network: " + str(network)
if vlan:
network.vlan = vlan[0]
else:
v, _ = Vlan.objects.get_or_create(
name="I need a name.", number=vlan_number)
network.vlan = v
network.save()
print network_start
print network_end
print netmask
print pool_start
print pool_end
print ntp_servers
print domain_name
print dns_servers
print allow_booting
print allow_bootp
if str(scope) == "phx1-vlan75":
continue
if ntp_servers:
real_ntp_servers = []
for server in ntp_servers:
if server == '':
continue
else:
real_ntp_servers.append(server)
if real_ntp_servers:
kv = NetworkKeyValue(
key="ntp-servers", value=", ".join(real_ntp_servers),
network=network)
try:
kv.clean()
kv.save()
except IntegrityError, e:
# Duplicate error
pass
continue
else:
real_ntp_servers.append(server)
if real_ntp_servers:
kv = NetworkKeyValue(
key="ntp-servers", value=", ".join(real_ntp_servers),
network=network)
try:
kv.clean()
kv.save()
except IntegrityError, e:
# Duplicate error
pass
if domain_name:
pdb.set_trace()
kv = NetworkKeyValue(key="domain-name", value=domain_name,
network=network)
try:
kv.clean()
kv.save()
except IntegrityError, e:
# Duplicate error
pass
if dns_servers:
kv = NetworkKeyValue(
key="ntp-servers", value=", ".join(dns_servers),
network=network)
try:
kv.clean()
kv.save()
except IntegrityError, e:
# Duplicate error
