This commit closes #13

This commit is contained in:
2023-08-14 12:11:53 +02:00
parent 7a587ececa
commit fdd5ad7ff0
8 changed files with 91 additions and 18 deletions

View File

@@ -22,6 +22,7 @@ class PyroGroup:
id: int
locale: Union[str, None]
locale_auto: bool
ban_failed: bool
@classmethod
async def create_if_not_exists(
@@ -29,6 +30,7 @@ class PyroGroup:
id: int,
locale: Union[str, None] = sync.config_get("locale", "defaults", "group"),
locale_auto: bool = sync.config_get("locale_auto", "defaults", "group"),
ban_failed: bool = sync.config_get("ban_failed", "defaults", "group"),
):
db_entry = await col_groups.find_one(
{
@@ -42,6 +44,7 @@ class PyroGroup:
"id": id,
"locale": locale,
"locale_auto": locale_auto,
"ban_failed": ban_failed,
}
)
db_entry = await col_groups.find_one({"_id": inserted.inserted_id})
@@ -65,6 +68,16 @@ class PyroGroup:
{"_id": self._id}, {"$set": {"locale_auto": enabled}}
)
async def set_ban_failed(self, enabled: bool) -> None:
logger.debug(
"Banning users that failed the captcha in group %s has been set to %s",
self.id,
enabled,
)
await col_groups.update_one(
{"_id": self._id}, {"$set": {"ban_failed": enabled}}
)
# Group settings
# User locale
def select_locale(