chore: use zoneinfo instead of pytz

Author: ttak-apphelix

xmodule/assetstore/__init__.py

@@ -7,9 +7,9 @@
 from datetime import datetime
 
 import dateutil.parser
-import pytz
 from lxml import etree
 from opaque_keys.edx.keys import AssetKey, CourseKey
+from zoneinfo import ZoneInfo
 
 
 class AssetMetadata:
@@ -80,7 +80,7 @@ def __init__(self, asset_id,
         self.thumbnail = thumbnail
         self.curr_version = curr_version
         self.prev_version = prev_version
-        now = datetime.now(pytz.utc)
+        now = datetime.now(ZoneInfo("UTC"))
         self.edited_by = edited_by
         self.edited_by_email = edited_by_email
         self.edited_on = edited_on or now

xmodule/capa/capa_problem.py

@@ -26,7 +26,6 @@
 
 from django.conf import settings
 from lxml import etree
-from pytz import UTC
 
 from openedx.core.djangolib.markup import HTML, Text
 from openedx.core.lib.safe_lxml.xmlparser import XML
@@ -35,6 +34,7 @@
 from xmodule.capa.safe_exec import safe_exec
 from xmodule.capa.util import contextualize_text, convert_files_to_filenames, get_course_id_from_capa_block
 from xmodule.stringify import stringify_children
+from zoneinfo import ZoneInfo
 
 # extra things displayed after "show answers" is pressed
 solution_tags = ["solution"]
@@ -437,7 +437,10 @@ def get_recentmost_queuetime(self):
             if self.correct_map.is_queued(answer_id)
         ]
         queuetimes = [
-            datetime.strptime(qt_str, xqueue_interface.DATEFORMAT).replace(tzinfo=UTC) for qt_str in queuetime_strs
+            datetime.strptime(qt_str, xqueue_interface.DATEFORMAT).replace(
+                tzinfo=ZoneInfo("UTC")
+            )
+            for qt_str in queuetime_strs
         ]
 
         return max(queuetimes)

xmodule/capa/responsetypes.py

@@ -33,7 +33,6 @@
 from lxml import etree
 from lxml.html.soupparser import fromstring as fromstring_bs  # uses Beautiful Soup!!! FIXME?
 from pyparsing import ParseException
-from pytz import UTC
 from shapely.geometry import MultiPoint, Point
 from six.moves import map, range, zip
 from symmath import symmath_check
@@ -42,6 +41,7 @@
 from openedx.core.lib.grade_utils import round_away_from_zero
 from xmodule.capa.safe_exec import safe_exec
 from xmodule.capa.xqueue_interface import DATEFORMAT, make_hashkey, make_xheader
+from zoneinfo import ZoneInfo
 
 from . import correctmap
 from .registry import TagRegistry
@@ -2653,7 +2653,7 @@ def get_score(self, student_answers):  # pylint: disable=too-many-locals
         # ------------------------------------------------------------
 
         qinterface = self.capa_system.xqueue.interface
-        qtime = datetime.strftime(datetime.now(UTC), DATEFORMAT)
+        qtime = datetime.strftime(datetime.now(ZoneInfo("UTC")), DATEFORMAT)
 
         anonymous_student_id = self.capa_system.anonymous_student_id
 

xmodule/capa/tests/test_responsetypes.py

@@ -17,7 +17,6 @@
 import pytest
 import random2 as random
 import requests
