def save(self, node): """ Completes subnet allocation. """ network = self.cleaned_data.get('network') cidr = self.cleaned_data.get('cidr') pool = self.cleaned_data.get('pool') if network: if self.cleaned_data['reserve']: # We should reserve this manually allocated subnet in the project pool if not pool.reserve_subnet(network, cidr): return subnet = Subnet(node = node) subnet.subnet = network subnet.cidr = cidr subnet.allocated = True subnet.allocated_at = datetime.now() subnet.status = SubnetStatus.NotAnnounced subnet.description = self.cleaned_data.get('description') subnet.gen_iface_type = IfaceType.LAN subnet.gen_dhcp = self.cleaned_data.get('dhcp') subnet.save() else: allocation = pool.allocate_subnet(int(self.cleaned_data.get('prefix_len')) or 27) subnet = Subnet(node = node, subnet = allocation.network, cidr = allocation.cidr) subnet.allocated = True subnet.allocated_at = datetime.now() subnet.status = SubnetStatus.NotAnnounced subnet.description = self.cleaned_data.get('description') subnet.gen_iface_type = IfaceType.LAN subnet.gen_dhcp = self.cleaned_data.get('dhcp') subnet.save() # Remove any already announced subnets that are the same subnet Subnet.objects.filter(node = node, subnet = subnet.subnet, cidr = subnet.cidr, allocated = False).delete() return node
def save(self): """ Performs the actual renumbering. """ # We must ensure exclusive access during node updates as otherwise this might happen # in the middle of a monitor update and this would cause unwanted consequences self.__node.ensure_exclusive_access() # Determine what subnet primary IP belonged to primary = self.__node.get_primary_subnet() renumber_primary = False old_router_id = self.__node.ip # Renumber subnets first for subnet in queryset_by_ip(self.__node.subnet_set.filter(allocated = True), 'ip_subnet')[:]: action = int(self.cleaned_data.get('subnet_%s' % subnet.pk)) prefix_len = int(self.cleaned_data.get('prefix_%s' % subnet.pk) or 27) manual_ip = self.cleaned_data.get('manual_%s' % subnet.pk) if action == RenumberAction.Keep: pass elif action == RenumberAction.Remove: subnet.delete() else: # This means we should renumber to some other pool pool = Pool.objects.get(pk = action) if manual_ip: new_subnet = pool.reserve_subnet(manual_ip, prefix_len) else: new_subnet = pool.allocate_subnet(prefix_len) # If the old subnet has been the source of node's primary IP remember that save_primary = (not renumber_primary and primary and primary[0] == subnet) # Remove old subnet and create a new one; it is deleted here so the old allocation # is returned to the pool and all status info is reset subnet.delete() s = Subnet(node = self.__node, subnet = new_subnet.network, cidr = new_subnet.cidr) s.allocated = True s.allocated_at = datetime.now() s.status = SubnetStatus.NotAnnounced s.description = subnet.description s.gen_iface_type = subnet.gen_iface_type s.gen_dhcp = subnet.gen_dhcp s.save() if save_primary: primary = s renumber_primary = True # The subnet have now been renumbered, check if we need to renumber the primary IP router_id_changed = False if renumber_primary: net = ipcalc.Network(primary.subnet, primary.cidr) self.__node.ip = str(net.host_first()) router_id_changed = True # Remove conflicting invalid nodes (another node with the IP we just renumbered to) existing_nodes = Node.objects.filter(ip = self.__node.ip, status = NodeStatus.Invalid) if existing_nodes.count() > 0: self.warning_or_continue(_("There is an existing but unregistered node with the same primary IP address you are currently renumbering to! If you continue with this operation, this invalid node will be replaced.")) existing_nodes.delete() # Node has been renumbered, reset monitoring status as this node is obviously not # visible right after renumbering. if router_id_changed: # Update node's DNS record Record.update_for_node(self.__node) if not self.__node.is_pending(): self.__node.status = NodeStatus.Down self.__node.peers = 0 Link.objects.filter(src = self.__node).delete() Link.objects.filter(dst = self.__node).delete() self.__node.subnet_set.filter(allocated = False).delete() self.__node.subnet_set.all().update(status = SubnetStatus.NotAnnounced) # Setup a node renumbered notice (if one doesn't exist yet) try: notice = RenumberNotice.objects.get(node = self.__node) if notice.original_ip == self.__node.ip: notice.delete() self.__node.awaiting_renumber = False else: self.__node.awaiting_renumber = True except RenumberNotice.DoesNotExist: notice = RenumberNotice(node = self.__node) notice.original_ip = old_router_id notice.renumbered_at = datetime.now() notice.save() self.__node.awaiting_renumber = True self.__node.save() # Generate node renumbered event Event.create_event(self.__node, EventCode.NodeRenumbered, '', EventSource.NodeDatabase, data = 'Old address: %s' % old_router_id)