refactor: 删除检测

This commit is contained in:
KAAAsS 2024-07-01 00:56:16 +08:00
parent c32cdf05fb
commit 0f137076a7
Signed by: KAAAsS
GPG Key ID: D22F53AF662411FE
2 changed files with 1 additions and 46 deletions

View File

@ -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

View File

@ -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