|\s[^<>]*>)\s*)+", "\n", html_string)
75 |
76 | return html_string.strip(" \t")
77 |
78 |
79 | def replace_line_breaks_except_pre(html_string, replace_by=" "):
80 | # Remove all line breaks and empty strings, except pre tag
81 | # how to make it in one string? :\
82 | pre_ranges = [0]
83 | out = ""
84 |
85 | # replace non-breaking space with usual space
86 | html_string = html_string.replace("\u00a0", " ")
87 |
88 | # get start/end postion
89 | for x in pre_content_re.finditer(html_string):
90 | start, end = x.start(), x.end()
91 | pre_ranges.extend((start, end))
92 | pre_ranges.append(len(html_string))
93 |
94 | # all odd elements are , leave them untouched
95 | for k in range(1, len(pre_ranges)):
96 | part = html_string[pre_ranges[k - 1] : pre_ranges[k]]
97 | if k % 2 == 0:
98 | out += line_breaks_inside_pre.sub("\n", part)
99 | else:
100 | out += line_breaks_and_empty_strings.sub(replace_by, part)
101 | return out
102 |
--------------------------------------------------------------------------------
/utiles/filters.py:
--------------------------------------------------------------------------------
1 | from pyrogram import filters
2 | from pyrogram.types import Message, InlineQuery
3 |
4 | from parsehub import ParseHub
5 |
6 |
7 | async def _platform_filter(_, __, update: Message | InlineQuery):
8 | if isinstance(update, Message):
9 | t = update.caption or update.text
10 | else:
11 | t = update.query
12 | return bool(ParseHub()._select_parser(t))
13 |
14 |
15 | platform_filter = filters.create(_platform_filter)
16 |
17 |
18 | def filter_inline_query_results(command: str):
19 | """
20 | 过滤指定字符开头的内联查询结果
21 |
22 | :param command:
23 | :return:
24 | """
25 |
26 | async def func(_, __, update):
27 | return update.query.startswith(command)
28 |
29 | return filters.create(func, commands=command)
30 |
--------------------------------------------------------------------------------
/utiles/img_host.py:
--------------------------------------------------------------------------------
1 | import httpx
2 | from httpx._types import ProxiesTypes
3 | from tenacity import retry, stop_after_attempt
4 |
5 |
6 | class ImgHost:
7 | def __init__(self, proxies: ProxiesTypes = None):
8 | self.async_client = httpx.AsyncClient(proxies=proxies)
9 |
10 | @retry(stop=stop_after_attempt(5))
11 | async def litterbox(self, filename: str):
12 | host_url = "https://litterbox.catbox.moe/resources/internals/api.php"
13 |
14 | file = open(filename, "rb")
15 | try:
16 | data = {
17 | "reqtype": "fileupload",
18 | "time": "1h",
19 | }
20 | response = await self.async_client.post(
21 | host_url, data=data, files={"fileToUpload": file}
22 | )
23 | finally:
24 | file.close()
25 |
26 | return response.text
27 |
28 | def __aexit__(self, exc_type, exc_val, exc_tb):
29 | self.async_client.aclose()
30 |
--------------------------------------------------------------------------------
/utiles/ph.py:
--------------------------------------------------------------------------------
1 | from dataclasses import dataclass
2 |
3 | from telegraph.aio import Telegraph as TelegraphAPI
4 |
5 |
6 | class Telegraph:
7 | """Telegraph API 封装"""
8 |
9 | def __init__(self, token: str = None, domain: str = "telegra.ph"):
10 | self.token = token
11 | self.domain = domain
12 | self.telegraph = TelegraphAPI(access_token=token, domain=domain)
13 |
14 | async def create_account(
15 | self, short_name: str, author_name: str = None, author_url: str = None
16 | ) -> "TelegraphAccount":
17 | """创建 Telegraph 账户"""
18 | account = await self.telegraph.create_account(
19 | short_name, author_name, author_url
20 | )
21 | acc_info = await self.get_account_info(account)
22 | self.token = acc_info.access_token
23 | return acc_info
24 |
25 | async def get_account_info(self, account_info: dict = None) -> "TelegraphAccount":
26 | """获取 Telegraph 账户信息"""
27 | account_info = account_info or await self.telegraph.get_account_info(
28 | [
29 | "short_name",
30 | "author_name",
31 | "author_url",
32 | "auth_url",
33 | ]
34 | )
35 | return TelegraphAccount(
36 | self.telegraph.get_access_token(),
37 | account_info["short_name"],
38 | account_info["author_name"],
39 | account_info["author_url"],
40 | account_info["auth_url"],
41 | )
42 |
43 | async def create_page(
44 | self,
45 | title,
46 | content=None,
47 | html_content=None,
48 | author_name=None,
49 | author_url=None,
50 | return_content=False,
51 | auto_create_account=True,
52 | ) -> "TelegraphPage":
53 | """创建 Telegraph 页面"""
54 | if auto_create_account and not self.token:
55 | # 随机用户名
56 | short_name = "tg_" + str(
57 | int(100000 * (1 + 0.5 * (1 + 0.5 * (1 + 0.5 * 1))))
58 | )
59 | await self.create_account(short_name)
60 | response = await self.telegraph.create_page(
61 | title,
62 | content,
63 | html_content,
64 | author_name,
65 | author_url,
66 | return_content,
67 | )
68 | return TelegraphPage(
69 | response["path"],
70 | response["url"],
71 | response["title"],
72 | response["description"],
73 | response["views"],
74 | response["can_edit"],
75 | await self.get_account_info(),
76 | )
77 |
78 |
79 | @dataclass
80 | class TelegraphAccount:
81 | access_token: str
82 | short_name: str
83 | author_name: str
84 | author_url: str
85 | auth_url: str
86 |
87 |
88 | @dataclass
89 | class TelegraphPage:
90 | path: str
91 | url: str
92 | title: str
93 | description: str
94 | views: int
95 | can_edit: bool
96 | account: TelegraphAccount
97 |
--------------------------------------------------------------------------------
/utiles/utile.py:
--------------------------------------------------------------------------------
1 | import asyncio
2 | import hashlib
3 |
4 | from pyrogram import Client
5 |
6 |
7 | async def schedule_delete_messages(
8 | client: Client, chat_id: int, message_ids: int | list, delay_seconds: int = 2
9 | ):
10 | """定时删除消息"""
11 |
12 | await asyncio.sleep(delay_seconds)
13 |
14 | try:
15 | await client.delete_messages(chat_id, message_ids)
16 | except Exception:
17 | ...
18 |
19 |
20 | def progress(current, total, status):
21 | if total == 0:
22 | return status
23 |
24 | text = None
25 | if total >= 100:
26 | if round(current * 100 / total, 1) % 25 == 0:
27 | text = f"下 载 中...|{status}"
28 | else:
29 | if (current + 1) % 3 == 0 or (current + 1) == total:
30 | text = f"下 载 中...|{status}"
31 | return text
32 |
33 |
34 | def encrypt(text: str):
35 | """hash加密"""
36 | md5 = hashlib.md5()
37 | md5.update(text.encode("utf-8"))
38 | return md5.hexdigest()
39 |
--------------------------------------------------------------------------------