[英]How can I convert this SQL statement to Django QuerySet?
请注意:这个问题是我几天前提出的这个问题的后续问题 。 它不是重复的。 我正在尝试在Django中建模的SQL查询和我加载的虚拟数据之间存在轻微但显着的差异。
我正在编写一个Python / Django应用程序来进行一些库存分析。
我有两个非常简单的模型,如下所示:
class Stock(models.Model):
symbol = models.CharField(db_index=True, max_length=5, null=False, editable=False, unique=True)
class StockHistory(models.Model):
stock = models.ForeignKey(Stock, related_name='StockHistory_stock', editable=False)
trading_date = models.DateField(db_index=True, null=False, editable=False)
close = models.DecimalField(max_digits=12, db_index=True, decimal_places=5, null=False, editable=False)
class Meta:
unique_together = ('stock', 'trading_date')
这是我填充的虚拟数据:
import datetime
a = Stock.objects.create(symbol='A')
b = Stock.objects.create(symbol='B')
c = Stock.objects.create(symbol='C')
d = Stock.objects.create(symbol='D')
StockHistory.objects.create(trading_date=datetime.date(2018,1,1), close=200, stock=a)
StockHistory.objects.create(trading_date=datetime.date(2018,1,2), close=150, stock=a)
StockHistory.objects.create(trading_date=datetime.date(2018,1,3), close=120, stock=a)
StockHistory.objects.create(trading_date=datetime.date(2018,4,28), close=105, stock=a)
StockHistory.objects.create(trading_date=datetime.date(2018,5,2), close=105, stock=a)
StockHistory.objects.create(trading_date=datetime.date(2018,5,3), close=105, stock=a)
StockHistory.objects.create(trading_date=datetime.date(2017,5,2), close=400, stock=b)
StockHistory.objects.create(trading_date=datetime.date(2017,11,11), close=200, stock=b)
StockHistory.objects.create(trading_date=datetime.date(2017,11,12), close=300, stock=b)
StockHistory.objects.create(trading_date=datetime.date(2017,11,13), close=400, stock=b)
StockHistory.objects.create(trading_date=datetime.date(2017,11,14), close=500, stock=b)
StockHistory.objects.create(trading_date=datetime.date(2018,4,28), close=105, stock=c)
StockHistory.objects.create(trading_date=datetime.date(2018,4,29), close=106, stock=c)
StockHistory.objects.create(trading_date=datetime.date(2018,4,30), close=107, stock=c)
StockHistory.objects.create(trading_date=datetime.date(2018,5,1), close=108, stock=c)
StockHistory.objects.create(trading_date=datetime.date(2018,5,2), close=109, stock=c)
StockHistory.objects.create(trading_date=datetime.date(2018,5,3), close=110, stock=c)
StockHistory.objects.create(trading_date=datetime.date(2018,5,4), close=90, stock=c)
我想找到过去一周内年度低点的所有股票。
但是为了使这个问题更简单,只要假设我想找到自'2017-05-04'
发生在'2018-04-30'
之后或之后的最低点的所有股票。 下面是我写的SQL来找到它。 有用。
但是我需要帮助找出要写的Django Query以获得与此SQL相同的结果。 我该怎么做? 在我上一个问题的答案中提供的Django产生了3行结果而不是2行。
select
s.symbol,
min(sh.trading_date),
low_table.low
from
(
select
stock_id,
min(close) as low
from
stocks_stockhistory
where
trading_date >= '2017-05-04'
group by
stock_id
) as low_table,
stocks_stockhistory as sh,
stocks_stock as s
where
sh.stock_id = low_table.stock_id
and sh.stock_id = s.id
and sh.close = low_table.low
and sh.trading_date >= '2018-04-30'
group by
s.symbol,
low_table.low
order by
s.symbol asc;
+--------+----------------------+--------------------+
| symbol | min(sh.trading_date) | min(low_table.low) |
+--------+----------------------+--------------------+
| A | 2018-05-02 | 105.00000 |
| C | 2018-05-04 | 90.00000 |
+--------+----------------------+--------------------+
2 rows in set (0.01 sec)
你能试一下吗
from stocks.models import StockHistory, Stock
from django.db.models import OuterRef, Subquery, F, Min
low = StockHistory.objects.filter(
stock=OuterRef('stock'), trading_date__gt='2017-05-04'
).order_by('close')
qs = StockHistory.objects.annotate(
low=Subquery(low.values('close')[:1])
)
qs = qs.filter(low=F('close')).filter(trading_date__gte='2018-04-30')
qs = qs.values('stock__symbol', 'low').order_by('stock__symbol', 'low')
qs = qs.annotate(mtd=Min('trading_date'))
qs = qs.values_list('stock__symbol', 'mtd', 'low')
qs = qs.order_by('stock__symbol', 'low')
结果:
>>> qs
<QuerySet [('A', datetime.date(2018, 5, 2), Decimal('105.00000')), ('C', datetime.date(2018, 5, 4), Decimal('90.00000'))]>
sql代码是
>>> print(qs.query)
SELECT "stocks_stock"."symbol",
(SELECT U0."close"
FROM "stocks_stockhistory" U0
WHERE (U0."stock_id" = ("stocks_stockhistory"."stock_id")
AND U0."trading_date" > 2017-05-04)
ORDER BY U0."close" ASC LIMIT 1) AS "low",
MIN("stocks_stockhistory"."trading_date") AS "mtd"
FROM "stocks_stockhistory"
INNER JOIN "stocks_stock"
ON ("stocks_stockhistory"."stock_id" = "stocks_stock"."id")
WHERE (
(SELECT U0."close"
FROM "stocks_stockhistory" U0
WHERE (U0."stock_id" = ("stocks_stockhistory"."stock_id") AND U0."trading_date" > 2017-05-04)
ORDER BY U0."close" ASC LIMIT 1) = ("stocks_stockhistory"."close")
AND "stocks_stockhistory"."trading_date" >= 2018-04-30)
GROUP BY "stocks_stock"."symbol",
(SELECT U0."close"
FROM "stocks_stockhistory" U0
WHERE (U0."stock_id" = ("stocks_stockhistory"."stock_id") AND U0."trading_date" > 2017-05-04)
ORDER BY U0."close" ASC LIMIT 1)
ORDER BY "stocks_stock"."symbol" ASC, "low" ASC
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.