main.py

import logging
import json
import re
import sys
from threading import Timer

from environs import Env
import telebot
from telebot import types

env = Env()
env.read_env()
main_admin = env.int('main_admin', default=None)

settings = {
    '-1001749011309': dict(greeting_text="Здравствуй {username}, очень рады тебя приветствовать в нашем чате.",
                           greeting_video='https://telegra.ph/file/4a150e0856f8c20ca65ea.mp4',
                           greeting_timeout=10),
    'admins': [main_admin] if main_admin else []
}
default = dict(
    greeting_text=env('greeting_text'),
    greeting_video=env('greeting_video', default=None),
    greeting_timeout=env.int('greeting_timeout', default=60),
)
restricted: set = set(env.list('restricted', default=['url', 'tag', 'photo', 'document', 'voice']))
bot = telebot.TeleBot(env('bot_token'))

logging.basicConfig(filename='moderator.log', format='%(asctime)s - %(levelname)s - %(message)s', level=logging.INFO)
logging.getLogger().addHandler(logging.StreamHandler(sys.stdout))

tag_regex = "@[a-zA-Z]"
url_regex = r"\b((?:https?://)?(?:(?:www\.)?" \
            r"(?:[\da-z\.-]+)\.(?:[a-z]{2,6})|(?:(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}" \
            r"(?:25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)|(?:(?:[0-9a-fA-F]{1,4}:){7,7}[0-9a-fA-F]{1,4}|" \
            r"(?:[0-9a-fA-F]{1,4}:){1,7}:|(?:[0-9a-fA-F]{1,4}:){1,6}:[0-9a-fA-F]{1,4}|(?:[0-9a-fA-F]{1,4}:){1,5}" \
            r"(?::[0-9a-fA-F]{1,4}){1,2}|(?:[0-9a-fA-F]{1,4}:){1,4}(?::[0-9a-fA-F]{1,4}){1,3}|" \
            r"(?:[0-9a-fA-F]{1,4}:){1,3}(?::[0-9a-fA-F]{1,4}){1,4}|(?:[0-9a-fA-F]{1,4}:){1,2}" \
            r"(?::[0-9a-fA-F]{1,4}){1,5}|[0-9a-fA-F]{1,4}:(?:(?::[0-9a-fA-F]{1,4}){1,6})|:" \
            r"(?:(?::[0-9a-fA-F]{1,4}){1,7}|:)|fe80:(?::[0-9a-fA-F]{0,4}){0,4}%[0-9a-zA-Z]{1,}|::" \
            r"(?:ffff(?::0{1,4}){0,1}:){0,1}(?:(?:25[0-5]|(?:2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}" \
            r"(?:25[0-5]|(?:2[0-4]|1{0,1}[0-9]){0,1}[0-9])|(?:[0-9a-fA-F]{1,4}:){1,4}:(?:" \
            r"(?:25[0-5]|(?:2[0-4]|1{0,1}[0-9]){0,1}[0-9])\.){3,3}(?:25[0-5]|(?:2[0-4]|1{0,1}[0-9]){0,1}[0-9])))" \
            r"(?::[0-9]{1,4}|[1-5][0-9]{4}|6[0-4][0-9]{3}|65[0-4][0-9]{2}|655[0-2][0-9]|6553[0-5])?(?:/[\w\.-]*)*/?)\b"

temp = dict()


@bot.message_handler(content_types=['left_chat_member'])
def delete_leave_message(m):
    if m.left_chat_member.id != bot.get_me().id:
        try:
            # удаляем сообщение о выходе
            bot.delete_message(m.chat.id, m.message_id)
        except Exception as exc:
            logging.info(f"delete_leave_message: {exc}. Not admin?")


