Attempted a fix for #8

classes/callbacks.py

@@ -10,7 +10,7 @@ class CallbackLanguage:
     language: str
 
     @classmethod
-    def from_callback(cls, callback: CallbackQuery) -> "CallbackLanguage":
+    def from_callback(cls, callback: CallbackQuery):
         """Parse callback query and extract language data from it.
 
         ### Args:

classes/garbage_entry.py

@@ -23,7 +23,7 @@ class GarbageEntry:
     date: datetime
 
     @classmethod
-    async def from_dict(cls, data: Mapping[str, Any]) -> "GarbageEntry":
+    async def from_dict(cls, data: Mapping[str, Any]):
         """Generate GarbageEntry object from the mapping provided
 
         ### Args:
@@ -60,7 +60,7 @@ class GarbageEntry:
         )
 
     @classmethod
-    async def from_record(cls, data: Mapping[str, Any]) -> "GarbageEntry":
+    async def from_record(cls, data: Mapping[str, Any]):
         locations = [
             await Location.get(location_id) for location_id in data["locations"]
         ]

classes/location.py

@@ -28,7 +28,7 @@ class Location:
     timezone: Union[BaseTzInfo, DstTzInfo]
 
     @classmethod
-    async def get(cls, id: int) -> "Location":
+    async def get(cls, id: int):
         db_entry = await col_locations.find_one({"id": id})
 
         if db_entry is None:
@@ -40,7 +40,7 @@ class Location:
         return cls(**db_entry)
 
     @classmethod
-    async def find(cls, name: str) -> "Location":
+    async def find(cls, name: str):
         db_entry = await col_locations.find_one({"name": {"$regex": name}})
 
         if db_entry is None:
@@ -52,7 +52,7 @@ class Location:
         return cls(**db_entry)
 
     @classmethod
-    async def nearby(cls, lat: float, lon: float) -> "Location":
+    async def nearby(cls, lat: float, lon: float):
         db_entry = await col_locations.find_one({"location": {"$near": [lon, lat]}})
 
         if db_entry is None:

classes/pyrouser.py

@@ -1,9 +1,8 @@
 import logging
 from dataclasses import dataclass
-from datetime import datetime, timedelta
-from typing import Any, Mapping, Tuple, Union
+from datetime import datetime, timedelta, timezone
+from typing import Any, Union
 
-import pytz
 from bson import ObjectId
 
 from classes.location import Location
@@ -46,7 +45,7 @@ class PyroUser:
         offset: int = 1,
         time_hour: int = 16,
         time_minute: int = 0,
-    ) -> "PyroUser":
+    ):
         db_entry = await col_users.find_one({"id": id})
 
         if db_entry is None:
@@ -74,7 +73,7 @@ class PyroUser:
         return cls(**db_entry)
 
     @classmethod
-    async def from_dict(cls, **kwargs) -> "PyroUser":
+    async def from_dict(cls, **kwargs):
         if "location" in kwargs:
             try:
                 kwargs["location"] = await Location.get(kwargs["location"])  # type: ignore
@@ -82,85 +81,46 @@ class PyroUser:
                 kwargs["location"] = None  # type: ignore
         return cls(**kwargs)
 
-    async def update_locale(self, locale: Union[str, None]) -> Union[str, None]:
+    async def update_locale(self, locale: Union[str, None]) -> None:
         """Change user's locale stored in the database.
 
         ### Args:
             * locale (`Union[str, None]`): New locale to be set.
         """
-
         logger.debug("%s's locale has been set to %s", self.id, locale)
-
         await col_users.update_one({"_id": self._id}, {"$set": {"locale": locale}})
-
         self.locale = locale
 
-        return self.locale
-
-    async def update_state(self, enabled: bool = False) -> bool:
+    async def update_state(self, enabled: bool = False) -> None:
         logger.debug("%s's state has been set to %s", self.id, enabled)
-
         await col_users.update_one({"_id": self._id}, {"$set": {"enabled": enabled}})
-
         self.enabled = enabled
 
-        return self.enabled
-
-    async def update_location(self, location_id: int = 0) -> Location:
+    async def update_location(self, location_id: int = 0) -> None:
         logger.debug("%s's location has been set to %s", self.id, location_id)
-
         await col_users.update_one(
             {"_id": self._id}, {"$set": {"location": location_id}}
         )
+        self.location = await Location.get(location_id)
 
-        location = await Location.get(location_id)
-
-        # Execute if timezones of old and new locations are different
-        if self.location and (self.location.timezone.zone != location.timezone.zone):
-            # Get UTC time for selected reminder time
-            now_utc = datetime.now(pytz.utc).replace(
-                hour=self.time_hour, minute=self.time_minute, second=0, microsecond=0
-            )
-
-            # Get the time for the reminder time of old and new location
-            local_old = now_utc.astimezone(self.location.timezone)
-            local_new = (
-                location.timezone.localize(local_old.replace(tzinfo=None))
-            ).astimezone(pytz.utc)
-
-            # Update the time to match the new timezone
-            await self.update_time(hour=local_new.hour, minute=local_new.minute)
-
-        self.location = location
-
-        return self.location
-
-    async def update_offset(self, offset: int = 1) -> int:
+    async def update_offset(self, offset: int = 1) -> None:
         logger.debug("%s's offset has been set to %s", self.id, offset)
