From c4d9254824a5bc34dc0500dfdc9e6fbc1838272d Mon Sep 17 00:00:00 2001 From: A B Date: Mon, 21 Oct 2024 20:36:30 +0000 Subject: [PATCH] Update task --- vpn/models.py | 4 ++-- vpn/tasks.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/vpn/models.py b/vpn/models.py index 29509fc..a733170 100644 --- a/vpn/models.py +++ b/vpn/models.py @@ -48,8 +48,8 @@ class ACL(models.Model): @receiver(post_save, sender=ACL) def acl_created_or_updated(sender, instance, created, **kwargs): - sync_user.delay_on_commit(instance.user.id, instance.server.id) + sync_user.delay_on_commit(instance.user, instance.server.id) @receiver(pre_delete, sender=ACL) def acl_deleted(sender, instance, **kwargs): - sync_user.delay_on_commit(instance.user.id, instance.server.id) \ No newline at end of file + sync_user.delay_on_commit(instance.user, instance.server.id) \ No newline at end of file diff --git a/vpn/tasks.py b/vpn/tasks.py index efb37f4..537235e 100644 --- a/vpn/tasks.py +++ b/vpn/tasks.py @@ -47,13 +47,12 @@ def sync_server(id): return {"status": Server.objects.get(id=id).sync()} @shared_task(name="sync_user_on_server") -def sync_user(user_id, server_id): +def sync_user(user, server_id): from .models import User, ACL from vpn.server_plugins import Server errors = {} result = {} - user = User.objects.get(id=user_id) acls = ACL.objects.filter(user=user) server = Server.objects.get(id=server_id)