@bot.message_handler(content_types=['new_chat_members'])
def delete_join_message(message):
    """Новый пользователь в группе"""
    chat_id = message.chat.id
    greeting = settings.get(str(chat_id)) or default
    user = message.json["new_chat_member"]
    username = f'<a href="tg://user?id={user["id"]}">{user["first_name"]}</a>'
    greeting_text = greeting['greeting_text'].format(username=username)
    # .replace("_", "\\_").replace("*", "\\*").replace("`", "\\`").replace(".", "\\.")
    # .replace("]", "\\]").replace("(", "\\(").replace(")", "\\)").replace("[", "\\[")
    greeting_video = greeting.get('greeting_video')
    greeting_timeout = greeting.get('greeting_timeout') or default['greeting_timeout']

    try:
        # удаляем сообщение о вступлении
        bot.delete_message(chat_id, message.message_id)
    except Exception as exc:
        logging.info(f"delete_join_message: {exc}. Not admin?")

    try:
        if greeting_video:
            msg = bot.send_video(chat_id, greeting_video, caption=greeting_text, parse_mode='HTML')
        else:
            msg = bot.send_message(chat_id, text=greeting_text, parse_mode='HTML')
        t = Timer(greeting_timeout, bot.delete_message, args=[msg.chat.id, msg.message_id])
        t.start()
    except Exception as exc:
        logging.info(f"greeting message: {exc}")


def is_admin(message):
    return bot.get_chat_member(message.chat.id, message.from_user.id).status in ['administrator', 'creator']


def reload_settings():
    global settings
    try:
        with open('settings.json') as file:
            settings = json.loads(file.read())
    except FileNotFoundError:
        settings = dict(admins=[])


def save_settings():
    with open('settings.json', 'w') as file:
        file.write(json.dumps(settings))


def add_chat(message):
    text: str = message.text
    if text.lstrip('-').isdigit():
        settings[text] = default
        msg = bot.reply_to(message, f'Чат {text} добавлен, не забудьте сохранить настройки')
        private_message(message)
    elif text == 'Отмена':
        private_message(message)
    else:
        msg = bot.reply_to(message, 'Неверный id чата')
        proceed_settings(message, chat=True)


def chat_settings(message):
    text = message.text
    if text == 'Отмена':
        private_message(message)
    else:
        key = message.text
        temp[message.chat.id]['key'] = key
        chat_id = temp[message.chat.id]['chat_id']
        key_now = settings[chat_id].get(key)
        if key == 'greeting_text':
            additional = '(Можно использовать {username})'
        else:
            additional = ''
        markup = types.ReplyKeyboardMarkup(one_time_keyboard=True)
        markup.add('Отмена')
        msg = bot.reply_to(message, f'Введите {key} {additional}\nТекущий: {key_now}', reply_markup=markup)
        bot.register_next_step_handler(msg, proceed_chat)


def proceed_chat(message):
    if message.text == 'Отмена':
        private_message(message)
    else:
        key = temp[message.chat.id]['key']
        chat_id = temp[message.chat.id]['chat_id']
        settings[chat_id][key] = message.text
        msg = bot.reply_to(message, f'{key} изменён, не забудьте сохранить настройки')
        msg.text = chat_id
        proceed_settings(msg)


def add_admin(message):
    text: str = message.text
    if text.isdigit():
        settings['admins'].append(int(text))
        msg = bot.reply_to(message, f'Админ {text} добавлен, не забудьте сохранить настройки')
        private_message(message)
    else:
        msg = bot.reply_to(message, 'Неверный id админа')
        proceed_admin(message, force=True)


def del_admin(message):
    text: str = message.text
    if text.startswith('Удалить '):
        admin = int(text[8:])
        settings['admins'].remove(admin)
        msg = bot.reply_to(message, f'Админ {admin} удалён, не забудьте сохранить настройки')
        private_message(message)
    else:
        private_message(message)


