如何在Django-admin命令中按用户存储时区发送消息？

核心思路 #

不要依赖服务器时区直接判断，而是基于用户数据库中存储的时区偏好，切换线程时区后验证本地时间是否符合发送条件，同时按时区分组批量处理，减少时区切换的性能开销。

步骤1：确保用户模型存储时区字段 #

先在自定义用户模型中添加时区选择字段，用pytz提供的标准时区列表：

# models.py
from django.contrib.auth.models import AbstractUser
from django.db import models
from pytz import all_timezones

class CustomUser(AbstractUser):
    timezone = models.CharField(
        max_length=100,
        choices=[(tz, tz) for tz in all_timezones],
        default='UTC',
        verbose_name='时区'
    )

步骤2：编写Admin命令的handle方法 #

在handle函数中，按时区分组用户，切换时区后验证本地时间是否到达目标发送时间（周一10点），符合条件则批量发送：

# management/commands/send_scheduled_messages.py
from django.core.management.base import BaseCommand
from django.utils.timezone import activate, get_current_timezone, now
from myapp.models import CustomUser
from datetime import time

class Command(BaseCommand):
    help = '给不同时区的用户发送定时消息（本地时间每周一10点）'

    def handle(self, *args, **options):
        # 目标发送条件：本地时间周一10点（允许±1分钟误差）
        target_local_time = time(hour=10)
        target_weekday = 0  # datetime.weekday() 0=周一，6=周日
        time_tolerance = 1  # 允许的分钟误差

        # 按时区分组用户，减少重复切换时区的开销
        users_grouped_by_tz = {}
        for user in CustomUser.objects.all():
            users_grouped_by_tz.setdefault(user.timezone, []).append(user)

        for tz_str, user_batch in users_grouped_by_tz.items():
            # 激活当前线程的时区
            activate(tz_str)
            current_tz = get_current_timezone()

            # 将UTC时间转换为当前时区的本地时间
            local_now = now().astimezone(current_tz)

            # 验证是否符合发送条件
            if (local_now.weekday() == target_weekday and
                local_now.hour == target_local_time.hour and
                abs(local_now.minute - target_local_time.minute) <= time_tolerance):
                # 执行消息发送逻辑
                self._send_messages_to_batch(user_batch, tz_str)

    def _send_messages_to_batch(self, user_batch, tz_str):
        """批量给一组用户发送消息"""
        for user in user_batch:
            # 替换为你的实际消息发送逻辑（邮件/推送等）
            self.stdout.write(f"已给用户 {user.username}（时区：{tz_str}）发送消息")

关键说明 #

• django.utils.timezone.activate(tz_str)：设置当前线程的时区，后续get_current_timezone()会返回这个时区，自动处理夏令时等时区规则变化。
• 按时区分组：避免给每个用户单独切换时区，大幅提升命令执行效率。
• 时间误差容忍：脚本不可能精准卡在整10点执行，设置±1分钟的误差范围，确保不会错过发送时机。
• 触发方式：用cron或类似定时工具每分钟执行一次这个命令，就能在每个时区的10点左右自动触发发送。

内容的提问来源于stack exchange，提问作者Naftali Shtern