18d5abb2f6
First PR to replace pytz with zoneinfo for UTC handling across codebase. This PR migrates all UTC timezone handling from pytz to Python’s standard library zoneinfo. The pytz library is now deprecated, and its documentation recommends using zoneinfo for all new code. This update modernizes our codebase, removes legacy pytz usage, and ensures compatibility with current best practices for timezone management in Python 3.9+. No functional changes to timezone logic - just a direct replacement for UTC handling. https://github.com/openedx/edx-platform/issues/33980
69 lines
2.3 KiB
Python
69 lines
2.3 KiB
Python
# lint-amnesty, pylint: disable=missing-module-docstring
|
|
|
|
import datetime
|
|
import logging
|
|
|
|
from zoneinfo import ZoneInfo
|
|
from django.db import transaction
|
|
|
|
from openedx.core.djangoapps.schedules.models import Schedule
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
# TODO: consider using a LoggerAdapter instead of this mixin:
|
|
# https://docs.python.org/2/library/logging.html#logging.LoggerAdapter
|
|
class PrefixedDebugLoggerMixin: # lint-amnesty, pylint: disable=missing-class-docstring
|
|
log_prefix = None
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
if self.log_prefix is None:
|
|
self.log_prefix = self.__class__.__name__
|
|
|
|
def log_debug(self, message, *args, **kwargs):
|
|
"""
|
|
Wrapper around LOG.debug that prefixes the message.
|
|
"""
|
|
LOG.debug(self.log_prefix + ': ' + message, *args, **kwargs)
|
|
|
|
def log_info(self, message, *args, **kwargs):
|
|
"""
|
|
Wrapper around LOG.info that prefixes the message.
|
|
"""
|
|
LOG.info(self.log_prefix + ': ' + message, *args, **kwargs)
|
|
|
|
|
|
def reset_self_paced_schedule(user, course_key, use_enrollment_date=False):
|
|
"""
|
|
Reset the user's schedule if self-paced.
|
|
|
|
It does not create a new schedule, just resets an existing one.
|
|
This is used, for example, when a user requests it or when an enrollment mode changes.
|
|
|
|
Arguments:
|
|
user (User)
|
|
course_key (CourseKey or str)
|
|
use_enrollment_date (bool): if False, reset to now, else reset to original enrollment creation date
|
|
"""
|
|
with transaction.atomic(savepoint=False):
|
|
try:
|
|
schedule = Schedule.objects.select_related('enrollment', 'enrollment__course').get(
|
|
enrollment__user=user,
|
|
enrollment__course__id=course_key,
|
|
enrollment__course__self_paced=True,
|
|
)
|
|
except Schedule.DoesNotExist:
|
|
return
|
|
|
|
if use_enrollment_date:
|
|
new_start_date = schedule.enrollment.created
|
|
else:
|
|
new_start_date = datetime.datetime.now(ZoneInfo("UTC"))
|
|
|
|
# Make sure we don't start the clock on the learner's schedule before the course even starts
|
|
new_start_date = max(new_start_date, schedule.enrollment.course.start)
|
|
|
|
schedule.start_date = new_start_date
|
|
schedule.save()
|