import re
from core.elements import Plain, Image, MessageSession
from core.loader.decorator import command
from modules.wiki.dbutils import WikiTargetInfo
from .userlib import GetUser
@command('user', ['u'], ('~user <username> [-r | -p] {获取一个MediaWiki用户的信息。（-r - 获取详细信息。-p - 生成一张图片。）}'))
async def user(msg: MessageSession):
mode = None
metaurl = None
username = None
if msg.parsed_msg['-r'] is True:
mode = '-r'
if msg.parsed_msg['-p'] is True:
mode = '-p'
get_url = WikiTargetInfo(msg).get_start_wiki()
if get_url:
metaurl = get_url
username = msg.parsed_msg['<username>']
else:
await msg.sendMessage('未设置起始wiki且没有提供Interwiki。')
match_interwiki = re.match(r'(.*?):(.*)', username)
if match_interwiki:
get_iw = WikiTargetInfo(msg).get_interwikis()
if get_iw and match_interwiki.group(1) in get_iw:
metaurl = get_iw
username = match_interwiki.group(2)
result = await GetUser(metaurl, username, mode)
if result:
matchimg = re.match('.*\[\[uimgc:(.*)]]', result)
if matchimg:
imgchain = [Image(path=matchimg.group(1))]
result = re.sub('\[\[uimgc:.*]]', '', result)
msgchain = [Plain(result)]
msgchain = msgchain.append(imgchain)
await msg.sendMessage(msgchain)
