awx/awx/main/middleware.py

from django.conf import settings
from django.contrib.auth.models import User, AnonymousUser
from django.db.models.signals import pre_save, post_save
from django.db import IntegrityError
from django.utils.functional import curry
from awx.main.models import ActivityStream, AuthToken
import json
import uuid
import urllib2

import logging
logger = logging.getLogger('awx.main.middleware')

class ActivityStreamMiddleware(object):

    def __init__(self):
        self.disp_uid = None
        self.instances = []

    def process_request(self, request):
        if hasattr(request, 'user') and hasattr(request.user, 'is_authenticated') and request.user.is_authenticated():
            user = request.user
        else:
            user = None

        set_actor = curry(self.set_actor, user)
        self.disp_uid = str(uuid.uuid1())
        post_save.connect(set_actor, sender=ActivityStream, dispatch_uid=self.disp_uid, weak=False)

    def process_response(self, request, response):
        drf_request = getattr(request, 'drf_request', None)
        drf_user = getattr(drf_request, 'user', None)
        if self.disp_uid is not None:
            post_save.disconnect(dispatch_uid=self.disp_uid)
        for instance_id in self.instances:
            instance = ActivityStream.objects.filter(id=instance_id)
            if instance.exists():
                instance = instance[0]
            else:
                logger.debug("Failed to look up Activity Stream instance for id : " + str(instance_id))
                continue

            if drf_user is not None and drf_user.__class__ != AnonymousUser:
                instance.user = drf_user
                try:
                    instance.save()
                except IntegrityError, e:
                    logger.debug("Integrity Error saving Activity Stream instance for id : " + str(instance_id))
            else:
                obj1_type_actual = instance.object1_type.split(".")[-1]
                if obj1_type_actual in ("InventoryUpdate", "ProjectUpdate", "Job") and instance.id is not None:
                    instance.delete()
        return response

    def set_actor(self, user, sender, instance, **kwargs):
        if sender == ActivityStream:
            if isinstance(user, User) and instance.user is None:
                instance.user = user
                instance.save()
            else:
                if instance.id not in self.instances:
                    self.instances.append(instance.id)