-
         await col_users.update_one({"_id": self._id}, {"$set": {"offset": offset}})
-
         self.offset = offset
 
-        return offset
-
-    async def update_time(self, hour: int = 16, minute: int = 0) -> Tuple[int, int]:
+    async def update_time(self, hour: int = 16, minute: int = 0) -> None:
         logger.debug("%s's time has been set to %s h. %s m.", self.id, hour, minute)
-
         await col_users.update_one(
             {"_id": self._id}, {"$set": {"time_hour": hour, "time_minute": minute}}
         )
-
         self.time_hour = hour
         self.time_minute = minute
 
-        return self.time_hour, self.time_minute
-
     async def delete(self) -> None:
         logger.debug("%s's data has been deleted", self.id)
         await col_users.delete_one({"_id": self._id})
 
-    async def checkout(self) -> Mapping[str, Any]:
+    async def checkout(self) -> Any:
         logger.debug("%s's data has been checked out", self.id)
         db_entry = await col_users.find_one({"_id": self._id})
 
@@ -195,7 +155,7 @@ class PyroUser:
             logger.warning("Location %s does not have a timezone set", self.location.id)
 
         return (
-            datetime.now(self.location.timezone or pytz.utc) + timedelta(days=1)
+            datetime.now(self.location.timezone or timezone.utc) + timedelta(days=1)
         ).replace(hour=0, minute=0, second=0, microsecond=0)
 
     def get_reminder_time(self) -> datetime:
@@ -220,12 +180,12 @@ class PyroUser:
             logger.warning("Location %s does not have a timezone set", self.location.id)
 
         return (
-            datetime.now(pytz.utc)
+            datetime.now(timezone.utc)
             .replace(
                 hour=self.time_hour,
                 minute=self.time_minute,
                 second=0,
                 microsecond=0,
             )
-            .astimezone(self.location.timezone or pytz.utc)
+            .astimezone(self.location.timezone or timezone.utc)
         )

migrations/202405261500.py

@@ -13,4 +13,4 @@ class Migration(BaseMigration):
 
     def downgrade(self):
         sync.config_delete("update_checker", missing_ok=True)
-        sync.config_delete("url_updater", "strings", missing_ok=True)
+        sync.config_delete("url_updater", "strings")

modules/reminder.py

@@ -1,7 +1,6 @@
 import logging
-from datetime import datetime
+from datetime import datetime, timezone
 
-import pytz
 from bson import json_util
 from libbot.pyrogram.classes import PyroClient
 
@@ -15,7 +14,7 @@ logger = logging.getLogger(__name__)
 
 
 async def remind(app: PyroClient) -> None:
-    utcnow = datetime.now(pytz.utc)
+    utcnow = datetime.now(timezone.utc)
 
     logger.debug("Performing reminder lookup for %s (UTCNOW)", utcnow)
 
@@ -43,7 +42,13 @@ async def remind(app: PyroClient) -> None:
             logger.warning("Skipping reminder for %s due to invalid location", user.id)
             continue
 
