36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
from config import Config
|
|
from core.builtins import Bot
|
|
from database import BotDBUtil
|
|
|
|
WARNING_COUNTS = Config('tos_warning_counts', 5)
|
|
|
|
|
|
async def warn_target(msg: Bot.MessageSession, reason=None):
|
|
if WARNING_COUNTS >= 1:
|
|
current_warns = int(msg.target.sender_info.query.warns) + 1
|
|
msg.target.sender_info.edit('warns', current_warns)
|
|
warn_template = [msg.locale.t("tos.warning")]
|
|
if reason is not None:
|
|
warn_template.append(msg.locale.t("tos.reason") + reason)
|
|
if current_warns < WARNING_COUNTS:
|
|
warn_template.append(msg.locale.t('tos.warning.count', current_warns=current_warns, warn_counts=WARNING_COUNTS))
|
|
if current_warns <= 2:
|
|
warn_template.append(msg.locale.t('tos.warning.appeal', issue_url=Config('issue_url')))
|
|
if current_warns == WARNING_COUNTS:
|
|
warn_template.append(msg.locale.t('tos.warning.last'))
|
|
if current_warns > WARNING_COUNTS:
|
|
msg.target.sender_info.edit('isInBlockList', True)
|
|
return
|
|
await msg.send_message('\n'.join(warn_template))
|
|
|
|
|
|
async def pardon_user(user: str):
|
|
BotDBUtil.SenderInfo(user).edit('warns', 0)
|
|
|
|
|
|
async def warn_user(user: str, count=1):
|
|
current_warns = int(BotDBUtil.SenderInfo(user).query.warns) + count
|
|
BotDBUtil.SenderInfo(user).edit('warns', current_warns)
|
|
if current_warns > WARNING_COUNTS and WARNING_COUNTS >= 1:
|
|
BotDBUtil.SenderInfo(user).edit('isInBlockList', True)
|
|
return current_warns
|