mirror of
https://github.com/django-q2/django-q2.git
synced 2026-09-15 21:47:53 +08:00
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
from datetime import datetime
|
|
from django import db
|
|
import calendar
|
|
from datetime import date
|
|
|
|
import django
|
|
from django.utils import timezone
|
|
from django.conf import settings
|
|
|
|
from django_q.conf import Conf, logger
|
|
|
|
if django.VERSION < (4, 0):
|
|
# pytz is the default in django 3.2. Remove when no support for 3.2
|
|
from pytz import timezone as ZoneInfo
|
|
else:
|
|
try:
|
|
from zoneinfo import ZoneInfo
|
|
except ImportError:
|
|
from backports.zoneinfo import ZoneInfo
|
|
|
|
|
|
# credits: https://stackoverflow.com/a/4131114
|
|
# Made them aware of timezone
|
|
def add_months(d, months):
|
|
month = d.month - 1 + months
|
|
year = d.year + month // 12
|
|
month = month % 12 + 1
|
|
day = min(d.day, calendar.monthrange(year, month)[1])
|
|
return d.replace(year=year, month=month, day=day)
|
|
|
|
|
|
# credits: https://stackoverflow.com/a/15743908
|
|
# Changed the last line to make it a little easier to read and changed it to move
|
|
# February 29 to 28 next year.
|
|
def add_years(d, years):
|
|
"""Return a date that's `years` years after the date (or datetime)
|
|
object `d`. Return the same calendar date (month and day) in the
|
|
destination year, if it exists, otherwise use the previous day
|
|
(thus changing February 29 to February 28).
|
|
"""
|
|
try:
|
|
return d.replace(year=d.year + years)
|
|
except ValueError:
|
|
new_date = d + (date(d.year + years, 3, 1) - date(d.year, 3, 1))
|
|
return d.replace(year=new_date.year, month=new_date.month, day=new_date.day)
|
|
|
|
|
|
def localtime(value=None) -> datetime:
|
|
"""Override for timezone.localtime to deal with naive times and local times"""
|
|
if settings.USE_TZ:
|
|
if django.VERSION >= (4, 0) and settings.USE_DEPRECATED_PYTZ:
|
|
import pytz
|
|
|
|
convert_to_tz = pytz.timezone(Conf.TIME_ZONE)
|
|
else:
|
|
convert_to_tz = ZoneInfo(Conf.TIME_ZONE)
|
|
|
|
return timezone.localtime(value=value, timezone=convert_to_tz)
|
|
if value is None:
|
|
return datetime.now()
|
|
else:
|
|
return value
|
|
|
|
|
|
def close_old_django_connections():
|
|
"""
|
|
Close django connections unless running with sync=True.
|
|
"""
|
|
if Conf.SYNC:
|
|
logger.warning(
|
|
"Preserving django database connections because sync=True. Beware "
|
|
"that tasks are now injected in the calling context/transactions "
|
|
"which may result in unexpected behaviour."
|
|
)
|
|
else:
|
|
db.close_old_connections()
|
|
|