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))