From ec74f8a02a2d7353e6577362ca4cd2f9beb9885a Mon Sep 17 00:00:00 2001
From: topilski <atopilski@gmail.com>
Date: Sat, 1 Jun 2019 11:39:08 -0400
Subject: [PATCH] Remove user fix

---
 app/service/service_entry.py | 6 ++++++
 app/user/view.py             | 5 +++++
 2 files changed, 11 insertions(+)

diff --git a/app/service/service_entry.py b/app/service/service_entry.py
index 9c62607..bfceb2a 100644
--- a/app/service/service_entry.py
+++ b/app/service/service_entry.py
@@ -42,3 +42,9 @@ class ServiceSettings(Document, ServerSettings):
     def add_user(self, user: UserPair):
         self.users.append(user)
         self.save()
+
+    def remove_user(self, uid):
+        for user in self.users:
+            if user.id == uid:
+                self.users.remove(user)
+        self.save()
diff --git a/app/user/view.py b/app/user/view.py
index 8ef5cee..8bf31c6 100644
--- a/app/user/view.py
+++ b/app/user/view.py
@@ -3,6 +3,7 @@ from flask import render_template, redirect, url_for, request
 from flask_login import login_required, current_user
 
 from app.home.forms import SettingsForm
+from app.service.service_entry import ServiceSettings
 
 
 # routes
@@ -50,5 +51,9 @@ class UserView(FlaskView):
 
     @login_required
     def remove(self):
+        servers = ServiceSettings.objects()
+        for server in servers:
+            server.remove_user(current_user.id)
+
         current_user.delete()
         return redirect(url_for('HomeView:index'))