External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-01-13 11:00:03 -03:30
Code Issues Packages Projects Releases Wiki Activity

Catch some errors that pop up during ActivityStream testing

Browse Source
This commit is contained in:
Matthew Jones 2013-11-08 10:21:30 -05:00
parent 2ab4cab495
commit b7b22fa7f8
3 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
awx/main/middleware.py
View File

@ -1,10 +1,10 @@
from django.conf import settings
from django.db.models.signals import pre_save
from django.utils.functional import curry
from awx.main.models import ActivityStream
from awx.main.models.base import ActivityStream
class ActvitiyStreamMiddleware(object):
class ActivityStreamMiddleware(object):
def process_request(self, request):
if hasattr(request, 'user') and hasattr(request.user, 'is_authenticated') and request.user.is_authenticated():

2
awx/main/models/__init__.py
View File

@ -44,4 +44,4 @@ activity_stream_registrar.connect(JobTemplate)
activity_stream_registrar.connect(Job)
activity_stream_registrar.connect(JobHostSummary)
activity_stream_registrar.connect(JobEvent)
activity_stream_registrar.connect(Profile)
#activity_stream_registrar.connect(Profile)

13
awx/main/registrar.py
View File

@ -1,6 +1,10 @@
import logging
from django.db.models.signals import pre_save, post_save, post_delete, m2m_changed
from signals import activity_stream_create, activity_stream_update, activity_stream_delete, activity_stream_associate
logger = logging.getLogger('awx.main.registrar')
class ActivityStreamRegistrar(object):
def __init__(self):
@ -15,9 +19,12 @@ class ActivityStreamRegistrar(object):
post_delete.connect(activity_stream_delete, sender=model, dispatch_uid=str(self.__class__) + str(model) + "_delete")
for m2mfield in model._meta.many_to_many:
m2m_attr = getattr(model, m2mfield.name)
m2m_changed.connect(activity_stream_associate, sender=m2m_attr.through,
dispatch_uid=str(self.__class__) + str(m2m_attr.through) + "_associate")
try:
m2m_attr = getattr(model, m2mfield.name)
m2m_changed.connect(activity_stream_associate, sender=m2m_attr.through,
dispatch_uid=str(self.__class__) + str(m2m_attr.through) + "_associate")
except AttributeError:
logger.warning("Failed to attach m2m activity stream tracker on class %s attribute %s" % (model, m2mfield.name))
def disconnect(self, model):
if model in self.models: