2021-08-19 12:17:48 +00:00
|
|
|
|
import re
|
2022-04-04 06:16:39 +00:00
|
|
|
|
from typing import Union
|
|
|
|
|
|
|
|
|
|
|
|
from discord import Embed as DiscordEmbed
|
2021-08-19 12:17:48 +00:00
|
|
|
|
|
2022-06-12 07:07:53 +00:00
|
|
|
|
from core.elements.message.internal import Embed, EmbedField
|
|
|
|
|
|
|
2021-08-19 12:17:48 +00:00
|
|
|
|
|
2022-06-12 07:07:53 +00:00
|
|
|
|
def removeIneffectiveText(prefix: str, lst: list) -> list:
|
2021-10-01 01:18:04 +00:00
|
|
|
|
'''删除命令首尾的空格和换行以及重复命令。
|
|
|
|
|
|
|
|
|
|
|
|
:param prefix: 机器人的命令前缀。
|
|
|
|
|
|
:param lst: 字符串（List/Union）。
|
|
|
|
|
|
:returns: 净化后的字符串。'''
|
2021-08-19 12:17:48 +00:00
|
|
|
|
remove_list = ['\n', ' '] # 首尾需要移除的东西
|
|
|
|
|
|
for x in remove_list:
|
|
|
|
|
|
list_cache = []
|
|
|
|
|
|
for y in lst:
|
|
|
|
|
|
split_list = y.split(x)
|
|
|
|
|
|
for _ in split_list:
|
|
|
|
|
|
if split_list[0] == '':
|
|
|
|
|
|
del split_list[0]
|
|
|
|
|
|
if len(split_list) > 0:
|
|
|
|
|
|
if split_list[-1] == '':
|
|
|
|
|
|
del split_list[-1]
|
|
|
|
|
|
for _ in split_list:
|
|
|
|
|
|
if len(split_list) > 0:
|
2022-01-26 09:52:17 +00:00
|
|
|
|
spl0 = split_list[0][0]
|
|
|
|
|
|
if spl0 in prefix and spl0 != '':
|
2021-08-19 12:17:48 +00:00
|
|
|
|
split_list[0] = re.sub(r'^' + split_list[0][0], '', split_list[0])
|
|
|
|
|
|
list_cache.append(x.join(split_list))
|
|
|
|
|
|
lst = list_cache
|
|
|
|
|
|
duplicated_list = [] # 移除重复命令
|
|
|
|
|
|
for x in lst:
|
|
|
|
|
|
if x not in duplicated_list:
|
|
|
|
|
|
duplicated_list.append(x)
|
|
|
|
|
|
lst = duplicated_list
|
|
|
|
|
|
return lst
|
|
|
|
|
|
|
|
|
|
|
|
|
2022-06-12 07:07:53 +00:00
|
|
|
|
def removeDuplicateSpace(text: str) -> str:
|
2021-10-01 01:18:04 +00:00
|
|
|
|
'''删除命令中间多余的空格。
|
|
|
|
|
|
|
|
|
|
|
|
:param text: 字符串。
|
|
|
|
|
|
:returns: 净化后的字符串。'''
|
2021-08-19 12:17:48 +00:00
|
|
|
|
strip_display_space = text.split(' ')
|
|
|
|
|
|
display_list = [] # 清除指令中间多余的空格
|
|
|
|
|
|
for x in strip_display_space:
|
|
|
|
|
|
if x != '':
|
|
|
|
|
|
display_list.append(x)
|
|
|
|
|
|
text = ' '.join(display_list)
|
2021-08-23 12:44:31 +00:00
|
|
|
|
return text
|
2022-04-04 06:16:39 +00:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def convertDiscordEmbed(embed: Union[DiscordEmbed, dict]) -> Embed:
|
|
|
|
|
|
'''将DiscordEmbed转换为Embed。
|
|
|
|
|
|
:param embed: DiscordEmbed。
|
|
|
|
|
|
:returns: Embed。'''
|
|
|
|
|
|
embed_ = Embed()
|
|
|
|
|
|
if isinstance(embed, DiscordEmbed):
|
|
|
|
|
|
embed = embed.to_dict()
|
|
|
|
|
|
if isinstance(embed, dict):
|
|
|
|
|
|
if 'title' in embed:
|
|
|
|
|
|
embed_.title = embed['title']
|
|
|
|
|
|
if 'description' in embed:
|
|
|
|
|
|
embed_.description = embed['description']
|
|
|
|
|
|
if 'url' in embed:
|
|
|
|
|
|
embed_.url = embed['url']
|
|
|
|
|
|
if 'color' in embed:
|
|
|
|
|
|
embed_.color = embed['color']
|
|
|
|
|
|
if 'timestamp' in embed:
|
|
|
|
|
|
embed_.timestamp = embed['timestamp']
|
|
|
|
|
|
if 'footer' in embed:
|
|
|
|
|
|
embed_.footer = embed['footer']['text']
|
|
|
|
|
|
if 'image' in embed:
|
|
|
|
|
|
embed_.image = embed['image']
|
|
|
|
|
|
if 'thumbnail' in embed:
|
|
|
|
|
|
embed_.thumbnail = embed['thumbnail']
|
|
|
|
|
|
if 'author' in embed:
|
|
|
|
|
|
embed_.author = embed['author']
|
|
|
|
|
|
if 'fields' in embed:
|
|
|
|
|
|
fields = []
|
|
|
|
|
|
for field_value in embed['fields']:
|
|
|
|
|
|
fields.append(EmbedField(field_value['name'], field_value['value'], field_value['inline']))
|
|
|
|
|
|
embed_.fields = fields
|
|
|
|
|
|
return embed_
|
|
|
|
|
|
|
|
|
|
|
|
|
2022-06-12 07:07:53 +00:00
|
|
|
|
__all__ = ['removeDuplicateSpace', 'removeIneffectiveText', 'convertDiscordEmbed']
|
