api: servers: delete associated ports upon server removal

wilfred-dev · Nov 19, 2021 · 8d20053 · 8d20053
1 parent 7247705
commit 8d20053
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/wilfred/api/servers.py b/wilfred/api/servers.py
@@ -141,11 +141,16 @@ def remove(self, server: Server):
 
         path = f"{self._configuration['data_path']}/{server.name}_{server.id}"
 
+        # delete all environment variables associated to this server
         for x in (
             session.query(EnvironmentVariable).filter_by(server_id=server.id).all()
         ):
             session.delete(x)
 
+        # delete all additional ports associated to this server
+        for x in session.query(Port).filter_by(server_id=server.id).all():
+            session.delete(x)
+
         session.delete(server)
         session.commit()