def rebuild_message(message):
data = {
'text': '',
'text_digits': '',
'text_pinyin': '',
'user_id': message['sender']['id'],
'is_at': False
}
if message['type'] == 'FriendMessage':
# todo 仅限管理员使用私聊
if data['user_id'] != config['admin_id']:
return False
data['type'] = 'friend'
data['nickname'] = message['sender']['nickname']
elif message['type'] == 'GroupMessage':
data['type'] = 'group'
data['nickname'] = message['sender']['memberName']
data['group_id'] = message['sender']['group']['id']
data['permission'] = message['sender']['permission']
else:
return False
for chain in message['messageChain']:
if chain['type'] == 'At' and chain['target'] == config['self_id']:
data['is_at'] = True
if chain['type'] == 'Plain':
text = chain['text'].strip()
data['text'] = text
data['text_digits'] = chinese_to_digits(text)
data['text_pinyin'] = text_to_pinyin(text)
if chain['type'] == 'Image':
data['image'] = chain['url'].strip()
return data
def init_stories_titles(self):
stories_titles = database.operator.get_all_stories_title()
self.stories_title = {
chinese_to_digits(item['story_title']): item['story_title']
for item in stories_titles
}
stories_keyword = []
for index, item in self.stories_title.items():
item = re.compile(r'？+', re.S).sub('', item)
if item:
stories_keyword.append(item + ' 100 n')
with open('resource/stories.txt', mode='w', encoding='utf-8') as file:
file.write('\n'.join(stories_keyword))