10 Commits

11 changed files with 30 additions and 82 deletions

View File

@@ -19,11 +19,11 @@ logger = logging.getLogger(__name__)
class PyroClient(LibPyroClient): class PyroClient(LibPyroClient):
def __init__(self, **kwargs): def __init__(self, **kwargs):
self.__version__ = (0, 1, 3) self.__version__ = (0, 1, 2)
super().__init__(**kwargs) super().__init__(**kwargs)
self.updater = Updater() self.updater = Updater(ClientSession())
self.contexts = [] self.contexts = []
if self.scheduler is not None: if self.scheduler is not None:
@@ -91,7 +91,6 @@ class PyroClient(LibPyroClient):
] ]
async def check_updates(self) -> None: async def check_updates(self) -> None:
"""Check for updates and send a message to the owner if newer version was found"""
if await self.updater.check_updates( if await self.updater.check_updates(
self.__version__, self.config["strings"]["url_updater"] self.__version__, self.config["strings"]["url_updater"]
): ):

View File

@@ -88,7 +88,8 @@ class PyroUser:
### Args: ### Args:
* locale (`Union[str, None]`): New locale to be set. * locale (`Union[str, None]`): New locale to be set.
""" """
logger.info("%s's locale has been set to %s", self.id, locale)
logger.debug("%s's locale has been set to %s", self.id, locale)
await col_users.update_one({"_id": self._id}, {"$set": {"locale": locale}}) await col_users.update_one({"_id": self._id}, {"$set": {"locale": locale}})
@@ -97,15 +98,7 @@ class PyroUser:
return self.locale return self.locale
async def update_state(self, enabled: bool = False) -> bool: async def update_state(self, enabled: bool = False) -> bool:
"""Update user's state (enabled/disabled) logger.debug("%s's state has been set to %s", self.id, enabled)
### Args:
* enabled (`bool`, *optional*): Whether the user is enabled. Defaults to `False`.
### Returns:
* `bool`: User's current state
"""
logger.info("%s's state has been set to %s", self.id, enabled)
await col_users.update_one({"_id": self._id}, {"$set": {"enabled": enabled}}) await col_users.update_one({"_id": self._id}, {"$set": {"enabled": enabled}})
@@ -113,16 +106,8 @@ class PyroUser:
return self.enabled return self.enabled
async def update_location(self, location_id: int) -> Location: async def update_location(self, location_id: int = 0) -> Location:
"""Update user's location and move their time to the new timezone (if the user had a location set previously) logger.debug("%s's location has been set to %s", self.id, location_id)
### Args:
* location_id (`int`): ID of the location
### Returns:
`Location`: New location
"""
logger.info("%s's location has been set to %s", self.id, location_id)
await col_users.update_one( await col_users.update_one(
{"_id": self._id}, {"$set": {"location": location_id}} {"_id": self._id}, {"$set": {"location": location_id}}
@@ -151,15 +136,7 @@ class PyroUser:
return self.location return self.location
async def update_offset(self, offset: int = 1) -> int: async def update_offset(self, offset: int = 1) -> int:
"""Update the offset of the reminder (in days) logger.debug("%s's offset has been set to %s", self.id, offset)
### Args:
* offset (`int`, *optional*): Offset in days. Defaults to `1`.
### Returns:
* `int`: Offset in days
"""
logger.info("%s's offset has been set to %s", self.id, offset)
await col_users.update_one({"_id": self._id}, {"$set": {"offset": offset}}) await col_users.update_one({"_id": self._id}, {"$set": {"offset": offset}})
@@ -168,16 +145,7 @@ class PyroUser:
return 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) -> Tuple[int, int]:
"""Update the time of the reminder (hour and minute, for UTC timezone) logger.debug("%s's time has been set to %s h. %s m.", self.id, hour, minute)
### Args:
* hour (`int`, *optional*): Hour of the reminder. Defaults to `16`.
* minute (`int`, *optional*): Minute of the reminder. Defaults to `0`.
### Returns:
* `Tuple[int, int]`: Hour and minute of the reminder
"""
logger.info("%s's time has been set to %s h. %s m.", self.id, hour, minute)
await col_users.update_one( await col_users.update_one(
{"_id": self._id}, {"$set": {"time_hour": hour, "time_minute": minute}} {"_id": self._id}, {"$set": {"time_hour": hour, "time_minute": minute}}
@@ -189,27 +157,16 @@ class PyroUser:
return self.time_hour, self.time_minute return self.time_hour, self.time_minute
async def delete(self) -> None: async def delete(self) -> None:
"""Delete the database record of the user""" logger.debug("%s's data has been deleted", self.id)
logger.info("%s's data has been deleted", self.id)
await col_users.delete_one({"_id": self._id}) await col_users.delete_one({"_id": self._id})
async def checkout(self) -> Mapping[str, Any]: async def checkout(self) -> Mapping[str, Any]:
"""Checkout the user's database record logger.debug("%s's data has been checked out", self.id)
### Raises:
* `KeyError`: Database record of the user was not found
### Returns:
* `Mapping[str, Any]`: Database record
"""
logger.info("%s's data has been checked out", self.id)
db_entry = await col_users.find_one({"_id": self._id}) db_entry = await col_users.find_one({"_id": self._id})
if db_entry is None: if db_entry is None:
raise KeyError( raise KeyError(
f"DB record with id {self._id} of user {self.id} was not found" f"DB record with id {self._id} of user {self.id} is not found"
) )
del db_entry["_id"] # type: ignore del db_entry["_id"] # type: ignore

View File

@@ -1,5 +1,5 @@
import logging import logging
from typing import Any, Dict, Tuple, Union from typing import Any, Dict, Tuple
from aiohttp import ClientSession from aiohttp import ClientSession
@@ -7,15 +7,12 @@ logger = logging.getLogger(__name__)
class Updater: class Updater:
def __init__(self, client_session: Union[ClientSession, None] = None) -> None: def __init__(self, client_session: ClientSession) -> None:
self.client_session: Union[ClientSession, None] = client_session self.client_session: ClientSession = client_session
async def check_updates( async def check_updates(
self, version_current: Tuple[int, int, int], api_url: str self, version_current: Tuple[int, int, int], api_url: str
) -> bool: ) -> bool:
if not self.client_session:
self.client_session = ClientSession()
response = await self.client_session.get(api_url) response = await self.client_session.get(api_url)
if response.status != 200: if response.status != 200:
@@ -33,9 +30,6 @@ class Updater:
) )
async def get_latest_release(self, api_url: str) -> Dict[str, Any]: async def get_latest_release(self, api_url: str) -> Dict[str, Any]:
if not self.client_session:
self.client_session = ClientSession()
response = await self.client_session.get(api_url) response = await self.client_session.get(api_url)
if response.status != 200: if response.status != 200:

View File

@@ -8,7 +8,7 @@
}, },
"bot": { "bot": {
"name": "Garbage Reminder", "name": "Garbage Reminder",
"about": "Nie wieder Müllabfuhrtermin verpassen. Mehr erfahren: https://garbagebot.eu", "about": "Nie wieder Müllabfuhrtermin verpassen. Quellcode: https://garbagebot.eu",
"description": "Sie können Erinnerungen an die Müllabfuhr für Orte Ihrer Wahl erhalten.\n\nVerwenden Sie /help, um die Funktionsweise des Bots besser zu verstehen, oder verwenden Sie /setup, um Ihre Erinnerungen zu konfigurieren." "description": "Sie können Erinnerungen an die Müllabfuhr für Orte Ihrer Wahl erhalten.\n\nVerwenden Sie /help, um die Funktionsweise des Bots besser zu verstehen, oder verwenden Sie /setup, um Ihre Erinnerungen zu konfigurieren."
}, },
"formats": { "formats": {
@@ -100,4 +100,4 @@
"callbacks": { "callbacks": {
"locale_set": "Ihre Sprache ist jetzt: {locale}" "locale_set": "Ihre Sprache ist jetzt: {locale}"
} }
} }

