from .es_query import HQESQuery
from . import filters


class DomainES(HQESQuery):
    index = 'domains'

    @property
    def builtin_filters(self):
        return [
            real_domains,
            commconnect_domains,
            created,
        ] + super(DomainES, self).builtin_filters


def real_domains():
    return filters.term("is_test", False)


def commconnect_domains():
    return filters.term("commconnect_enabled", True)


def created(gt=None, gte=None, lt=None, lte=None):
    return filters.date_range('date_created', gt, gte, lt, lte)

from .es_query import HQESQuery
from . import filters


class DomainES(HQESQuery):
    index = 'domains'

    @property
    def builtin_filters(self):
        return [
            real_domains,
            commcare_domains,
            commconnect_domains,
            commtrack_domains,
            created,
        ] + super(DomainES, self).builtin_filters


def real_domains():
    return filters.term("is_test", False)


def commcare_domains():
    return filters.AND(filters.term("commconnect_enabled", False),
                       filters.term("commtrack_enabled", False))


def commconnect_domains():
    return filters.term("commconnect_enabled", True)


def commtrack_domains():
    return filters.term("commtrack_enabled", True)


def created(gt=None, gte=None, lt=None, lte=None):
    return filters.date_range('date_created', gt, gte, lt, lte)

Add CommCare, CommTrack filters for DomainES
Add CommCare, CommTrack filters for DomainES

  from .es_query import HQESQuery
  from . import filters
  
  
  class DomainES(HQESQuery):
      index = 'domains'
  
      @property
      def builtin_filters(self):
          return [
              real_domains,
+             commcare_domains,
              commconnect_domains,
+             commtrack_domains,
              created,
          ] + super(DomainES, self).builtin_filters
  
  
  def real_domains():
      return filters.term("is_test", False)
  
  
+ def commcare_domains():
+     return filters.AND(filters.term("commconnect_enabled", False),
+                        filters.term("commtrack_enabled", False))
+ 
+ 
  def commconnect_domains():
      return filters.term("commconnect_enabled", True)
+ 
+ 
+ def commtrack_domains():
+     return filters.term("commtrack_enabled", True)
  
  
  def created(gt=None, gte=None, lt=None, lte=None):
      return filters.date_range('date_created', gt, gte, lt, lte)
  
Add CommCare, CommTrack filters for DomainES
## Code Before:
from .es_query import HQESQuery
from . import filters


class DomainES(HQESQuery):
    index = 'domains'

    @property
    def builtin_filters(self):
        return [
            real_domains,
            commconnect_domains,
            created,
        ] + super(DomainES, self).builtin_filters


def real_domains():
    return filters.term("is_test", False)


def commconnect_domains():
    return filters.term("commconnect_enabled", True)


def created(gt=None, gte=None, lt=None, lte=None):
    return filters.date_range('date_created', gt, gte, lt, lte)

## Instruction:
Add CommCare, CommTrack filters for DomainES
## Code After:
from .es_query import HQESQuery
from . import filters


class DomainES(HQESQuery):
    index = 'domains'

    @property
    def builtin_filters(self):
        return [
            real_domains,
            commcare_domains,
            commconnect_domains,
            commtrack_domains,
            created,
        ] + super(DomainES, self).builtin_filters


def real_domains():
    return filters.term("is_test", False)


def commcare_domains():
    return filters.AND(filters.term("commconnect_enabled", False),
                       filters.term("commtrack_enabled", False))


def commconnect_domains():
    return filters.term("commconnect_enabled", True)


def commtrack_domains():
    return filters.term("commtrack_enabled", True)


def created(gt=None, gte=None, lt=None, lte=None):
    return filters.date_range('date_created', gt, gte, lt, lte)
