button_utils.py 658 Bytes
Newer Older
dwxh's avatar
dwxh committed
1
2
from typing import List

dwxh's avatar
dwxh committed
3
4
5
6
7
from telethon import events

from w_tg_bot import W_BOT_SET


dwxh's avatar
dwxh committed
8
def split_list(list_data: List, n: int) -> List[List]:
dwxh's avatar
dwxh committed
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
    """
    一维列表转二维列表。

    :param list_data: 原始的一维列表
    :param n: 每行的数量
    :return: 对应的二维列表
    """
    return [list_data[n * i:n * (i + 1)]
            for i in range((len(list_data) + n - 1) // n)]


def press_event(user_id=W_BOT_SET["admin_id"]) -> events.CallbackQuery:
    """
    判断是否由指定用户按下。

    :param user_id: 指定用户的 id
    :return:
    """
    return events.CallbackQuery(func=lambda e: e.sender_id == user_id)