View File

@@ -8,7 +8,7 @@
}, },
"bot": { "bot": {
"name": "Garbage Reminder", "name": "Garbage Reminder",
"about": "Never forget about garbage collection again. Learn more: https://garbagebot.eu", "about": "Never forget about garbage collection again. Source code: https://garbagebot.eu",
"description": "You can receive reminders about garbage collection for locations of your choice.\n\nUse /help to better understand how the bot works or use /setup to configure your reminders." "description": "You can receive reminders about garbage collection for locations of your choice.\n\nUse /help to better understand how the bot works or use /setup to configure your reminders."
}, },
"formats": { "formats": {

View File

@@ -8,7 +8,7 @@
}, },
"bot": { "bot": {
"name": "Garbage Reminder 🇺🇦", "name": "Garbage Reminder 🇺🇦",
"about": "Більше ніколи не забувайте про вивезення сміття. Дізнатись більше: https://garbagebot.eu", "about": "Більше ніколи не забувайте про вивезення сміття. Вихідний код: https://garbagebot.eu",
"description": "Ви можете отримувати нагадування про вивезення сміття для обраних вами місць.\n\nВикористовуйте /help, щоб краще зрозуміти, як працює бот, або /setup, щоб налаштувати нагадування." "description": "Ви можете отримувати нагадування про вивезення сміття для обраних вами місць.\n\nВикористовуйте /help, щоб краще зрозуміти, як працює бот, або /setup, щоб налаштувати нагадування."
}, },
"formats": { "formats": {
@@ -100,4 +100,4 @@
"set_offset": "Кількість днів", "set_offset": "Кількість днів",
"set_time": "Час у вигляді ГГ:ХХ" "set_time": "Час у вигляді ГГ:ХХ"
} }
} }

