我正在使用 Django 3.2 和 Django Rest Framework 作為 API。我很難讓它按我期望的方式作業。我有一個position表，其中包含與此類似的示例資料：
[
{ id: 1, position_date: '2022-01-01', symbol: 'AAPL', shares: 100 },
{ id: 2, position_date: '2022-01-01', symbol: 'TSLA', shares: 100 },
{ id: 3, position_date: '2022-01-01', symbol: 'GOOG', shares: 100 },
{ id: 4, position_date: '2022-01-02', symbol: 'AAPL', shares: 200 },
{ id: 5, position_date: '2022-01-02', symbol: 'TSLA', shares: 200 },
{ id: 6, position_date: '2022-01-02', symbol: 'GOOG', shares: 200 },
{ id: 7, position_date: '2022-01-05', symbol: 'AAPL', shares: 300 },
{ id: 8, position_date: '2022-01-05', symbol: 'TSLA', shares: 300 },
{ id: 9, position_date: '2022-01-05', symbol: 'GOOG', shares: 300 },
]
此資料包含特定日期的股票頭寸。此每日快照頭寸資訊僅在市場開市的作業日/日創建，因此周末/市場假期的日期存在差距。
我想按 position_date 查詢職位/api/v1/positions?position_date=<YYYY-MM-DD>，但需要注意的是，當您傳入與職位不直接關聯的日期時，您將獲得小于或等于傳入日期的最大日期的職位。例如，如果我傳遞了一個星期天的日期，該日期沒有與之關聯的職位，我希望從星期五收到職位。在上面的示例資料中，如果我傳入的 position_date 為“2022-01-04”，我希望收到三個 position_date 為“2022-01-02”的記錄。
下面是我的視圖集。目前，如果我沒有傳入 position_date，它會按預期作業并回傳最新的職位。但是，當我傳入 position_date 時，它??只回傳該日期的職位。當日期傳入時， find max_date 的東西似乎在某處/以某種方式被覆寫。關于如何解決這個問題的任何想法？
class PositionViewSet(viewsets.ModelViewSet):
"""
API endpoint that allows positions to be viewed
"""
def get_queryset(self):
# only include params from filterset_fields list
params = dict(self.request.GET)
filtered_params = dict((k, params[k][0]) for k in self.filterset_fields if k in params and params[k][0])
# set default value to latest positions
if 'position_date' not in filtered_params:
filtered_params['position_date'] = date.today().isoformat()
# return all positions
if 'position_date' in filtered_params and filtered_params['position_date'] == 'all':
del filtered_params['position_date']
# find the greatest date that is lte to the passed in date
if 'position_date' in filtered_params:
max_date = Position.objects.filter(position_date__lte=filtered_params['position_date']).aggregate(position_date=Max('position_date'))['position_date'].isoformat()
# use the max date in place of original date
filtered_params['position_date'] = max_date
# fix some janky nested route issues between django and drf
filtered_kwargs = {k.replace('_pk', '_id'): v for k, v in self.kwargs.items()}
# merge kwargs and params
qs_filter = {**filtered_params, **filtered_kwargs}
qs = Position.objects.filter(**qs_filter)
return qs
serializer_class = PositionSerializer
filter_backends = [SearchFilter, OrderingFilter, DjangoFilterBackend]
filterset_fields = [
'symbol',
'shares',
'position_date',
]
search_fields = [
'^symbol',
]
ordering_fields = [
'symbol',
'shares',
'position_date',
]
ordering = ['position_date', 'symbol']
uj5u.com熱心網友回復：
您應該嘗試從中洗掉position_date，filterset_fields因為我認為DjangoFilterBackend會覆寫您的查詢集。
還有一條建議：您必須對自定義過濾器position_date中的欄位進行所有過濾，而不是在函式中。get_queryset
轉載請註明出處，本文鏈接：https://www.uj5u.com/gongcheng/419409.html
標籤：