# Copyright (c) 2014 AnsibleWorks, Inc.
# All Rights Reserved.

# Django
from django.conf import settings

# AWX
from awx.main.models.base import *
from awx.main.models.unified_jobs import *
from awx.main.models.organization import *
from awx.main.models.projects import *
from awx.main.models.inventory import *
from awx.main.models.jobs import *
from awx.main.models.schedules import *
from awx.main.models.activity_stream import *

if getattr(settings, 'UNIFIED_JOBS_STEP') in (0, 2):
    # Monkeypatch Django serializer to ignore django-taggit fields (which break
    # the dumpdata command; see https://github.com/alex/django-taggit/issues/155).
    from django.core.serializers.python import Serializer as _PythonSerializer
    _original_handle_m2m_field = _PythonSerializer.handle_m2m_field
    def _new_handle_m2m_field(self, obj, field):
        try:
            field.rel.through._meta
        except AttributeError:
            return
        return _original_handle_m2m_field(self, obj, field)
    _PythonSerializer.handle_m2m_field = _new_handle_m2m_field

    # Add custom methods to User model for permissions checks.
    from django.contrib.auth.models import User
    from awx.main.access import *
    User.add_to_class('get_queryset', get_user_queryset)
    User.add_to_class('can_access', check_user_access)

    # Import signal handlers only after models have been defined.
    import awx.main.signals

    from awx.main.registrar import activity_stream_registrar
    activity_stream_registrar.connect(Organization)
    activity_stream_registrar.connect(Inventory)
    activity_stream_registrar.connect(Host)
    activity_stream_registrar.connect(Group)
    activity_stream_registrar.connect(InventorySource)
    #activity_stream_registrar.connect(InventoryUpdate)
    activity_stream_registrar.connect(Credential)
    activity_stream_registrar.connect(Team)
    activity_stream_registrar.connect(Project)
    #activity_stream_registrar.connect(ProjectUpdate)
    activity_stream_registrar.connect(Permission)
    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)