def proceed_admin(message, force=False):
    if message.text == 'Добавить админа' or force:
        markup = types.ReplyKeyboardMarkup(one_time_keyboard=True)
        markup.add('Отмена')
        msg = bot.reply_to(message, 'Введите id админа', reply_markup=markup)
        bot.register_next_step_handler(msg, add_admin)
    elif message.text == 'Отмена':
        private_message(message)
    else:
        markup = types.ReplyKeyboardMarkup(one_time_keyboard=True)
        markup.add(f'Удалить {message.text}', 'Отмена')
        msg = bot.reply_to(message, f'Настройка админа {message.text}', reply_markup=markup)
        bot.register_next_step_handler(msg, del_admin)


def proceed_settings(message, chat=False):
    if message.text == 'admins':
        markup = types.ReplyKeyboardMarkup(one_time_keyboard=True)
        admins = [str(admin) for admin in settings.get('admins', [])]
        markup.add(*admins, 'Добавить админа', 'Отмена')
        msg = bot.reply_to(message, 'Админы', reply_markup=markup)
        bot.register_next_step_handler(msg, proceed_admin)
    elif message.text == 'Добавить чат' or chat:
        markup = types.ReplyKeyboardMarkup(one_time_keyboard=True)
        markup.add('Отмена')
        msg = bot.reply_to(message, 'Введите id чата', reply_markup=markup)
        bot.register_next_step_handler(msg, add_chat)
    elif message.text == 'Сохранить настройки':
        save_settings()
        msg = bot.reply_to(message, 'Настройки сохранены')
        private_message(message)
    else:
        chat_id = message.text
        temp[message.chat.id] = dict(chat_id=chat_id)
        markup = types.ReplyKeyboardMarkup(one_time_keyboard=True)
        markup.add(*default.keys(), 'Отмена')
        msg = bot.reply_to(message, f'Настройки чата {chat_id}', reply_markup=markup)
        bot.register_next_step_handler(msg, chat_settings)


def private_message(message):
    admins = settings.get('admins', []) + [main_admin]
    if message.from_user.id in admins:
        # https://ru.stackoverflow.com/questions/1062669/%D0%95%D1%81%D1%82%D1%8C-%D0%BB%D0%B8-%D1%83-pytelegrambotapi-%D0%B0%D0%BD%D0%B0%D0%BB%D0%BE%D0%B3-conversationhandler-%D0%B8%D0%B7-python-telegram-bot
        logging.info(f'private: {message.from_user.id} is admin')
        markup = types.ReplyKeyboardMarkup(one_time_keyboard=True)
        markup.add(*settings.keys(), 'Добавить чат', 'Сохранить настройки')
        msg = bot.send_message(message.chat.id, 'Выберите настройку', reply_markup=markup)
        bot.register_next_step_handler(msg, proceed_settings)
    else:
        logging.info(f'private: {message.from_user.id} not an admin')
        bot.send_message(message.chat.id, 'You are not an admin')


"""
content_types:
text, audio, document, photo, sticker, video, video_note, voice, location, contact, new_chat_members, 
left_chat_member, new_chat_title, new_chat_photo, delete_chat_photo, group_chat_created, supergroup_chat_created, 
channel_chat_created, migrate_to_chat_id, migrate_from_chat_id, pinned_message
"""


@bot.message_handler(func=lambda message: True, content_types=['audio', 'photo', 'voice', 'video', 'document',
                                                               'text', 'location', 'contact', 'sticker'])
def message(message):
    logging.debug(f'{message.chat.id}, {message.content_type}, {message.from_user.id}, {message.from_user.username}, '
                  f'{message.from_user.full_name}')
    if message.chat.id > 0:
        private_message(message)
    elif not is_admin(message) and (
            'url' in restricted and message.text and re.search(url_regex, message.text)
            or 'tag' in restricted and message.text and re.search(tag_regex, message.text)
            or message.content_type in restricted - {'url', 'tag'}
    ):
        bot.delete_message(message.chat.id, message.message_id)


if __name__ == '__main__':
    reload_settings()
    logging.info('Moderator started')
    bot.infinity_polling()