Django: consulta de filtro basada en la función personalizada

Question

Tengo una función incorporada en mi clase de modelo Django y quiero usar esa función para filtrar los resultados de mi consulta.

class service: 
     ...... 
     def is_active(self): 
      if datetime.now() > self.end_time: 
        return False 
      return True 

Ahora quiero utilizar esta función en mi filtro de consulta, algo así como

nserv = service.objects.filter(is_active=True) 

que sé, en este caso sencillo 'is_active', puedo directamente hacer esta comparación en la consulta de filtro, pero para situaciones más complejas, eso puede no ser posible. ¿Cómo debería hacer una consulta, basada en funciones personalizadas?

Answer 1

Yo te sugeriría utilizar un gestor personalizado para su clase, como esto puede usar:

nserv = service.objects.are_active()

Esto se lograría con algo como:

class ServiceManager(models.Manager): 
    def are_active(self): 
     # use your method to filter results 
     return you_custom_queryset 

Ver custom managers

Answer 2

Es posible que no pueda, en su lugar puede post-procesar el conjunto de preguntas con un list comprehension o generator expression.

Por ejemplo:

[x for x in Q if x.somecond()] 

Answer 3

Acabo de tener un problema similar. El problema fue que tuve que devolver una instancia de QuerySet. Una solución rápida para mí era hacer algo como:

active_serv_ids = [service.id for service in Service.objects.all() if service.is_active()] 
nserv = Service.objects.filter(id__in=active_serv_ids) 

bastante seguro de que esto no es la forma más bonita y performante para hacer esto, pero funciona para mí.

una manera más prolija de hacer esto sería:

active_serv_ids = [] 

for service in Service.objects.all(): 
if service.is_active(): 
    active_serv_ids.append(service.id) 

nserv = Service.objects.filter(id__in=active_serv_ids) 

Answer 4

La respuesta de Ignacio es interesante, pero no devuelve un conjunto de consultas. Esto se hace:

def users_by_role(role): 
    users = User.objects.all() 
    ids = [user.id for user in users if user.role == role] 
    return users.filter(id__in=ids)