-from pytz import UTC
 
 from xmodule.capa.correctmap import CorrectMap
 from xmodule.capa.responsetypes import (
@@ -49,6 +48,7 @@
 from xmodule.capa.tests.test_util import UseUnsafeCodejail
 from xmodule.capa.util import convert_files_to_filenames
 from xmodule.capa.xqueue_interface import DATEFORMAT
+from zoneinfo import ZoneInfo
 
 
 class ResponseTest(unittest.TestCase):
@@ -999,7 +999,7 @@ def test_is_queued(self):
         # Now we queue the LCP
         cmap = CorrectMap()
         for i, answer_id in enumerate(answer_ids):
-            queuestate = CodeResponseTest.make_queuestate(i, datetime.now(UTC))
+            queuestate = CodeResponseTest.make_queuestate(i, datetime.now(ZoneInfo("UTC")))
             cmap.update(CorrectMap(answer_id=answer_id, queuestate=queuestate))
         self.problem.correct_map.update(cmap)
 
@@ -1015,7 +1015,7 @@ def test_update_score(self):  # pylint: disable=too-many-locals
         old_cmap = CorrectMap()
         for i, answer_id in enumerate(answer_ids):
             queuekey = 1000 + i
-            queuestate = CodeResponseTest.make_queuestate(queuekey, datetime.now(UTC))
+            queuestate = CodeResponseTest.make_queuestate(queuekey, datetime.now(ZoneInfo("UTC")))
             old_cmap.update(CorrectMap(answer_id=answer_id, queuestate=queuestate))
 
         # Message format common to external graders
@@ -1083,14 +1083,14 @@ def test_recentmost_queuetime(self):
         cmap = CorrectMap()
         for i, answer_id in enumerate(answer_ids):
             queuekey = 1000 + i
-            latest_timestamp = datetime.now(UTC)
+            latest_timestamp = datetime.now(ZoneInfo("UTC"))
             queuestate = CodeResponseTest.make_queuestate(queuekey, latest_timestamp)
             cmap.update(CorrectMap(answer_id=answer_id, queuestate=queuestate))
         self.problem.correct_map.update(cmap)
 
         # Queue state only tracks up to second
         latest_timestamp = datetime.strptime(datetime.strftime(latest_timestamp, DATEFORMAT), DATEFORMAT).replace(
-            tzinfo=UTC
+            tzinfo=ZoneInfo("UTC")
         )
 
         assert self.problem.get_recentmost_queuetime() == latest_timestamp
@@ -1153,7 +1153,7 @@ def test_parse_score_msg_of_responder(self):
         old_cmap = CorrectMap()
         for i, answer_id in enumerate(answer_ids):
             queuekey = 1000 + i
-            queuestate = CodeResponseTest.make_queuestate(queuekey, datetime.now(UTC))
+            queuestate = CodeResponseTest.make_queuestate(queuekey, datetime.now(ZoneInfo("UTC")))
             old_cmap.update(CorrectMap(answer_id=answer_id, queuestate=queuestate))
 
         for grader_msg in valid_grader_msgs:

xmodule/capa_block.py

@@ -23,7 +23,6 @@
 from django.utils.encoding import smart_str
 from django.utils.functional import cached_property
 from lxml import etree
-from pytz import utc
 from web_fragments.fragment import Fragment
 from xblock.core import XBlock
 from xblock.exceptions import NotFoundError, ProcessingError
@@ -62,6 +61,7 @@
 from xmodule.util.sandboxing import SandboxService
 from xmodule.x_module import XModuleMixin, XModuleToXBlockMixin, shim_xmodule_js
 from xmodule.xml_block import XmlMixin
+from zoneinfo import ZoneInfo
 
 from .capa.xqueue_interface import XQueueService
 
@@ -931,7 +931,7 @@ def set_last_submission_time(self):
         """
         Set the module's last submission time (when the problem was submitted)
         """
-        self.last_submission_time = datetime.datetime.now(utc)
+        self.last_submission_time = datetime.datetime.now(ZoneInfo("UTC"))
 
     def get_progress(self):
         """
@@ -1448,7 +1448,7 @@ def is_past_due(self):
         """
         Is it now past this problem's due date, including grace period?
         """
-        return self.close_date is not None and datetime.datetime.now(utc) > self.close_date
+        return self.close_date is not None and datetime.datetime.now(ZoneInfo("UTC")) > self.close_date
 
     def closed(self):
         """
@@ -1774,7 +1774,7 @@ def submit_problem(  # pylint: disable=too-many-statements,too-many-branches,too
         event_info["answers"] = answers_without_files
 
         # Can override current time
-        current_time = datetime.datetime.now(utc)
+        current_time = datetime.datetime.now(ZoneInfo("UTC"))
         if override_time is not False:
             current_time = override_time
 

xmodule/course_block.py

@@ -15,7 +15,6 @@
 from lazy import lazy
 from lxml import etree
 from path import Path as path
-from pytz import utc
 from xblock.fields import Boolean, Date, Dict, Float, Integer, List, Scope, String
 from openedx.core.djangoapps.video_pipeline.models import VideoUploadsEnabledByDefault
 from openedx.core.djangoapps.video_config.sharing import (
@@ -31,6 +30,7 @@
 from xmodule.graders import grader_from_conf
 from xmodule.seq_block import SequenceBlock
 from xmodule.tabs import CourseTabList, InvalidTabsException
+from zoneinfo import ZoneInfo
 
 from .modulestore.exceptions import InvalidProctoringProvider
 
@@ -163,7 +163,7 @@ def table_of_contents(self):
             # see if we already fetched this
             if toc_url in _cached_toc:
                 (table_of_contents, timestamp) = _cached_toc[toc_url]
-                age = datetime.now(utc) - timestamp
+                age = datetime.now(ZoneInfo("UTC")) - timestamp
                 # expire every 10 minutes
                 if age.seconds < 600:
                     return table_of_contents
@@ -1485,7 +1485,7 @@ def forum_posts_allowed(self):
 
         blackouts = self.get_discussion_blackout_datetimes()
         posting_restrictions = self.discussions_settings.get('posting_restrictions', 'disabled')
-        now = datetime.now(utc)
+        now = datetime.now(ZoneInfo("UTC"))
 
         if posting_restrictions == 'enabled':
             return False
@@ -1601,7 +1601,7 @@ def can_toggle_course_pacing(self):
         """
         if not self.start:
             return False
-        return datetime.now(utc) <= self.start
+        return datetime.now(ZoneInfo("UTC")) <= self.start
 
 
 class CourseSummary:
@@ -1674,5 +1674,5 @@ def has_ended(self):
                     course_id=str(self.id), end_date=self.end, err=e
                 )
             )
-            modified_end = self.end.replace(tzinfo=utc)
+            modified_end = self.end.replace(tzinfo=ZoneInfo("UTC"))
             return course_metadata_utils.has_course_ended(modified_end)

xmodule/course_metadata_utils.py

@@ -12,9 +12,9 @@
 from math import exp
 
 import dateutil.parser
-from pytz import utc
+from zoneinfo import ZoneInfo
 
-DEFAULT_START_DATE = datetime(2030, 1, 1, tzinfo=utc)
+DEFAULT_START_DATE = datetime(2030, 1, 1, tzinfo=ZoneInfo("UTC"))
 
 """
 Default grading policy for a course run.
@@ -95,7 +95,7 @@ def has_course_started(start_date):
         start_date (datetime): The start datetime of the course in question.
     """
     # TODO: This will throw if start_date is None... consider changing this behavior?
-    return datetime.now(utc) > start_date
+    return datetime.now(ZoneInfo("UTC")) > start_date
 
 
 def has_course_ended(end_date):
@@ -107,7 +107,7 @@ def has_course_ended(end_date):
     Arguments:
         end_date (datetime): The end datetime of the course in question.
     """
-    return datetime.now(utc) > end_date if end_date is not None else False
+    return datetime.now(ZoneInfo("UTC")) > end_date if end_date is not None else False
 
 
 def is_enrollment_open(enrollment_start_date, enrollment_end_date):
@@ -118,9 +118,9 @@ def is_enrollment_open(enrollment_start_date, enrollment_end_date):
         enrollment_start_date (datetime): The enrollment start datetime of the course.
         enrollment_end_date (datetime): The enrollment end datetime of the course.
     """
-    now = datetime.now(utc)
-    enrollment_start_date = enrollment_start_date or datetime.min.replace(tzinfo=utc)
-    enrollment_end_date = enrollment_end_date or datetime.max.replace(tzinfo=utc)
+    now = datetime.now(ZoneInfo("UTC"))
+    enrollment_start_date = enrollment_start_date or datetime.min.replace(tzinfo=ZoneInfo("UTC"))
+    enrollment_end_date = enrollment_end_date or datetime.max.replace(tzinfo=ZoneInfo("UTC"))
     return enrollment_start_date < now < enrollment_end_date
 
 
@@ -133,7 +133,7 @@ def course_starts_within(start_date, look_ahead_days):
         start_date (datetime): The start datetime of the course in question.
         look_ahead_days (int): number of days to see in future for course start date.
     """
-    return datetime.now(utc) + timedelta(days=look_ahead_days) > start_date
+    return datetime.now(ZoneInfo("UTC")) + timedelta(days=look_ahead_days) > start_date
 
 
 def course_start_date_is_default(start, advertised_start):
@@ -179,10 +179,10 @@ def sorting_dates(start, advertised_start, announcement):
     try:
         start = dateutil.parser.parse(advertised_start)
         if start.tzinfo is None:
-            start = start.replace(tzinfo=utc)
+            start = start.replace(tzinfo=ZoneInfo("UTC"))
     except (TypeError, ValueError, AttributeError):
         start = start  # lint-amnesty, pylint: disable=self-assigning-variable
 
-    now = datetime.now(utc)
+    now = datetime.now(ZoneInfo("UTC"))
 
     return announcement, start, now

xmodule/graders.py

@@ -13,6 +13,7 @@
 from django.utils.translation import gettext_lazy as _
 
 from xmodule.util.misc import get_short_labeler
+from zoneinfo import ZoneInfo
 
 
 log = logging.getLogger("edx.courseware")

xmodule/lti_block.py

@@ -69,12 +69,12 @@
 from lxml import etree
 from oauthlib.oauth1.rfc5849 import signature
 from opaque_keys.edx.keys import CourseKey
-from pytz import UTC
 from web_fragments.fragment import Fragment
 from webob import Response
 from xblock.core import List, Scope, String, XBlock
 from xblock.fields import Boolean, Float
 from xblocks_contrib.lti import LTIBlock as _ExtractedLTIBlock
+from zoneinfo import ZoneInfo
 
 from common.djangoapps.xblock_django.constants import (
     ATTR_KEY_ANONYMOUS_USER_ID,
@@ -989,7 +989,7 @@ def is_past_due(self):
             close_date = due_date + self.graceperiod  # pylint: disable=no-member
         else:
             close_date = due_date
-        return close_date is not None and datetime.datetime.now(UTC) > close_date
+        return close_date is not None and datetime.datetime.now(ZoneInfo("UTC")) > close_date
 
 
 LTIBlock = (

xmodule/modulestore/__init__.py

@@ -16,7 +16,6 @@
 
 from opaque_keys.edx.keys import AssetKey, CourseKey
 from opaque_keys.edx.locations import Location  # For import backwards compatibility
-from pytz import UTC
 from sortedcontainers import SortedKeyList
 from xblock.core import XBlock
 from xblock.plugin import default_select
@@ -28,6 +27,7 @@
 from xmodule.assetstore import AssetMetadata
 from xmodule.errortracker import make_error_tracker
 from xmodule.util.misc import get_library_or_course_attribute
+from zoneinfo import ZoneInfo
 
 from .exceptions import InsufficientSpecificationError, InvalidLocationError
 
@@ -719,7 +719,7 @@ def _save_assets_by_type(self, course_key, asset_metadata_list, course_assets, u
                 ))
                 continue
             if not import_only:
-                asset_md.update({'edited_by': user_id, 'edited_on': datetime.datetime.now(UTC)})
+                asset_md.update({'edited_by': user_id, 'edited_on': datetime.datetime.now(ZoneInfo("UTC"))})
             asset_type = asset_md.asset_id.asset_type
             all_assets = assets_by_type[asset_type]
             all_assets.insert_or_update(asset_md)
@@ -862,7 +862,7 @@ def _block_matches(self, block, qualifiers):
         For substring matching:
             pass a regex object.
         For arbitrary function comparison such as date time comparison:
-            pass the function as in start=lambda x: x < datetime.datetime(2014, 1, 1, 0, tzinfo=pytz.UTC)
+            pass the function as in start=lambda x: x < datetime.datetime(2014, 1, 1, 0, tzinfo=ZoneInfo("UTC"))
 
         Args:
             block (dict, XBlock, or BlockData): either the BlockData (transformed from the db) -or-