46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
import datetime
|
|
|
|
import django.utils.timezone
|
|
from background_task import background
|
|
from telegram.bot import Bot
|
|
|
|
from . import models
|
|
|
|
|
|
@background()
|
|
def captcha_kicker():
|
|
for pendingCaptchaUser in models.PendingCaptchaUser.objects.all():
|
|
td: datetime.timedelta = (pendingCaptchaUser.created + datetime.timedelta(seconds=pendingCaptchaUser.lifetime)) - django.utils.timezone.now()
|
|
sec_remain = int(td.total_seconds())
|
|
if sec_remain < 0:
|
|
if sec_remain < -1800:
|
|
pendingCaptchaUser.delete()
|
|
continue
|
|
bot_token = pendingCaptchaUser.bot_token
|
|
bot = Bot(token=bot_token)
|
|
try:
|
|
bot.kick_chat_member(
|
|
chat_id=pendingCaptchaUser.group.telegram_id,
|
|
user_id=pendingCaptchaUser.user.telegram_id,
|
|
)
|
|
except:
|
|
pass
|
|
try:
|
|
bot.unban_chat_member(
|
|
chat_id=pendingCaptchaUser.group.telegram_id,
|
|
user_id=pendingCaptchaUser.user.telegram_id,
|
|
)
|
|
except:
|
|
pass
|
|
try:
|
|
bot.delete_message(
|
|
chat_id=pendingCaptchaUser.group.telegram_id,
|
|
message_id=pendingCaptchaUser.captcha_message_id,
|
|
)
|
|
except:
|
|
pass
|
|
pendingCaptchaUser.delete()
|
|
|
|
|
|
captcha_kicker(schedule=1, repeat=10, repeat_until=None)
|