根据 django 中的条件查找公共元素

Question

我通过 m2m 关系连接了三个模型。 Stop 代表一个公共汽车站，Route 代表一个公共汽车路线，RouteStation 是一种将 Stop 和 Route 相互连接起来的中间模型，其中包含有关该站在路线上的位置的信息 (RouteStation.order)。

class Stop(models.Model):
    lat = models.FloatField()
    lon = models.FloatField()
    name = models.CharField(max_length=250, blank=True, default="None")
    stop_id = models.IntegerField(unique=True)


class Route(models.Model):
    route_id = models.CharField(max_length=20)
    journey_pattern = models.CharField(max_length=20)
    stops = models.ManyToManyField(Stop, through="RouteStation")


class RouteStation(models.Model):
    stop = models.ForeignKey(Stop, on_delete=models.CASCADE)
    route = models.ForeignKey(Route, on_delete=models.CASCADE)
    order = models.IntegerField()

给定两个站点，我需要返回一个查询集，其中包含两个站点共有的所有路线，其中第二个站点的 RouteStation.order 值高于第一个站点。此条件可确保可通过该路线上的第一站到达第二站。

我可以使用以下方法找到常见的停靠点：

stop1 = Stop.objects.get(stop_id=origin)
stop2 = Stop.objects.get(stop_id=destination)

routes1 = stop1.route_set.all()
routes2 = stop2.route_set.all()
common = routes1 & routes2

但是，我无法根据 RouteStation.order 值过滤它。有谁知道最好的解决方法？

Answer 1

在这种情况下，我建议performing raw SQL queries，因为使用 Django orm 比必要的复杂。

sql = '''
SELECT * FROM yourapp_routestation r1, yourapp_routestation r2
WHERE r1.stop_id = {s1} AND 
  r2.stop_id = {s2} AND
  r1.route_id = r2.route_id AND
  r1.order < r2.order
'''

rs_query = RouteStation.objects.raw(sql.format(s1=stop1.stop_id, s2=stop2.stop_id))
for rs in rs_query:
    print(rs.route_id)

注意：

将yourapp_routestation 更改为实际的表名。

rs_query 是 RouteStation 对象，因此您将获得带有 rs.route 的 Route 对象。