72 lines
1.7 KiB
Python
72 lines
1.7 KiB
Python
import re
|
|
|
|
from .orm import WikiWhitelist
|
|
from database.orm import DBSession
|
|
from tenacity import retry, stop_after_attempt
|
|
|
|
session = DBSession().session
|
|
|
|
|
|
@retry(stop=stop_after_attempt(3))
|
|
def audit_query(apiLinkRegex: str):
|
|
try:
|
|
session.expire_all()
|
|
link = session.query(WikiWhitelist).filter_by(apiLinkRegex=apiLinkRegex).first()
|
|
if link is not None:
|
|
return True
|
|
else:
|
|
return False
|
|
except Exception:
|
|
session.rollback()
|
|
raise
|
|
|
|
|
|
@retry(stop=stop_after_attempt(3))
|
|
def audit_allow(apiLinkRegex: str, operator: str):
|
|
if audit_query(apiLinkRegex):
|
|
return False
|
|
try:
|
|
session.add(WikiWhitelist(apiLinkRegex=apiLinkRegex, operator=operator))
|
|
session.commit()
|
|
session.expire_all()
|
|
return True
|
|
except Exception:
|
|
session.rollback()
|
|
raise
|
|
|
|
|
|
@retry(stop=stop_after_attempt(3))
|
|
def audit_remove(apiLinkRegex_: str):
|
|
if not audit_query(apiLinkRegex_):
|
|
return False
|
|
try:
|
|
session.query(WikiWhitelist).filter(WikiWhitelist.apiLinkRegex == apiLinkRegex_).delete()
|
|
session.expire_all()
|
|
return True
|
|
except Exception:
|
|
session.rollback()
|
|
raise
|
|
|
|
|
|
@retry(stop=stop_after_attempt(3))
|
|
def audit_list():
|
|
try:
|
|
return session.query(WikiWhitelist.apiLinkRegex, WikiWhitelist.operator)
|
|
except Exception:
|
|
session.rollback()
|
|
raise
|
|
|
|
|
|
def check_whitelist(apiLink: str):
|
|
whitepair = audit_list()
|
|
whitelist = []
|
|
for pair in whitepair:
|
|
whitelist.append(pair[0])
|
|
for pattern in whitelist:
|
|
if re.match(pattern, apiLink):
|
|
return True
|
|
return False
|
|
|
|
|
|
class WikiWhitelistError(Exception):
|
|
'''The wiki is not in the bot whitelist'''
|