xtex/akari-bot
Archived
1
0
Fork 0
Code Issues Pull requests Activity

Update dbutils.py

Browse source
This commit is contained in:
yzhh 2021-10-28 23:44:45 +08:00
parent 952072a8f9
commit 82d0021f0f
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
modules/wiki/dbutils.py
View file

@ -1,3 +1,5 @@
from typing import Union
import ujson as json
from core.elements import MessageSession
@ -37,10 +39,9 @@ class WikiTargetInfo:
session.rollback()
raise
def get_start_wiki(self):
def get_start_wiki(self) -> Union[str, None]:
if self.query is not None:
return self.query.link if self.query.link is not None else False
return False
return self.query.link if self.query.link is not None else None
@retry(stop=stop_after_attempt(3))
def config_interwikis(self, iw: str, iwlink: str = None, let_it=True):
@ -59,13 +60,13 @@ class WikiTargetInfo:
session.rollback()
raise
def get_interwikis(self):
def get_interwikis(self) -> dict:
q = self.query.iws
if q is not None:
iws = json.loads(q)
return iws
else:
return False
return {}
@retry(stop=stop_after_attempt(3))
def config_headers(self, headers, let_it: [bool, None] = True):