1
0
Fork 0
mirror of https://github.com/fastogt/fastocloud_admin.git synced 2025-03-09 23:38:52 +00:00

Remove user fix

This commit is contained in:
topilski 2019-06-01 11:39:08 -04:00
parent e3bd69d0de
commit ec74f8a02a
2 changed files with 11 additions and 0 deletions

View file

@ -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()

View file

@ -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'))