View File

@@ -8,7 +8,7 @@
}, },
"bot": { "bot": {
"name": "Garbage Reminder 🇺🇦", "name": "Garbage Reminder 🇺🇦",
"about": "Більше ніколи не забувайте про вивезення сміття. Дізнатись більше: https://garbagebot.eu", "about": "Більше ніколи не забувайте про вивезення сміття. Вихідний код: https://garbagebot.eu",
"description": "Ви можете отримувати нагадування про вивезення сміття для обраних вами місць.\n\nВикористовуйте /help, щоб краще зрозуміти, як працює бот, або /setup, щоб налаштувати нагадування." "description": "Ви можете отримувати нагадування про вивезення сміття для обраних вами місць.\n\nВикористовуйте /help, щоб краще зрозуміти, як працює бот, або /setup, щоб налаштувати нагадування."
}, },
"formats": { "formats": {
@@ -100,4 +100,4 @@
"set_offset": "Кількість днів", "set_offset": "Кількість днів",
"set_time": "Час у вигляді ГГ:ХХ" "set_time": "Час у вигляді ГГ:ХХ"
} }
} }

View File

@@ -4,7 +4,6 @@ from argparse import ArgumentParser
from os import getpid from os import getpid
from pathlib import Path from pathlib import Path
from aiohttp import ClientSession
from convopyro import Conversation from convopyro import Conversation
from libbot import sync from libbot import sync
@@ -42,8 +41,7 @@ def main():
exit() exit()
client = PyroClient( client = PyroClient(
scheduler=scheduler, scheduler=scheduler, commands_source=sync.json_read(Path("commands.json"))
commands_source=sync.json_read(Path("commands.json")),
) )
Conversation(client) Conversation(client)

View File

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

View File

@@ -13,7 +13,7 @@ async def command_toggle(app: PyroClient, message: Message):
await user.update_state(not user.enabled) await user.update_state(not user.enabled)
if not user.enabled: if user.enabled:
await message.reply_text( await message.reply_text(
app._("toggle_disabled", "messages", locale=user.locale) app._("toggle_disabled", "messages", locale=user.locale)
) )

View File

@@ -1,12 +1,12 @@
aiohttp~=3.10.2 aiohttp~=3.9.5
apscheduler~=3.10.4 apscheduler~=3.10.4
async_pymongo==0.1.9
convopyro==0.5 convopyro==0.5
mongodb-migrations==1.3.1 mongodb-migrations==1.3.1
pytz>=2024.1 pytz<=2023.2
tgcrypto==1.2.5 tgcrypto==1.2.5
ujson>=5.0.0 ujson>=5.0.0
uvloop==0.20.0 uvloop==0.19.0
--extra-index-url https://git.end-play.xyz/api/packages/profitroll/pypi/simple --extra-index-url https://git.end-play.xyz/api/packages/profitroll/pypi/simple
libbot[speed,pyrogram]==3.2.3 async_pymongo==0.1.4
libbot[speed,pyrogram]==3.2.2
pykeyboard==0.1.7 pykeyboard==0.1.7