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)