如何在Django中将年度周四日期集转为表单下拉列表选项

要实现trade_expiry字段的下拉列表显示当前年度所有周四，需要动态生成选项（静态choices只会在服务启动时生成一次，无法自动更新），具体步骤如下：

1. 编写生成年度周四的工具函数 #

这个函数会生成指定年份（默认当前年）所有周四的元组列表，格式符合Django表单choices的要求（(日期对象, 显示文本)）：

from datetime import date, timedelta

def get_year_thursdays(year=None):
    if year is None:
        year = date.today().year
    # 定位当年第一个周四
    first_day = date(year, 1, 1)
    # weekday() 中周一为0，周四为3，计算与第一个周四的天数差
    days_to_thursday = (3 - first_day.weekday()) % 7
    first_thursday = first_day + timedelta(days=days_to_thursday)
    
    thursdays = []
    current_thursday = first_thursday
    # 遍历添加当年所有周四
    while current_thursday.year == year:
        display_text = current_thursday.strftime("%Y-%m-%d (周四)")
        thursdays.append((current_thursday, display_text))
        current_thursday += timedelta(weeks=1)
    return thursdays

2. 调整模型字段 #

移除原字段的静态choices，改用动态生成的默认值：

import uuid
from django.db import models
from .utils import get_year_thursdays  # 假设函数放在utils.py中

class Trade(models.Model):
    user_id = models.CharField(max_length=5, null=True, blank=True)
    nse_index = models.ForeignKey('NseIndex', on_delete=models.CASCADE)
    # 默认值设为当年第一个周四，移除静态choices
    trade_expiry = models.DateField(default=get_year_thursdays()[0][0])
    trade_type = models.CharField(max_length=25, choices=TRADE_TYPE, default="paper")
    id = models.UUIDField(default=uuid.uuid4, unique=True, primary_key=True, editable=False)

3. 自定义表单实现动态下拉选项 #

通过重写表单的__init__方法，动态设置trade_expiry的下拉选项：

from django import forms
from .models import Trade
from .utils import get_year_thursdays

class TradeForm(forms.ModelForm):
    class Meta:
        model = Trade
        fields = '__all__'

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 给trade_expiry字段设置动态生成的周四选项
        self.fields['trade_expiry'].choices = get_year_thursdays()
        # 编辑已有记录时，确保选中当前值
        if self.instance.pk and self.instance.trade_expiry:
            self.fields['trade_expiry'].initial = self.instance.trade_expiry

4. 在视图和Admin中使用自定义表单 #

视图中使用 #

from django.shortcuts import render, redirect
from .forms import TradeForm

def create_trade(request):
    if request.method == 'POST':
        form = TradeForm(request.POST)
        if form.is_valid():
            form.save()
            return redirect('trade_list')
    else:
        form = TradeForm()
    return render(request, 'trade_form.html', {'form': form})

Admin后台中使用 #

from django.contrib import admin
from .models import Trade
from .forms import TradeForm

class TradeAdmin(admin.ModelAdmin):
    form = TradeForm

admin.site.register(Trade, TradeAdmin)

可选扩展：支持选择不同年份 #

如果需要让用户选择其他年份的周四，可以在表单中添加年份选择字段，然后根据用户选择的年份重新生成周四列表：

class TradeForm(forms.ModelForm):
    selected_year = forms.IntegerField(initial=date.today().year)

    class Meta:
        model = Trade
        fields = '__all__'

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        # 根据选中的年份生成周四选项
        selected_year = self.data.get('selected_year') or self.initial.get('selected_year') or date.today().year
        self.fields['trade_expiry'].choices = get_year_thursdays(int(selected_year))

内容的提问来源于stack exchange，提问作者Thomas Amal