770 lines
38 KiB
Python
770 lines
38 KiB
Python
import asyncio
|
||
import datetime
|
||
import re
|
||
import traceback
|
||
import urllib.parse
|
||
from typing import Union, Dict, List
|
||
|
||
import ujson as json
|
||
|
||
import core.utils.html2text as html2text
|
||
from config import Config, CFG
|
||
from core.builtins import Url
|
||
from core.dirty_check import check
|
||
from core.logger import Logger
|
||
from core.utils.http import get_url
|
||
from core.utils.i18n import Locale, default_locale
|
||
from core.exceptions import NoReportException
|
||
from modules.wiki.utils.dbutils import WikiSiteInfo as DBSiteInfo, Audit
|
||
|
||
web_render = CFG.get_url('web_render')
|
||
web_render_local = CFG.get_url('web_render_local')
|
||
|
||
redirect_list = {'https://zh.moegirl.org.cn/api.php': 'https://mzh.moegirl.org.cn/api.php', # 萌娘百科强制使用移动版 API
|
||
'https://minecraft.fandom.com/api.php': 'https://minecraft.wiki/api.php', # no more Fandom then
|
||
'https://minecraft.fandom.com/zh/api.php': 'https://zh.minecraft.wiki/api.php'
|
||
}
|
||
|
||
request_by_web_render_list = [re.compile(r'.*minecraft\.wiki'), # sigh
|
||
re.compile(r'.*runescape\.wiki'),
|
||
]
|
||
|
||
|
||
class InvalidPageIDError(Exception):
|
||
pass
|
||
|
||
|
||
class InvalidWikiError(Exception):
|
||
pass
|
||
|
||
|
||
class DangerousContentError(Exception):
|
||
pass
|
||
|
||
|
||
class PageNotFound(Exception):
|
||
pass
|
||
|
||
|
||
class WhatAreUDoingError(Exception):
|
||
pass
|
||
|
||
|
||
class QueryInfo:
|
||
def __init__(self, api, headers=None, prefix=None, locale=None):
|
||
self.api = api
|
||
self.headers = headers if headers is not None else {
|
||
'accept-language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6'}
|
||
self.prefix = prefix
|
||
self.locale = Locale(locale if locale is not None else default_locale)
|
||
|
||
|
||
class WikiInfo:
|
||
def __init__(self,
|
||
api: str = '',
|
||
articlepath: str = '',
|
||
extensions=None,
|
||
interwiki=None,
|
||
realurl: str = '',
|
||
name: str = '',
|
||
namespaces=None,
|
||
namespaces_local=None,
|
||
namespacealiases=None,
|
||
in_allowlist=False,
|
||
in_blocklist=False,
|
||
script: str = '',
|
||
logo_url: str = ''):
|
||
if extensions is None:
|
||
extensions = []
|
||
if interwiki is None:
|
||
interwiki = {}
|
||
self.api = api
|
||
self.articlepath = articlepath
|
||
self.extensions = extensions
|
||
self.interwiki = interwiki
|
||
self.realurl = realurl
|
||
self.name = name
|
||
self.namespaces = namespaces
|
||
self.namespaces_local = namespaces_local
|
||
self.namespacealiases = namespacealiases
|
||
self.in_allowlist = in_allowlist
|
||
self.in_blocklist = in_blocklist
|
||
self.script = script
|
||
self.logo_url = logo_url
|
||
|
||
|
||
class WikiStatus:
|
||
def __init__(self,
|
||
available: bool,
|
||
value: Union[WikiInfo, bool],
|
||
message: str):
|
||
self.available = available
|
||
self.value = value
|
||
self.message = message
|
||
|
||
|
||
class PageInfo:
|
||
def __init__(self,
|
||
info: WikiInfo,
|
||
title: str,
|
||
id: int = -1,
|
||
before_title: str = None,
|
||
link: str = None,
|
||
edit_link: str = None,
|
||
file: str = None,
|
||
desc: str = None,
|
||
args: str = None,
|
||
section: str = None,
|
||
interwiki_prefix: str = '',
|
||
status: bool = True,
|
||
templates: List[str] = None,
|
||
before_page_property: str = 'page',
|
||
page_property: str = 'page',
|
||
has_template_doc: bool = False,
|
||
invalid_namespace: Union[str, bool] = False,
|
||
possible_research_title: List[str] = None
|
||
):
|
||
self.info = info
|
||
self.id = id
|
||
self.title = title
|
||
self.before_title = before_title
|
||
self.link = link
|
||
self.edit_link = edit_link
|
||
self.file = file
|
||
self.desc = desc
|
||
self.args = args
|
||
self.section = section
|
||
self.interwiki_prefix = interwiki_prefix
|
||
self.templates = templates
|
||
self.status = status
|
||
self.before_page_property = before_page_property
|
||
self.page_property = page_property
|
||
self.has_template_doc = has_template_doc
|
||
self.invalid_namespace = invalid_namespace
|
||
self.possible_research_title = possible_research_title
|
||
|
||
|
||
class WikiLib:
|
||
def __init__(self, url: str, headers=None, locale='zh_cn'):
|
||
self.url = url
|
||
self.wiki_info = WikiInfo()
|
||
self.headers = headers
|
||
self.locale = Locale(locale)
|
||
|
||
async def get_json_from_api(self, api, **kwargs) -> dict:
|
||
if api in redirect_list:
|
||
api = redirect_list[api]
|
||
if kwargs is not None:
|
||
api = api + '?' + urllib.parse.urlencode(kwargs) + '&format=json'
|
||
Logger.debug(api)
|
||
else:
|
||
raise ValueError('kwargs is None')
|
||
request_local = False
|
||
for x in request_by_web_render_list:
|
||
if x.match(api):
|
||
if web_render:
|
||
use_local = True if web_render_local else False
|
||
api = (web_render_local if use_local else web_render) + 'source?url=' + urllib.parse.quote(api)
|
||
request_local = True
|
||
break
|
||
try:
|
||
return await get_url(api, status_code=200, headers=self.headers, fmt="json", request_private_ip=request_local)
|
||
except Exception as e:
|
||
if api.find('moegirl.org.cn') != -1:
|
||
raise InvalidWikiError(self.locale.t("wiki.message.utils.wikilib.get_failed.moegirl"))
|
||
raise NoReportException(str(e))
|
||
|
||
def rearrange_siteinfo(self, info: Union[dict, str], wiki_api_link) -> WikiInfo:
|
||
if isinstance(info, str):
|
||
info = json.loads(info)
|
||
extensions = info['query']['extensions']
|
||
ext_list = []
|
||
for ext in extensions:
|
||
ext_list.append(ext['name'])
|
||
real_url = info['query']['general']['server']
|
||
if real_url.startswith('//'):
|
||
real_url = self.url.split('//')[0] + real_url
|
||
namespaces = {}
|
||
namespaces_local = {}
|
||
namespacealiases = {}
|
||
for x in info['query']['namespaces']:
|
||
try:
|
||
ns = info['query']['namespaces'][x]
|
||
if '*' in ns:
|
||
namespaces[ns['*']] = ns['id']
|
||
if 'canonical' in ns:
|
||
namespaces[ns['canonical']] = ns['id']
|
||
if '*' in ns and 'canonical' in ns:
|
||
namespaces_local.update({ns['*']: ns['canonical']})
|
||
except Exception:
|
||
traceback.print_exc()
|
||
for x in info['query']['namespacealiases']:
|
||
if '*' in x:
|
||
namespaces[x['*']] = x['id']
|
||
namespacealiases[x['*'].lower()] = x['*']
|
||
interwiki_map = info['query']['interwikimap']
|
||
interwiki_dict = {}
|
||
for interwiki in interwiki_map:
|
||
interwiki_dict[interwiki['prefix']] = interwiki['url']
|
||
api_url = wiki_api_link
|
||
audit = Audit(api_url)
|
||
return WikiInfo(articlepath=real_url + info['query']['general']['articlepath'],
|
||
extensions=ext_list,
|
||
name=info['query']['general']['sitename'],
|
||
realurl=real_url,
|
||
api=api_url,
|
||
namespaces=namespaces,
|
||
namespaces_local=namespaces_local,
|
||
namespacealiases=namespacealiases,
|
||
interwiki=interwiki_dict,
|
||
in_allowlist=audit.inAllowList,
|
||
in_blocklist=audit.inBlockList,
|
||
script=real_url + info['query']['general']['script'],
|
||
logo_url=info['query']['general'].get('logo'))
|
||
|
||
async def check_wiki_available(self):
|
||
try:
|
||
self.url = re.sub(r'https://zh\.moegirl\.org\.cn/', 'https://mzh.moegirl.org.cn/', self.url)
|
||
# 萌娘百科强制使用移动版 API
|
||
api_match = re.match(r'(https?://.*?/api.php$)', self.url)
|
||
wiki_api_link = api_match.group(1)
|
||
except Exception:
|
||
try:
|
||
get_page = await get_url(self.url, fmt='text', headers=self.headers)
|
||
if get_page.find('<title>Attention Required! | Cloudflare</title>') != -1:
|
||
return WikiStatus(available=False, value=False,
|
||
message=self.locale.t("wiki.message.utils.wikilib.get_failed.cloudflare"))
|
||
m = re.findall(
|
||
r'(?im)<\s*link\s*rel="EditURI"\s*type="application/rsd\+xml"\s*href="([^>]+?)\?action=rsd"\s*/?\s*>',
|
||
get_page)
|
||
api_match = m[0]
|
||
if api_match.startswith('//'):
|
||
api_match = self.url.split('//')[0] + api_match
|
||
# Logger.info(api_match)
|
||
wiki_api_link = api_match
|
||
except (TimeoutError, asyncio.TimeoutError):
|
||
return WikiStatus(available=False, value=False, message=self.locale.t(
|
||
"wiki.message.utils.wikilib.get_failed.timeout"))
|
||
except Exception as e:
|
||
Logger.debug(traceback.format_exc())
|
||
if e.args == (403,):
|
||
message = self.locale.t("wiki.message.utils.wikilib.get_failed.forbidden")
|
||
elif not re.match(r'^(https?://).*', self.url):
|
||
message = self.locale.t("wiki.message.utils.wikilib.get_failed.no_http_or_https_headers")
|
||
else:
|
||
message = self.locale.t("wiki.message.utils.wikilib.get_failed.not_a_mediawiki") + str(e)
|
||
if self.url.find('moegirl.org.cn') != -1:
|
||
message += '\n' + self.locale.t("wiki.message.utils.wikilib.get_failed.moegirl")
|
||
return WikiStatus(available=False, value=False, message=message)
|
||
if wiki_api_link in redirect_list:
|
||
wiki_api_link = redirect_list[wiki_api_link]
|
||
get_cache_info = DBSiteInfo(wiki_api_link).get()
|
||
if get_cache_info and datetime.datetime.now().timestamp() - get_cache_info[1].timestamp() < 43200:
|
||
return WikiStatus(available=True,
|
||
value=self.rearrange_siteinfo(get_cache_info[0], wiki_api_link),
|
||
message='')
|
||
try:
|
||
get_json = await self.get_json_from_api(wiki_api_link,
|
||
action='query',
|
||
meta='siteinfo',
|
||
siprop='general|namespaces|namespacealiases|interwikimap|extensions')
|
||
except Exception as e:
|
||
Logger.debug(traceback.format_exc())
|
||
message = self.locale.t("wiki.message.utils.wikilib.get_failed.api") + str(e)
|
||
if self.url.find('moegirl.org.cn') != -1:
|
||
message += '\n' + self.locale.t("wiki.message.utils.wikilib.get_failed.moegirl")
|
||
return WikiStatus(available=False, value=False, message=message)
|
||
DBSiteInfo(wiki_api_link).update(get_json)
|
||
info = self.rearrange_siteinfo(get_json, wiki_api_link)
|
||
return WikiStatus(available=True, value=info,
|
||
message=self.locale.t("wiki.message.utils.wikilib.no_textextracts")
|
||
if 'TextExtracts' not in info.extensions else '')
|
||
|
||
async def check_wiki_info_from_database_cache(self):
|
||
"""检查wiki信息是否已记录在数据库缓存（由于部分wiki通过path区分语言，此处仅模糊查询域名部分，返回结果可能不准确）"""
|
||
parse_url = urllib.parse.urlparse(self.url)
|
||
get = DBSiteInfo.get_like_this(parse_url.netloc)
|
||
if get is not None:
|
||
api_link = get.apiLink
|
||
if api_link in redirect_list:
|
||
api_link = redirect_list[api_link]
|
||
return WikiStatus(available=True, value=self.rearrange_siteinfo(get.siteInfo, api_link), message='')
|
||
else:
|
||
return WikiStatus(available=False, value=False, message='')
|
||
|
||
async def fixup_wiki_info(self):
|
||
if self.wiki_info.api == '':
|
||
wiki_info = await self.check_wiki_available()
|
||
if wiki_info.available:
|
||
self.wiki_info = wiki_info.value
|
||
else:
|
||
raise InvalidWikiError(wiki_info.message if wiki_info.message != '' else '')
|
||
|
||
async def get_json(self, **kwargs) -> dict:
|
||
await self.fixup_wiki_info()
|
||
api = self.wiki_info.api
|
||
return await self.get_json_from_api(api, **kwargs)
|
||
|
||
@staticmethod
|
||
def parse_text(text):
|
||
try:
|
||
desc = text.split('\n')
|
||
desc_list = []
|
||
for x in desc:
|
||
if x != '':
|
||
desc_list.append(x)
|
||
desc = '\n'.join(desc_list)
|
||
desc_end = re.findall(r'(.*?(?:!\s|\?\s|\.\s|！|？|。)).*', desc, re.S | re.M)
|
||
if desc_end:
|
||
if re.findall(r'[({\[>\"\'《【‘“「（]', desc_end[0]):
|
||
desc_end = re.findall(r'(.*?[)}\]>\"\'》】’”」）].*?(?:!\s|\?\s|\.\s|！|？|。)).*', desc, re.S | re.M)
|
||
desc = desc_end[0]
|
||
except Exception:
|
||
traceback.print_exc()
|
||
desc = ''
|
||
if desc in ['...', '…']:
|
||
desc = ''
|
||
ell = False
|
||
if len(desc) > 250:
|
||
desc = desc[0:250]
|
||
ell = True
|
||
split_desc = desc.split('\n')
|
||
for d in split_desc:
|
||
if d == '':
|
||
split_desc.remove('')
|
||
if len(split_desc) > 5:
|
||
split_desc = split_desc[0:5]
|
||
ell = True
|
||
return '\n'.join(split_desc) + ('...' if ell else '')
|
||
|
||
async def get_html_to_text(self, page_name, section=None):
|
||
await self.fixup_wiki_info()
|
||
get_parse = await self.get_json(action='parse',
|
||
page=page_name,
|
||
prop='text')
|
||
h = html2text.HTML2Text()
|
||
h.ignore_links = True
|
||
h.ignore_images = True
|
||
h.ignore_tables = True
|
||
h.single_line_break = True
|
||
t = h.handle(get_parse['parse']['text']['*'])
|
||
if section is not None:
|
||
for i in range(1, 7):
|
||
s = re.split(r'(.*' + '#' * i + r'[^#].*\[.*?])', t, re.M | re.S)
|
||
ls = len(s)
|
||
if ls > 1:
|
||
ii = 0
|
||
for x in s:
|
||
ii += 1
|
||
if re.match(r'' + '#' * i + '[^#]' + section + r'\[.*?]', x):
|
||
break
|
||
if ii != ls:
|
||
t = ''.join(s[ii:])
|
||
break
|
||
return t
|
||
|
||
async def get_wikitext(self, page_name):
|
||
await self.fixup_wiki_info()
|
||
try:
|
||
load_desc = await self.get_json(action='parse',
|
||
page=page_name,
|
||
prop='wikitext')
|
||
desc = load_desc['parse']['wikitext']['*']
|
||
except Exception:
|
||
traceback.print_exc()
|
||
desc = ''
|
||
return desc
|
||
|
||
async def search_page(self, search_text, namespace='*', limit=10, srwhat='text'):
|
||
await self.fixup_wiki_info()
|
||
title_split = search_text.split(':')
|
||
if title_split[0] in self.wiki_info.interwiki:
|
||
search_text = ':'.join(title_split[1:])
|
||
q_site = WikiLib(self.wiki_info.interwiki[title_split[0]], self.headers)
|
||
result = await q_site.search_page(search_text, namespace, limit)
|
||
result_ = []
|
||
for r in result:
|
||
result_.append(title_split[0] + ':' + r)
|
||
return result_
|
||
get_page = await self.get_json(action='query',
|
||
list='search',
|
||
srsearch=search_text,
|
||
srnamespace=namespace,
|
||
srwhat=srwhat,
|
||
srlimit=limit,
|
||
srenablerewrites=True)
|
||
pagenames = []
|
||
for x in get_page['query']['search']:
|
||
pagenames.append(x['title'])
|
||
return pagenames
|
||
|
||
async def research_page(self, page_name: str, namespace='*', srwhat='text'):
|
||
await self.fixup_wiki_info()
|
||
get_titles = await self.search_page(page_name, namespace=namespace, limit=1, srwhat=srwhat)
|
||
new_page_name = get_titles[0] if len(get_titles) > 0 else None
|
||
title_split = page_name.split(':')
|
||
invalid_namespace = False
|
||
if len(title_split) > 1 and title_split[0] not in self.wiki_info.namespaces \
|
||
and title_split[0].lower() not in self.wiki_info.namespacealiases:
|
||
invalid_namespace = title_split[0]
|
||
return new_page_name, invalid_namespace
|
||
|
||
async def parse_page_info(self, title: str = None, pageid: int = None, inline=False, lang=None, _doc=False,
|
||
_tried=0, _prefix='', _iw=False, _search=False) -> PageInfo:
|
||
"""
|
||
:param title: 页面标题，如果为None，则使用pageid
|
||
:param pageid: 页面id
|
||
:param inline: 是否为inline模式
|
||
:param lang: 所需的对应语言版本
|
||
:param _doc: 是否为文档模式，仅用作内部递归调用判断
|
||
:param _tried: 尝试iw跳转的次数，仅用作内部递归调用判断
|
||
:param _prefix: iw前缀，仅用作内部递归调用判断
|
||
:param _iw: 是否为iw模式，仅用作内部递归调用判断
|
||
:param _search: 是否为搜索模式，仅用作内部递归调用判断
|
||
:return:
|
||
"""
|
||
try:
|
||
await self.fixup_wiki_info()
|
||
except InvalidWikiError as e:
|
||
link = None
|
||
if self.url.find('$1') != -1:
|
||
link = self.url.replace('$1', title)
|
||
return PageInfo(title=title if title is not None else pageid, id=pageid,
|
||
link=link, desc=self.locale.t("error") + str(e), info=self.wiki_info, templates=[])
|
||
ban = False
|
||
if self.wiki_info.in_blocklist and not self.wiki_info.in_allowlist:
|
||
ban = True
|
||
if _tried > 5:
|
||
if Config('enable_tos'):
|
||
raise WhatAreUDoingError
|
||
section = None
|
||
if title is not None:
|
||
if title == '':
|
||
return PageInfo(title='', link=self.wiki_info.articlepath.replace("$1", ""), info=self.wiki_info,
|
||
interwiki_prefix=_prefix, templates=[])
|
||
if inline:
|
||
split_name = re.split(r'(#)', title)
|
||
else:
|
||
split_name = re.split(r'([#?])', title)
|
||
title = re.sub('_', ' ', split_name[0])
|
||
arg_list = []
|
||
_arg_list = []
|
||
section_list = []
|
||
quote_code = False
|
||
for a in split_name[1:]:
|
||
if len(a) > 0:
|
||
if a[0] == '#':
|
||
quote_code = True
|
||
if a[0] == '?':
|
||
quote_code = False
|
||
if quote_code:
|
||
arg_list.append(urllib.parse.quote(a))
|
||
section_list.append(a)
|
||
else:
|
||
_arg_list.append(a)
|
||
_arg = ''.join(_arg_list)
|
||
if _arg.find('=') != -1:
|
||
arg_list.append(_arg)
|
||
else:
|
||
if len(arg_list) > 0:
|
||
arg_list[-1] += _arg
|
||
else:
|
||
title += _arg
|
||
if len(section_list) > 1:
|
||
section = ''.join(section_list)[1:]
|
||
page_info = PageInfo(info=self.wiki_info, title=title, args=''.join(arg_list), interwiki_prefix=_prefix)
|
||
page_info.section = section
|
||
query_string = {'action': 'query', 'prop': 'info|imageinfo|langlinks|templates', 'llprop': 'url',
|
||
'inprop': 'url', 'iiprop': 'url',
|
||
'redirects': 'True', 'titles': title}
|
||
elif pageid is not None:
|
||
page_info = PageInfo(info=self.wiki_info, title=title, args='', interwiki_prefix=_prefix)
|
||
query_string = {'action': 'query', 'prop': 'info|imageinfo|langlinks|templates', 'llprop': 'url',
|
||
'inprop': 'url', 'iiprop': 'url', 'redirects': 'True', 'pageids': pageid}
|
||
else:
|
||
raise ValueError('title and pageid cannot be both None')
|
||
use_textextracts = True if 'TextExtracts' in self.wiki_info.extensions else False
|
||
if use_textextracts and section is None:
|
||
query_string.update({'prop': 'info|imageinfo|langlinks|templates|extracts|pageprops',
|
||
'ppprop': 'description|displaytitle|disambiguation|infoboxes', 'explaintext': 'true',
|
||
'exsectionformat': 'plain', 'exchars': '200'})
|
||
get_page = await self.get_json(**query_string)
|
||
query = get_page.get('query')
|
||
if query is None:
|
||
return PageInfo(title=title, link=None, desc=self.locale.t("wiki.message.utils.wikilib.error.empty"),
|
||
info=self.wiki_info)
|
||
redirects_: List[Dict[str, str]] = query.get('redirects')
|
||
if redirects_ is not None:
|
||
for r in redirects_:
|
||
if r['from'] == title:
|
||
page_info.before_title = r['from']
|
||
page_info.title = r['to']
|
||
normalized_: List[Dict[str, str]] = query.get('normalized')
|
||
if normalized_ is not None:
|
||
for n in normalized_:
|
||
if n['from'] == title:
|
||
page_info.before_title = n['from']
|
||
page_info.title = n['to']
|
||
pages: Dict[str, dict] = query.get('pages')
|
||
# print(pages)
|
||
if pages is not None:
|
||
for page_id in pages:
|
||
page_info.status = False
|
||
page_info.id = int(page_id)
|
||
page_raw = pages[page_id]
|
||
if 'title' in page_raw:
|
||
page_info.title = page_raw['title']
|
||
if 'editurl' in page_raw:
|
||
page_info.edit_link = page_raw['editurl']
|
||
if 'invalid' in page_raw:
|
||
match = re.search(r'"(.)"', page_raw['invalidreason'])
|
||
if match:
|
||
rs = self.locale.t("wiki.message.utils.wikilib.error.invalid_character", char=match.group(1))
|
||
else:
|
||
rs = self.locale.t("wiki.message.utils.wikilib.error.empty_title")
|
||
page_info.desc = rs
|
||
elif 'missing' in page_raw:
|
||
if 'title' in page_raw:
|
||
if 'known' in page_raw:
|
||
full_url = re.sub(r'\$1', urllib.parse.quote(page_info.title.encode('UTF-8')),
|
||
self.wiki_info.articlepath) \
|
||
+ page_info.args
|
||
page_info.link = full_url
|
||
file = None
|
||
if 'imageinfo' in page_raw:
|
||
file = page_raw['imageinfo'][0]['url']
|
||
page_info.file = file
|
||
page_info.status = True
|
||
else:
|
||
split_title = title.split(':')
|
||
reparse = False
|
||
if (len(split_title) > 1 and split_title[0] in self.wiki_info.namespaces_local
|
||
and self.wiki_info.namespaces_local[split_title[0]] == 'Template'):
|
||
rstitle = ':'.join(split_title[1:]) + page_info.args
|
||
reparse = await self.parse_page_info(rstitle)
|
||
page_info.before_page_property = 'template'
|
||
elif len(split_title) > 1 and split_title[
|
||
0].lower() in self.wiki_info.namespacealiases and not _search:
|
||
rstitle = f'{self.wiki_info.namespacealiases[split_title[0].lower()]}:' \
|
||
+ ':'.join(split_title[1:]) + page_info.args
|
||
reparse = await self.parse_page_info(rstitle, _search=True)
|
||
if reparse:
|
||
page_info.before_title = page_info.title
|
||
page_info.title = reparse.title
|
||
page_info.link = reparse.link
|
||
page_info.desc = reparse.desc
|
||
page_info.file = reparse.file
|
||
page_info.status = reparse.status
|
||
page_info.invalid_namespace = reparse.invalid_namespace
|
||
page_info.possible_research_title = reparse.possible_research_title
|
||
else:
|
||
namespace = '*'
|
||
if len(split_title) > 1 and split_title[0] in self.wiki_info.namespaces:
|
||
namespace = self.wiki_info.namespaces[split_title[0]]
|
||
srwhats = ['text', 'title', 'nearmatch']
|
||
preferred = None
|
||
invalid_namespace = False
|
||
|
||
async def search_something(srwhat):
|
||
Logger.debug(traceback.format_exc())
|
||
try:
|
||
research = await self.research_page(page_info.title, namespace, srwhat=srwhat)
|
||
if srwhat == 'text':
|
||
nonlocal preferred
|
||
nonlocal invalid_namespace
|
||
preferred = research[0]
|
||
invalid_namespace = research[1]
|
||
return research
|
||
except Exception:
|
||
Logger.debug(traceback.format_exc())
|
||
return None, False
|
||
|
||
searches = []
|
||
searched_result = []
|
||
for srwhat in srwhats:
|
||
searches.append(search_something(srwhat))
|
||
gather_search = await asyncio.gather(*searches)
|
||
for search in gather_search:
|
||
if search[0] is not None and search[0] not in searched_result:
|
||
searched_result.append(search[0])
|
||
|
||
if preferred is None and searched_result:
|
||
preferred = searched_result[0]
|
||
|
||
page_info.before_title = page_info.title
|
||
page_info.title = preferred
|
||
page_info.invalid_namespace = invalid_namespace
|
||
page_info.possible_research_title = searched_result
|
||
else:
|
||
page_info.status = True
|
||
templates = page_info.templates = [t['title'] for t in page_raw.get('templates', [])]
|
||
if 'special' in page_raw:
|
||
full_url = re.sub(r'\$1',
|
||
urllib.parse.quote(title.encode('UTF-8')),
|
||
self.wiki_info.articlepath) + page_info.args
|
||
page_info.link = full_url
|
||
page_info.status = True
|
||
else:
|
||
query_langlinks = False
|
||
if lang is not None:
|
||
langlinks_ = {}
|
||
for x in page_raw['langlinks']:
|
||
langlinks_[x['lang']] = x['url']
|
||
if lang in langlinks_:
|
||
query_wiki = WikiLib(url=self.wiki_info.interwiki[lang], headers=self.headers)
|
||
await query_wiki.fixup_wiki_info()
|
||
query_wiki_info = query_wiki.wiki_info
|
||
q_articlepath = query_wiki_info.articlepath.replace('$1', '(.*)')
|
||
get_title = re.sub(r'' + q_articlepath, '\\1', langlinks_[lang])
|
||
query_langlinks = await query_wiki.parse_page_info(urllib.parse.unquote(get_title))
|
||
if 'WikibaseClient' in self.wiki_info.extensions and not query_langlinks:
|
||
title = (await self.parse_page_info(title)).title
|
||
qc_string = {'action': 'query', 'meta': 'wikibase', 'wbprop': 'url|siteid'}
|
||
query_client_info = await self.get_json(**qc_string)
|
||
repo_url = query_client_info['query']['wikibase']['repo']['url']['base']
|
||
siteid = query_client_info['query']['wikibase']['siteid']
|
||
query_target_site = WikiLib(self.wiki_info.interwiki[lang], headers=self.headers)
|
||
target_siteid = (await query_target_site.get_json(**qc_string))['query']['wikibase'][
|
||
'siteid']
|
||
qr_wiki_info = WikiLib(repo_url)
|
||
qr_string = {'action': 'wbgetentities', 'sites': siteid, 'titles': title,
|
||
'props': 'sitelinks/urls', 'redirects': 'yes'}
|
||
qr = await qr_wiki_info.get_json(**qr_string)
|
||
if 'entities' in qr:
|
||
qr_result = qr['entities']
|
||
for x in qr_result:
|
||
if 'missing' not in qr_result[x]:
|
||
target_site_page_title = qr_result[x]['sitelinks'][target_siteid]['title']
|
||
q_target = await query_target_site.parse_page_info(target_site_page_title)
|
||
if q_target.status:
|
||
query_langlinks = q_target
|
||
break
|
||
|
||
if lang in self.wiki_info.interwiki and not query_langlinks:
|
||
query_wiki = WikiLib(url=self.wiki_info.interwiki[lang], headers=self.headers)
|
||
await query_wiki.fixup_wiki_info()
|
||
query_wiki_info = query_wiki.wiki_info
|
||
q_articlepath = query_wiki_info.articlepath
|
||
get_title_schema = re.sub(r'' + q_articlepath.replace('$1', '(.*)'), '\\1',
|
||
self.wiki_info.interwiki[lang])
|
||
query_langlinks_ = await query_wiki.parse_page_info(
|
||
get_title_schema.replace('$1', title))
|
||
if query_langlinks_.status:
|
||
query_langlinks = query_langlinks_
|
||
|
||
if not query_langlinks:
|
||
title = page_raw['title']
|
||
page_desc = ''
|
||
split_title = title.split(':')
|
||
get_desc = True
|
||
if not _doc and len(split_title) > 1 and split_title[0] in self.wiki_info.namespaces_local \
|
||
and self.wiki_info.namespaces_local[split_title[0]] == 'Template' \
|
||
and 'Template:Documentation' in templates:
|
||
get_all_text = await self.get_wikitext(title)
|
||
match_doc = re.match(r'.*{{documentation\|?(.*?)}}.*', get_all_text, re.I | re.S)
|
||
if match_doc:
|
||
match_link = re.match(r'link=(.*)', match_doc.group(1), re.I | re.S)
|
||
if match_link:
|
||
get_doc = match_link.group(1)
|
||
else:
|
||
get_doc = title + '/doc'
|
||
get_desc = False
|
||
get_doc_desc = await self.parse_page_info(get_doc, _doc=True)
|
||
page_desc = get_doc_desc.desc
|
||
if page_desc is not None:
|
||
page_info.has_template_doc = True
|
||
page_info.before_page_property = page_info.page_property = 'template'
|
||
if get_desc:
|
||
if use_textextracts and section is None:
|
||
raw_desc = page_raw.get('extract')
|
||
if raw_desc is not None:
|
||
page_desc = self.parse_text(raw_desc)
|
||
else:
|
||
page_desc = self.parse_text(await self.get_html_to_text(title, section))
|
||
full_url = page_raw['fullurl'] + page_info.args
|
||
file = None
|
||
if 'imageinfo' in page_raw:
|
||
file = page_raw['imageinfo'][0]['url']
|
||
page_info.title = title
|
||
page_info.link = full_url
|
||
page_info.file = file
|
||
page_info.desc = page_desc
|
||
if not _iw and page_info.args == '':
|
||
page_info.link = self.wiki_info.script + f'?curid={page_info.id}'
|
||
else:
|
||
page_info.title = query_langlinks.title
|
||
page_info.before_title = query_langlinks.title
|
||
page_info.link = query_langlinks.link
|
||
page_info.edit_link = query_langlinks.edit_link
|
||
page_info.file = query_langlinks.file
|
||
page_info.desc = query_langlinks.desc
|
||
interwiki_: List[Dict[str, str]] = query.get('interwiki')
|
||
if interwiki_ is not None:
|
||
for i in interwiki_:
|
||
if i['title'] == page_info.title:
|
||
iw_title = re.match(r'^' + i['iw'] + ':(.*)', i['title'])
|
||
iw_title = iw_title.group(1)
|
||
_prefix += i['iw'] + ':'
|
||
_iw = True
|
||
iw_query = await WikiLib(url=self.wiki_info.interwiki[i['iw']], headers=self.headers) \
|
||
.parse_page_info(iw_title, lang=lang,
|
||
_tried=_tried + 1,
|
||
_prefix=_prefix,
|
||
_iw=_iw)
|
||
before_page_info = page_info
|
||
page_info = iw_query
|
||
if iw_query.title == '':
|
||
page_info.title = ''
|
||
else:
|
||
page_info.before_title = before_page_info.title
|
||
t = page_info.title
|
||
if t != '' and t is not None:
|
||
if before_page_info.args is not None:
|
||
page_info.before_title += urllib.parse.unquote(before_page_info.args)
|
||
t += urllib.parse.unquote(before_page_info.args)
|
||
if page_info.link is not None:
|
||
page_info.link += before_page_info.args
|
||
else:
|
||
page_info.link = self.wiki_info.script + f'?curid={page_info.id}'
|
||
if _tried == 0:
|
||
if lang is not None and page_info.status:
|
||
page_info.before_title = page_info.title
|
||
else:
|
||
page_info.title = page_info.interwiki_prefix + t
|
||
if page_info.possible_research_title is not None:
|
||
page_info.possible_research_title = [page_info.interwiki_prefix + possible_title
|
||
for possible_title in
|
||
page_info.possible_research_title]
|
||
|
||
if before_page_info.section is not None:
|
||
page_info.section = before_page_info.section
|
||
if not self.wiki_info.in_allowlist:
|
||
checklist = []
|
||
if page_info.title is not None:
|
||
checklist.append(page_info.title)
|
||
if page_info.before_title is not None:
|
||
checklist.append(page_info.before_title)
|
||
if page_info.desc is not None:
|
||
checklist.append(page_info.desc)
|
||
chk = await check(*checklist)
|
||
for x in chk:
|
||
if not x['status']:
|
||
ban = True
|
||
if ban:
|
||
page_info.status = False
|
||
page_info.title = page_info.before_title = None
|
||
page_info.id = -1
|
||
if page_info.link is not None:
|
||
page_info.desc = str(Url(page_info.link, use_mm=True))
|
||
page_info.link = None
|
||
return page_info
|
||
|
||
async def random_page(self) -> PageInfo:
|
||
"""
|
||
获取随机页面
|
||
:return: 页面信息
|
||
"""
|
||
await self.fixup_wiki_info()
|
||
random_url = await self.get_json(action='query', list='random', rnnamespace='0')
|
||
page_title = random_url['query']['random'][0]['title']
|
||
return await self.parse_page_info(page_title)
|