Merge pull request 'Update to 3.0.0' (#52) from dev into main
All checks were successful
Tests / test (3.10) (push) Successful in 1m15s
Tests / test (3.11) (push) Successful in 1m14s
Tests / test (3.8) (push) Successful in 1m14s
Tests / test (3.9) (push) Successful in 1m22s

Reviewed-on: #52
This commit is contained in:
Profitroll 2024-01-04 00:06:50 +02:00
commit 5ba763246b
6 changed files with 93 additions and 21 deletions

View File

@ -3,9 +3,9 @@ build==1.0.3
isort==5.13.2
mypy==1.8.0
pylint==3.0.3
pytest-asyncio==0.23.2
pytest-asyncio==0.23.3
pytest-cov==4.1.0
pytest==7.4.3
pytest==7.4.4
tox==4.11.4
types-aiofiles==23.2.0.0
types-ujson==5.9.0.0

View File

@ -1,2 +1,2 @@
apscheduler~=3.10.4
pyrogram~=2.0.106
pyrofork~=2.3.16.post1

View File

@ -1,4 +1,4 @@
__version__ = "2.1.0"
__version__ = "3.0.0"
__license__ = "GPL3"
__author__ = "Profitroll"

View File

@ -103,7 +103,7 @@ class PyroClient(Client):
self.bot_locale: BotLocale = BotLocale(
default_locale=self.config["locale"],
locales_root=(Path("locale") if locales_root is None else locales_root)
locales_root=(Path("locale") if locales_root is None else locales_root),
)
self.default_locale: str = self.bot_locale.default
self.locales: dict = self.bot_locale.locales
@ -129,6 +129,53 @@ class PyroClient(Client):
getpid(),
)
# Register default bot's info
try:
await self.set_bot_info(
name=self._("name", "bot"),
about=self._("about", "bot"),
description=self._("description", "bot"),
lang_code="",
)
logger.info(
"Bot's info for the default locale %s has been updated",
self.default_locale,
)
except KeyError:
logger.warning(
"Default locale %s has incorrect keys or values in bot section",
self.default_locale,
)
# Register bot's info for each available locale
for locale_code in self.locales:
locale = self.locales[locale_code]
if "metadata" not in locale or ("codes" not in locale["metadata"]):
logger.warning(
"Locale %s is missing metadata or metadata.codes key", locale_code
)
continue
for code in locale["metadata"]["codes"]:
try:
await self.set_bot_info(
name=locale["bot"]["name"],
about=locale["bot"]["about"],
description=locale["bot"]["description"],
lang_code=code,
)
logger.info(
"Bot's info for the locale %s has been updated",
self.default_locale,
)
except KeyError:
logger.warning(
"Locale %s has incorrect keys or values in bot section",
locale_code,
)
# Send a message to the bot's reports chat about the startup
try:
await self.send_message(
chat_id=self.owner
@ -136,22 +183,23 @@ class PyroClient(Client):
else self.config["reports"]["chat_id"],
text=f"Bot started PID `{getpid()}`",
)
if self.scheduler is None:
return
if register_commands:
self.scheduler.add_job(
self.register_commands,
trigger="date",
run_date=datetime.now() + timedelta(seconds=5),
kwargs={"command_sets": await self.collect_commands()},
)
self.scheduler.start()
except BadRequest:
logger.warning("Unable to send message to report chat.")
if self.scheduler is None:
return
# Schedule the task to register all commands
if register_commands:
self.scheduler.add_job(
self.register_commands,
trigger="date",
run_date=datetime.now() + timedelta(seconds=5),
kwargs={"command_sets": await self.collect_commands()},
)
self.scheduler.start()
async def stop(self, exit_completely: bool = True) -> None:
try:
await self.send_message(
@ -170,16 +218,16 @@ class PyroClient(Client):
if exit_completely:
try:
exit()
except SystemExit as exp:
except SystemExit as exc:
raise SystemExit(
"Bot has been shut down, this is not an application error!"
) from exp
) from exc
async def collect_commands(self) -> Union[List[CommandSet], None]:
"""Gather list of the bot's commands
### Returns:
* `List[CommandSet]`: List of the commands' sets
* `List[CommandSet]`: List of the commands' sets.
"""
command_sets = None

View File

@ -1,4 +1,16 @@
{
"metadata": {
"flag": "🇬🇧",
"name": "English",
"codes": [
"en"
]
},
"bot": {
"name": "",
"about": "",
"description": ""
},
"foo": "bar",
"messages": {
"example": "okay"

View File

@ -1,4 +1,16 @@
{
"metadata": {
"flag": "🇺🇦",
"name": "Українська",
"codes": [
"uk"
]
},
"bot": {
"name": "",
"about": "",
"description": ""
},
"foo": "бар",
"messages": {
"example": "окей"