-        user_date = user.get_reminder_date().replace(tzinfo=None)
+        try:
+            user_date = user.get_reminder_date()
+        except AttributeError:
+            logger.warning(
+                "Skipping reminder for %s due to missing attributes", user.id
+            )
+            continue
 
         entries = await col_entries.find(
             {

modules/utils.py

@@ -1,7 +1,8 @@
 from datetime import datetime
 from typing import Union
 
-import pytz
+from pytz import UTC
+from pytz import timezone as pytz_timezone
 
 
 def to_utc(date: datetime, timezone: Union[str, None] = None) -> datetime:
@@ -17,9 +18,7 @@ def to_utc(date: datetime, timezone: Union[str, None] = None) -> datetime:
         * `datetime`: Timezone unaware datetime in UTC with timezone's offset applied to it.
     """
     timezone = "UTC" if timezone is None else timezone
-    return (
-        pytz.timezone(timezone).localize(date).astimezone(pytz.utc).replace(tzinfo=None)
-    )
+    return pytz_timezone(timezone).localize(date).astimezone(UTC).replace(tzinfo=None)
 
 
 def from_utc(date: datetime, timezone: Union[str, None] = None) -> datetime:
@@ -36,5 +35,8 @@ def from_utc(date: datetime, timezone: Union[str, None] = None) -> datetime:
     """
     timezone = "UTC" if timezone is None else timezone
     return (
-        pytz.utc.localize(date).astimezone(pytz.timezone(timezone)).replace(tzinfo=None)
+        pytz_timezone("UTC")
+        .localize(date)
+        .astimezone(pytz_timezone(timezone))
+        .replace(tzinfo=None)
     )

plugins/commands/set_offset.py

@@ -1,7 +1,6 @@
 import logging
-from datetime import datetime
+from datetime import datetime, timezone
 
-import pytz
 from convopyro import listen_message
 from pyrogram import filters
 from pyrogram.types import ForceReply, Message, ReplyKeyboardRemove
@@ -66,9 +65,9 @@ async def command_set_offset(app: PyroClient, message: Message):
     logger.info("User %s has set offset to %s", user.id, offset)
 
     garbage_time = (
-        datetime.now(pytz.utc)
+        datetime.now(timezone.utc)
         .replace(hour=user.time_hour, minute=user.time_minute)
-        .astimezone(user.location.timezone or pytz.utc)
+        .astimezone(user.location.timezone or timezone.utc)
     )
 
     await answer.reply_text(

plugins/commands/set_time.py

@@ -1,7 +1,6 @@
 import logging
-from datetime import datetime
+from datetime import datetime, timezone
 
-import pytz
 from convopyro import listen_message
 from pyrogram import filters
 from pyrogram.types import ForceReply, Message, ReplyKeyboardRemove
@@ -55,30 +54,34 @@ async def command_set_time(app: PyroClient, message: Message):
 
         break
 
-    # Time we got from the user
-    parsed_time = datetime.strptime(answer.text, "%H:%M")
+    now = datetime.now()
 
-    # Datetime user means in their timezone
-    user_time = datetime.now(user.location.timezone).replace(
-        hour=parsed_time.hour, minute=parsed_time.minute, second=0, microsecond=0
+    parsed_time = datetime.strptime(answer.text, "%H:%M").replace(
+        year=now.year,
+        month=now.month,
+        day=now.day,
+        second=0,
+        microsecond=0,
+        tzinfo=timezone.utc,
     )
 
-    # Datetime in user's timezone moved to UTC timezone
-    utc_time = user_time.astimezone(pytz.utc)
+    user_time = parsed_time.astimezone(user.location.timezone or timezone.utc)
 
-    await user.update_time(hour=utc_time.hour, minute=utc_time.minute)
+    await user.update_time(hour=user_time.hour, minute=user_time.minute)
 
     logger.info(
         "User %s has selected notification time of %s (%s UTC)",
         user.id,
+        parsed_time.strftime("%H:%M"),
         user_time.strftime("%H:%M"),
-        utc_time.strftime("%H:%M"),
     )
 
+    garbage_time = parsed_time.strftime(app._("time", "formats", locale=user.locale))
+
     await answer.reply_text(
         app._("set_time_finished", "messages", locale=user.locale).format(
             offset=user.offset,
-            time=user_time.strftime(app._("time", "formats", locale=user.locale)),
+            time=garbage_time,
             toggle_notice=(
                 "" if user.enabled else app._("toggle", "messages", locale=user.locale)
             ),

plugins/commands/setup.py

@@ -72,9 +72,12 @@ async def command_setup(app: PyroClient, message: Message):
 
     await user.update_location(location.id)
 
-    user_time = user.get_reminder_time().strftime(
-        app._("time", "formats", locale=user.locale)
-    )
+    try:
+        user_time = user.get_reminder_time().strftime(
+            app._("time", "formats", locale=user.locale)
+        )
+    except AttributeError:
+        user_time = "N/A"
 
     await message.reply_text(
         app._("setup_finished", "messages", locale=user.locale).format(

plugins/commands/start.py

@@ -88,9 +88,12 @@ async def command_start(app: PyroClient, message: Message):
 
         await user.update_location(location.id)
 
-        user_time = user.get_reminder_time().strftime(
-            app._("time", "formats", locale=user.locale)
-        )
+        try:
+            user_time = user.get_reminder_time().strftime(
+                app._("time", "formats", locale=user.locale)
+            )
+        except AttributeError:
+            user_time = "N/A"
 
         await answer.reply_text(
             app._("start_selection_yes", "messages", locale=user.locale).format(

plugins/commands/upcoming.py

@@ -1,6 +1,5 @@
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 
-import pytz
 from pyrogram import filters
 from pyrogram.types import Message
 
@@ -24,11 +23,11 @@ async def command_upcoming(app: PyroClient, message: Message):
 
     date_min = (
         datetime.now(user.location.timezone).replace(second=0, microsecond=0)
-    ).replace(tzinfo=pytz.utc)
+    ).replace(tzinfo=timezone.utc)
     date_max = (
         datetime.now(user.location.timezone).replace(second=0, microsecond=0)
         + timedelta(days=30)
-    ).replace(tzinfo=pytz.utc)
+    ).replace(tzinfo=timezone.utc)
 
     entries = [
         await GarbageEntry.from_record(entry)