Estoy ofreciendo servicios a empresas (muchos a muchos entre Service y Company ). Para estructurar mis servicios, utilizo una estructura similar a un árbol utilizando el campo muchos a muchos de Django. Por lo tanto, un servicio puede contener muchos otros servicios.
La pregunta: ¿cómo puedo recuperar todos los servicios que tiene una empresa? Usando company.services , solo obtengo los servicios directamente relacionados con esa empresa. Necesito los directamente relacionados + servicios_incluidos (de forma recursiva).
class Company(models.Model): services = models.ManyToManyField(Service) class Service(models.Model): name = models.CharField(max_length=255) included_services = models.ManyToManyField("self", blank=True, symmetrical=False)No creo que pueda hacer esto usando el marco ORM de Django o escribiendo una declaración SQL sin procesar. Es posible que desee echar un vistazo a las bibliotecas de terceros como django-mptt que le permite almacenar/recuperar modelos representados en estructuras en forma de árbol de manera más eficiente.
Sin embargo, si desea hacerlo a través de Python, esto es lo que puede hacer:
def get_services(company): services = list(company.services.all()) result = [] while services: service = services.pop(0) result.append(service) services.extend(list(service.included_services.all())) return resultLa idea anterior es muy similar a realizar una búsqueda en anchura en un árbol.
Olvidé mencionar que necesitaba un QuerySet como valor de retorno, así que modifiqué un poco la excelente respuesta de @ozgur. Por si alguien también necesita esto:
def get_services(company): services = list(company.services.all()) pks = [] while services: service = services.pop(0) if service.pk not in pks: pks.append(service.pk) services.extend(list(service.included_services.all())) return Service.objects.filter(pk__in=pks)EDITAR: También agregué una verificación de posibles bucles.