1 files changed, 57 insertions, 0 deletions
diff --git a/app/discover/events/event_subnet_delete.py b/app/discover/events/event_subnet_delete.py
new file mode 100644
index 0000000..900e701
--- /dev/null
+++ b/app/discover/events/event_subnet_delete.py
@@ -0,0 +1,57 @@
+###############################################################################
+# Copyright (c) 2017 Koren Lev (Cisco Systems), Yaron Yogev (Cisco Systems)   #
+# and others                                                                  #
+#                                                                             #
+# All rights reserved. This program and the accompanying materials            #
+# are made available under the terms of the Apache License, Version 2.0       #
+# which accompanies this distribution, and is available at                    #
+# http://www.apache.org/licenses/LICENSE-2.0                                  #
+###############################################################################
+from discover.events.event_base import EventResult
+from discover.events.event_delete_base import EventDeleteBase
+
+
+class EventSubnetDelete(EventDeleteBase):
+
+    def delete_children_documents(self, env, vservice_id):
+        vnic_parent_id = vservice_id + '-vnics'
+        vnic = self.inv.get_by_field(env, 'vnic', 'parent_id', vnic_parent_id, get_single=True)
+        if not vnic:
+            self.log.info("Vnic document not found.")
+            return EventResult(result=False, retry=False)
+
+        # delete port and vnic together by mac address.
+        self.inv.delete('inventory', {"mac_address": vnic.get("mac_address")})
+        return self.delete_handler(env, vservice_id, 'vservice')
+
+    def handle(self, env, notification):
+        subnet_id = notification['payload']['subnet_id']
+        network_document = self.inv.get_by_field(env, "network", "subnet_ids", subnet_id, get_single=True)
+        if not network_document:
+            self.log.info("network document not found, aborting subnet deleting")
+            return EventResult(result=False, retry=False)
+
+        # remove subnet_id from subnet_ids array
+        network_document["subnet_ids"].remove(subnet_id)
+
+        # find the subnet in network_document by subnet_id
+        subnet = next(
+            filter(lambda s: s['id'] == subnet_id,
+                   network_document['subnets'].values()),
+            None)
+
+        # remove cidr from cidrs and delete subnet document.
+        if subnet:
+            network_document['cidrs'].remove(subnet['cidr'])
+            del network_document['subnets'][subnet['name']]
+
+        self.inv.set(network_document)
+
+        # when network does not have any subnet, delete vservice DHCP, port and vnic documents.
+        if not network_document["subnet_ids"]:
+            vservice_dhcp_id = 'qdhcp-{}'.format(network_document['id'])
+            self.delete_children_documents(env, vservice_dhcp_id)
+
+        return EventResult(result=True,
+                           related_object=subnet['id'],
+                           display_context=network_document.get('id'))