-
Notifications
You must be signed in to change notification settings - Fork 470
fix: handle timezone correctly when DJANGO_CELERY_BEAT_TZ_AWARE=False (Issue #924) #926
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Open
NeverStopDreamingWang
wants to merge
10
commits into
celery:main
Choose a base branch
from
NeverStopDreamingWang:main
base: main
Could not load branches
Branch not found: {{ refName }}
Loading
Could not load tags
Nothing to show
Loading
Are you sure you want to change the base?
Some commits from the old base branch may be removed from the timeline,
and old review comments may become outdated.
+24
−18
Open
Changes from 1 commit
Commits
Show all changes
10 commits
Select commit
Hold shift + click to select a range
e5fe235
fix: handle timezone correctly when DJANGO_CELERY_BEAT_TZ_AWARE=False…
NeverStopDreamingWang 90eff76
fix: handle timezone correctly when DJANGO_CELERY_BEAT_TZ_AWARE=False…
NeverStopDreamingWang 5fbcd04
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] cf3a0e5
fix: The local time obtained from the test sample should be consisten…
NeverStopDreamingWang 027831f
Merge remote-tracking branch 'origin/main'
NeverStopDreamingWang 8b49ecc
fix: The issue where the local time obtained does not match the CELER…
NeverStopDreamingWang f61ce19
Update django_celery_beat/schedulers.py
auvipy 6933370
fix: Line too long (104 > 88). (Issue #924)
NeverStopDreamingWang baa65c9
fix: visually indented line with same indent as next logical line. (I…
NeverStopDreamingWang 343d368
fix: visually indented line with same indent as next logical line. (I…
NeverStopDreamingWang File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change | ||||||||
---|---|---|---|---|---|---|---|---|---|---|
|
@@ -12,13 +12,14 @@ | |||||||||
from celery import current_app, schedules | ||||||||||
from celery.beat import ScheduleEntry, Scheduler | ||||||||||
from celery.utils.log import get_logger | ||||||||||
from celery.utils.time import maybe_make_aware | ||||||||||
from celery.utils.time import make_aware | ||||||||||
from django.conf import settings | ||||||||||
from django.core.exceptions import ObjectDoesNotExist | ||||||||||
from django.db import close_old_connections, transaction | ||||||||||
from django.db.models import Case, F, IntegerField, Q, When | ||||||||||
from django.db.models.functions import Cast | ||||||||||
from django.db.utils import DatabaseError, InterfaceError | ||||||||||
from django.utils import timezone | ||||||||||
from kombu.utils.encoding import safe_repr, safe_str | ||||||||||
from kombu.utils.json import dumps, loads | ||||||||||
|
||||||||||
|
@@ -117,8 +118,6 @@ def is_due(self): | |||||||||
# START DATE: only run after the `start_time`, if one exists. | ||||||||||
if self.model.start_time is not None: | ||||||||||
now = self._default_now() | ||||||||||
if getattr(settings, 'DJANGO_CELERY_BEAT_TZ_AWARE', True): | ||||||||||
now = maybe_make_aware(self._default_now()) | ||||||||||
if now < self.model.start_time: | ||||||||||
# The datetime is before the start date - don't run. | ||||||||||
# send a delay to retry on start_time | ||||||||||
|
@@ -147,19 +146,19 @@ def is_due(self): | |||||||||
# Don't recheck | ||||||||||
return schedules.schedstate(False, NEVER_CHECK_TIMEOUT) | ||||||||||
|
||||||||||
# CAUTION: make_aware assumes settings.TIME_ZONE for naive datetimes, | ||||||||||
# while maybe_make_aware assumes utc for naive datetimes | ||||||||||
tz = self.app.timezone | ||||||||||
last_run_at_in_tz = maybe_make_aware(self.last_run_at).astimezone(tz) | ||||||||||
last_run_at_in_tz = make_aware(self.last_run_at, self.app.timezone) | ||||||||||
return self.schedule.is_due(last_run_at_in_tz) | ||||||||||
|
||||||||||
def _default_now(self): | ||||||||||
if getattr(settings, 'DJANGO_CELERY_BEAT_TZ_AWARE', True): | ||||||||||
now = datetime.datetime.now(self.app.timezone) | ||||||||||
if getattr(settings, 'USE_TZ', True): | ||||||||||
now = datetime.datetime.now(self.app.timezone) | ||||||||||
else: | ||||||||||
# Remove the time zone information (convert to naive datetime) | ||||||||||
now = datetime.datetime.now(self.app.timezone).replace(tzinfo=None) | ||||||||||
auvipy marked this conversation as resolved.
Show resolved
Hide resolved
|
||||||||||
else: | ||||||||||
# this ends up getting passed to maybe_make_aware, which expects | ||||||||||
# all naive datetime objects to be in utc time. | ||||||||||
now = datetime.datetime.utcnow() | ||||||||||
# Use the default time zone time of Django | ||||||||||
now = timezone.now() | ||||||||||
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. When
Suggested change
Copilot uses AI. Check for mistakes. Positive FeedbackNegative Feedback
auvipy marked this conversation as resolved.
Show resolved
Hide resolved
|
||||||||||
return now | ||||||||||
|
||||||||||
def __next__(self): | ||||||||||
|
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
The
make_aware
function expects a naive datetime as its first argument, butself.last_run_at
could already be timezone-aware. This will raise a ValueError ifself.last_run_at
is already aware. Consider usingtimezone.localtime(self.last_run_at, self.app.timezone)
or checking if the datetime is naive before callingmake_aware
.Copilot uses AI. Check for mistakes.