Добавил проверку на наличие домена в бд

This commit is contained in:
2025-07-26 01:05:13 +03:00
parent 803cdefe99
commit 8d5bac64fa

View File

@@ -729,11 +729,18 @@ class ModerModule(commands.Cog):
if not ext.domain or not ext.suffix: if not ext.domain or not ext.suffix:
return None return None
return f"{ext.domain}.{ext.suffix}".lower() return f"{ext.domain}.{ext.suffix}".lower()
new_link = link if not "http" in link else extract_root_domain(link) new_link = extract_root_domain(link)
if not new_link: if not new_link:
await ctx.send(embed = disnake.Embed(description = f'Некорректная ссылка!', colour = 0xff9900)) await ctx.send(embed = disnake.Embed(description = f'Некорректная ссылка!', colour = 0xff9900))
return 1 return 1
stmt = self.DataBaseManager.select(аllowed_domains_model).where(аllowed_domains_model.domain == new_link)
link_in_wl = (await session.execute(stmt)).scalars().first()
if link_in_wl is not None:
await ctx.send(embed = disnake.Embed(description = f'Этот домен уже есть в белом листе!', colour = 0xff9900))
return 1
domain = аllowed_domains_model(domain = new_link, initiator_id = ctx.author.id) domain = аllowed_domains_model(domain = new_link, initiator_id = ctx.author.id)
session.add(domain) session.add(domain)