diff --git a/sukuinote/plugins/einfo.py b/sukuinote/plugins/einfo.py index ca74d03..1c74c42 100644 --- a/sukuinote/plugins/einfo.py +++ b/sukuinote/plugins/einfo.py @@ -134,7 +134,7 @@ async def get_deai(client, entity): ntext.pop(0) reason = '\n'.join(ntext).strip() text += html.escape(reason) or 'None' - match = re.match(r'(?:AIdetection:)?((?:0x\d{2} )+)risk:(\S+) mod:X([0-8]) eng:(\S+)(?: cmt:(.+))?', reason) + match = re.match(r'(?:AIdetection:)?((?:0x\d{2} )+)risk:(\S+) mod:X([0-8])(?: eng:(\S+))?(?: cmt:(.+))?', reason) if match: text += '\n- Ban Codes:\n' for i in match.group(1).split(' '): @@ -142,8 +142,10 @@ async def get_deai(client, entity): i = DEAI_BAN_CODES.get(i.strip()[2:], i.strip()) text += f'--- {i}\n' text += f'- Risk Factor: {match.group(2).capitalize()}\n' - text += f'- Module: {DEAI_MODULE_CODES.get(match.group(3), match.group(3))}\n' - text += f'- Engine: {match.group(4).capitalize()}' + text += f'- Module: {DEAI_MODULE_CODES.get(match.group(3), match.group(3))}' + engine = (match.group(4) or '').strip() + if engine: + text += f'\n- Engine: {engine.capitalize()}' comment = (match.group(5) or '').strip() if comment: text += f'\n- Comment: {html.escape(comment)}'