Do not attempt to serialize User models

This commit is contained in:
Adarnof 2018-02-22 17:54:35 -05:00
parent 9c74952607
commit 980569de68
3 changed files with 5 additions and 4 deletions

View File

@ -9,5 +9,5 @@ class Command(BaseCommand):
def handle(self, *args, **options):
for u in User.objects.all():
validate_services(u)
validate_services(u.pk)
self.stdout.write(self.style.SUCCESS('Verified all user service accounts.'))

View File

@ -2,7 +2,7 @@ import logging
import redis
from celery import shared_task
from django.contrib.auth.models import User
from .hooks import ServicesHook
REDIS_CLIENT = redis.Redis()
@ -38,7 +38,8 @@ def only_one(function=None, key="", timeout=None):
@shared_task(bind=True)
def validate_services(self, user):
def validate_services(self, pk):
user = User.objects.get(pk=pk)
logger.debug('Ensuring user {} has permissions for active services'.format(user))
# Iterate through services hooks and have them check the validity of the user
for svc in ServicesHook.get_services():

View File

@ -18,7 +18,7 @@ class ServicesTasksTestCase(TestCase):
services_hook.get_services.return_value = [svc]
validate_services.delay(user=self.member)
validate_services.delay(self.member.pk)
self.assertTrue(services_hook.get_services.called)
self.assertTrue(svc.validate_user.called)