# -*- coding: utf-8 -*-

'''

Created on Feb 22, 2013

@author: ymh

'''

from django.db import models

from haystack import signals

class LdtSignalProcessor(signals.BaseSignalProcessor):

    def __connect_signals(self, klass):

        models.signals.post_save.connect(self.handle_save, sender=klass)

        models.signals.post_delete.connect(self.handle_delete, sender=klass)        

    def __disconnect_signals(self, klass):

        models.signals.post_save.disconnect(self.handle_save, sender=klass)

        models.signals.post_delete.disconnect(self.handle_delete, sender=klass)

    def setup(self):

        #put import here to avoid circular 

        from ldt.text.models import Annotation

        self.__connect_signals(Segment)

        self.__connect_signals(Annotation)

    def teardown(self):

        from ldt.text.models import Annotation

        self.__disconnect_signals(Annotation)

        self.__disconnect_signals(Segment)