message.py

#!coding:utf-8
#  Created by bluebird on 2018/12/3
"""
handle massage
"""
import filter
from admin import user_is_ban
from constant import WARN_MSG, SET_OK_MSG, LIMIT_DICT, BanMessageType, BAN_STATE, OpenState, OPITON_ERROR
from telegram import Update, Bot
from telegram.ext.dispatcher import run_async
from tool import messaage_warp, check_admin, kick_user, check_run, check_ban_state
from telegram.ext.filters import Filters


@messaage_warp(filters=filter.Run() & Filters.group & (~filter.Admin()) & (~filter.GroupAdmin()) & (
        filter.TelegramLink() | filter.Lang() | filter.Flood() |
        filter.Emoji() | filter.Gif() | filter.Numbers()
        | filter.BanWord() | filter.Lock() | filter.AutoLock() | filter.MaxMsg()),
               pass_chat_data=True,
               pass_user_data=True)
@run_async
def telegram_link_handler(bot, update, user_data, chat_data):
    """
    :param chat_data:
    :param user_data:
    :param bot:
    :type bot: Bot
    :param update:
    :type update: Update
    :return:
    """
    update.message.delete()
    warn_user(bot, update, user_data, chat_data)


@messaage_warp(filters=(filter.Run() & Filters.group & Filters.all & ~filter.Admin() & ~filter.GroupAdmin()),
               pass_chat_data=True, pass_user_data=True)
@run_async
def common_message_handler(bot, update, user_data, chat_data):
    """
    :param user_data:
    :param chat_data:
    :param bot:
    :param update:
    :type bot: Bot
    :type update: Update
    :type chat_data:dict
    :type user_data:dict
    :return:
    """
    ban_state = chat_data.get(BAN_STATE, {})
    for ban_type in ban_state.keys():
        if getattr(update.message, ban_type, False):
            update.message.delete()
            warn_user(bot, update, user_data, chat_data)
            return
    if update.message.entities:
        for entity in update.message.entities:
            if entity.type in ban_state.keys():
                update.message.delete()
                warn_user(bot, update, user_data, chat_data)
            return
    if ban_state.get('all'):
        update.message.delete()
        warn_user(bot, update, user_data, chat_data)
        return
    return


@check_admin(admin=True)
@check_run()
@run_async
def limit_set(bot, update, chat_data, groups):
    if not chat_data.get('ban_state'):
        chat_data['ban_state'] = {}

    if not groups[1].isalpha():
        groups[1] = groups[1][1:]
    if LIMIT_DICT.get(groups[0]):
        limits = LIMIT_DICT.get(groups[0])
        for limit in limits:
            if groups[1] == OpenState.OPEN:
                if chat_data['ban_state'].get(limit):
                    chat_data['ban_state'].pop(limit)
            elif groups[1] == OpenState.CLODE:
                chat_data['ban_state'][limit] = groups[1]
            else:
                bot.send_message(update.message.chat_id, text=OPITON_ERROR)
                return
    else:
        if groups[1] == OpenState.OPEN:
            if chat_data['ban_state'].get(groups[0]):
                chat_data['ban_state'].pop(groups[0])
        elif groups[1] == OpenState.CLODE:
            chat_data['ban_state'][groups[0]] = True
        else:
            bot.send_message(update.message.chat_id, text=OPITON_ERROR)
            return
    bot.send_message(update.message.chat_id, text=SET_OK_MSG)
    return


@messaage_warp(filters=filter.NewMember())
@run_async
def new_member(bot, update):
    if check_ban_state(update.message.chat_id, BanMessageType.ADDGROUP):
        update.message.delete()
    for members in update.message.new_chat_members:
        if user_is_ban(members.id):
            kick_user(bot=bot, update=update, user_list=[members.id])
    return


def warn_user(bot, update, user_data, chat_data):
    """

    :param chat_data:
    :param bot:
    :param update:
    :param user_data:
    :type user_data: dict
    :return:
    """
    if not chat_data.get(BAN_STATE, {}).get(BanMessageType.WARN):
        bot.send_message(chat_id=update.message.chat_id, text=WARN_MSG)
        user_data['warn'] = user_data.get('warn', 0) + 1
        if chat_data.get('maxwarn'):
            if user_data['warn'] > chat_data.get('maxwarn'):
                kick_user(bot, update, [update.message.from_user.id])