mirror of
https://github.com/MeoProject/lx-music-api-server.git
synced 2025-05-23 19:17:41 +08:00
692 lines
23 KiB
Python
692 lines
23 KiB
Python
# ----------------------------------------
|
||
# - mode: python -
|
||
# - author: helloplhm-qwq -
|
||
# - name: config.py -
|
||
# - project: lx-music-api-server -
|
||
# - license: MIT -
|
||
# ----------------------------------------
|
||
# This file is part of the "lx-music-api-server" project.
|
||
|
||
import ujson as json
|
||
import time
|
||
import os
|
||
import traceback
|
||
import sys
|
||
import sqlite3
|
||
from . import variable
|
||
from .log import log
|
||
import threading
|
||
|
||
logger = log('config_manager')
|
||
|
||
# 创建线程本地存储对象
|
||
local_data = threading.local()
|
||
|
||
|
||
def get_data_connection():
|
||
# 检查线程本地存储对象是否存在连接对象，如果不存在则创建一个新的连接对象
|
||
if not hasattr(local_data, 'connection'):
|
||
local_data.connection = sqlite3.connect('data.db')
|
||
return local_data.connection
|
||
|
||
|
||
# 创建线程本地存储对象
|
||
local_cache = threading.local()
|
||
|
||
|
||
def get_cache_connection():
|
||
# 检查线程本地存储对象是否存在连接对象，如果不存在则创建一个新的连接对象
|
||
if not hasattr(local_cache, 'connection'):
|
||
local_cache.connection = sqlite3.connect('cache.db')
|
||
return local_cache.connection
|
||
|
||
|
||
class ConfigReadException(Exception):
|
||
pass
|
||
|
||
|
||
default = {
|
||
"common": {
|
||
"host": "0.0.0.0",
|
||
"_host-desc": "服务器启动时所使用的HOST地址",
|
||
"port": "9763",
|
||
"_port-desc": "服务器启动时所使用的端口",
|
||
"debug_mode": False,
|
||
"_debug_mode-desc": "是否开启调试模式",
|
||
"log_length_limit": 500,
|
||
"_log_length_limit-desc": "单条日志长度限制",
|
||
"fakeip": "1.0.1.114",
|
||
"_fakeip-desc": "服务器在海外时的IP伪装值",
|
||
"proxy": {
|
||
"enable": False,
|
||
"http_value": "http://127.0.0.1:7890",
|
||
"https_value": "http://127.0.0.1:7890",
|
||
},
|
||
"_proxy-desc": "代理配置，HTTP与HTTPS协议需分开配置",
|
||
"log_file": True,
|
||
"_log_file-desc": "是否开启日志文件",
|
||
'cookiepool': False,
|
||
'_cookiepool-desc': '是否开启cookie池，这将允许用户配置多个cookie并在请求时随机使用一个，启用后请在module.cookiepool中配置cookie，在user处配置的cookie会被忽略，cookiepool中格式统一为列表嵌套user处的cookie的字典',
|
||
"allow_download_script": True,
|
||
'_allow_download_script-desc': '是否允许直接从服务端下载脚本，开启后可以直接访问 /script?key=你的请求key 下载脚本',
|
||
"download_config": {
|
||
"desc": "源脚本的相关配置，dev为是否启用开发模式",
|
||
"name": "修改为你的源脚本名称",
|
||
"intro": "修改为你的源脚本描述",
|
||
"author": "修改为你的源脚本作者",
|
||
"version": "修改为你的源版本",
|
||
"dev": True,
|
||
"quality": {
|
||
"kw": ["128k", "320k", "flac"],
|
||
"kg": ["128k"],
|
||
"tx": ["128k"],
|
||
"wy": ["128k"],
|
||
"mg": ["128k"],
|
||
}
|
||
}
|
||
},
|
||
"security": {
|
||
"rate_limit": {
|
||
"global": 0,
|
||
"ip": 0,
|
||
"desc": "请求速率限制，global为全局，ip为单个ip，填入的值为至少间隔多久才能进行一次请求，单位：秒，不限制请填为0"
|
||
},
|
||
"key": {
|
||
"enable": False,
|
||
"_enable-desc": "是否开启请求key，开启后只有请求头中包含key，且值一样时可以访问API",
|
||
"ban": True,
|
||
"value": "114514",
|
||
},
|
||
"whitelist_host": [
|
||
"localhost",
|
||
"0.0.0.0",
|
||
"127.0.0.1",
|
||
],
|
||
"_whitelist_host-desc": "强制白名单HOST，不需要加端口号（即不受其他安全设置影响的HOST）",
|
||
"check_lxm": False,
|
||
"_check_lxm-desc": "是否检查lxm请求头（正常的LX Music请求时都会携带这个请求头）",
|
||
"lxm_ban": True,
|
||
"_lxm_ban-desc": "lxm请求头不存在或不匹配时是否将用户IP加入黑名单",
|
||
"allowed_host": {
|
||
"desc": "HOST允许列表，启用后只允许列表内的HOST访问服务器，不需要加端口号",
|
||
"enable": False,
|
||
"blacklist": {
|
||
"desc": "当用户访问的HOST并不在允许列表中时是否将请求IP加入黑名单，长度单位：秒",
|
||
"enable": False,
|
||
"length": 0,
|
||
},
|
||
"list": [
|
||
"localhost",
|
||
"0.0.0.0",
|
||
"127.0.0.1",
|
||
],
|
||
},
|
||
"banlist": {
|
||
"desc": "是否启用黑名单（全局设置，关闭后已存储的值并不受影响，但不会再检查）",
|
||
"enable": True,
|
||
"expire": {
|
||
"desc": "是否启用黑名单IP过期（关闭后其他地方的配置会失效）",
|
||
"enable": True,
|
||
"length": 86400 * 7, # 七天
|
||
},
|
||
},
|
||
},
|
||
"module": {
|
||
"kg": {
|
||
"desc": "酷狗音乐相关配置",
|
||
"client": {
|
||
"desc": "客户端请求配置，不懂请保持默认，修改请统一为字符串格式",
|
||
"appid": "1005",
|
||
"_appid-desc": "酷狗音乐的appid，官方安卓为1005，官方PC为1001",
|
||
"signatureKey": "OIlwieks28dk2k092lksi2UIkp",
|
||
"_signatureKey-desc": "客户端signature采用的key值，需要与appid对应",
|
||
"clientver": "12029",
|
||
"_clientver-desc": "客户端versioncode，pidversionsecret可能随此值而变化",
|
||
"pidversionsecret": "57ae12eb6890223e355ccfcb74edf70d",
|
||
"_pidversionsecret-desc": "获取URL时所用的key值计算验证值",
|
||
},
|
||
"tracker": {
|
||
"desc": "trackerapi请求配置，不懂请保持默认，修改请统一为字符串格式",
|
||
"host": "https://gateway.kugou.com",
|
||
"path": "/v5/url",
|
||
"version": "v5",
|
||
"x-router": {
|
||
"desc": "当host为gateway.kugou.com时需要追加此头，为tracker类地址时则不需要",
|
||
"enable": True,
|
||
"value": "tracker.kugou.com",
|
||
},
|
||
"extra_params": {},
|
||
"_extra_params-desc": "自定义添加的param，优先级大于默认，填写类型为普通的JSON数据，会自动转换为请求param",
|
||
},
|
||
"user": {
|
||
"desc": "此处内容请统一抓包获取，需要vip账号来获取会员歌曲，如果没有请留为空值，mid必填，可以瞎填一段数字",
|
||
"token": "",
|
||
"userid": "0",
|
||
"mid": "114514",
|
||
}
|
||
},
|
||
"tx": {
|
||
"desc": "QQ音乐相关配置",
|
||
"vkeyserver": {
|
||
"desc": "请求官方api时使用的guid，uin等信息，不需要与cookie中信息一致",
|
||
"guid": "114514",
|
||
"uin": "10086",
|
||
},
|
||
"user": {
|
||
"desc": "用户数据，可以通过浏览器获取，需要vip账号来获取会员歌曲，如果没有请留为空值，qqmusic_key可以从Cookie中/客户端的请求体中（comm.authst）获取",
|
||
"qqmusic_key": "",
|
||
"uin": "",
|
||
"_uin-desc": "key对应的QQ号",
|
||
'refresh_login': {
|
||
'desc': '刷新登录相关配置，enable是否启动，interval刷新间隔',
|
||
'enable': False,
|
||
'interval': 86000
|
||
}
|
||
},
|
||
"cdnaddr": "http://ws.stream.qqmusic.qq.com/",
|
||
},
|
||
"wy": {
|
||
"desc": "网易云音乐相关配置",
|
||
"user": {
|
||
"desc": "账号cookie数据，可以通过浏览器获取，需要vip账号来获取会员歌曲，如果没有请留为空值",
|
||
"cookie": ""
|
||
},
|
||
},
|
||
"mg": {
|
||
"desc": "咪咕音乐相关配置",
|
||
"user": {
|
||
"desc": "研究不深，后两项自行抓包获取，在header里",
|
||
"aversionid": "",
|
||
"token": "",
|
||
"osversion": "10",
|
||
"useragent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.82 Safari/537.36",
|
||
},
|
||
},
|
||
'cookiepool': {
|
||
'kg': [
|
||
{
|
||
'userid': '0',
|
||
'token': '',
|
||
'mid': '114514',
|
||
},
|
||
],
|
||
'tx': [
|
||
{
|
||
'qqmusic_key': '',
|
||
'uin': '',
|
||
'refresh_login': {
|
||
'desc': 'cookie池中对于此账号刷新登录的配置，账号间互不干扰',
|
||
'enable': False,
|
||
'interval': 86000,
|
||
}
|
||
}
|
||
],
|
||
'wy': [
|
||
{
|
||
'cookie': '',
|
||
}
|
||
],
|
||
'mg': [
|
||
{
|
||
'aversionid': '',
|
||
'token': '',
|
||
'osversion': '10',
|
||
'useragent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.82 Safari/537.36',
|
||
}
|
||
],
|
||
},
|
||
},
|
||
}
|
||
|
||
|
||
def handle_default_config():
|
||
with open("./config.json", "w", encoding="utf-8") as f:
|
||
f.write(json.dumps(default, indent=2, ensure_ascii=False,
|
||
escape_forward_slashes=False))
|
||
f.close()
|
||
if (not os.getenv('build')):
|
||
logger.info('首次启动或配置文件被删除，已创建默认配置文件')
|
||
logger.info(
|
||
f'\n建议您到{variable.workdir + os.path.sep}config.json修改配置后重新启动服务器')
|
||
return default
|
||
|
||
|
||
class ConfigReadException(Exception):
|
||
pass
|
||
|
||
|
||
def load_data():
|
||
config_data = {}
|
||
try:
|
||
# Connect to the database
|
||
conn = get_data_connection()
|
||
cursor = conn.cursor()
|
||
|
||
# Retrieve all configuration data from the 'config' table
|
||
cursor.execute("SELECT key, value FROM data")
|
||
rows = cursor.fetchall()
|
||
|
||
for row in rows:
|
||
key, value = row
|
||
config_data[key] = json.loads(value)
|
||
|
||
except Exception as e:
|
||
logger.error(f"Error loading config: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
|
||
return config_data
|
||
|
||
|
||
def save_data(config_data):
|
||
try:
|
||
# Connect to the database
|
||
conn = get_data_connection()
|
||
cursor = conn.cursor()
|
||
|
||
# Clear existing data in the 'data' table
|
||
cursor.execute("DELETE FROM data")
|
||
|
||
# Insert the new configuration data into the 'data' table
|
||
for key, value in config_data.items():
|
||
cursor.execute(
|
||
"INSERT INTO data (key, value) VALUES (?, ?)", (key, json.dumps(value)))
|
||
|
||
conn.commit()
|
||
|
||
except Exception as e:
|
||
logger.error(f"Error saving config: {str(e)}")
|
||
logger.error(traceback.format_exc())
|
||
|
||
|
||
def getCache(module, key):
|
||
try:
|
||
# 连接到数据库（如果数据库不存在，则会自动创建）
|
||
conn = get_cache_connection()
|
||
|
||
# 创建一个游标对象
|
||
cursor = conn.cursor()
|
||
|
||
cursor.execute("SELECT data FROM cache WHERE module=? AND key=?",
|
||
(module, key))
|
||
|
||
result = cursor.fetchone()
|
||
if result:
|
||
cache_data = json.loads(result[0])
|
||
if (not cache_data['expire']):
|
||
return cache_data
|
||
if (int(time.time()) < cache_data['time']):
|
||
return cache_data
|
||
except:
|
||
pass
|
||
# traceback.print_exc()
|
||
return False
|
||
|
||
|
||
def updateCache(module, key, data):
|
||
try:
|
||
# 连接到数据库（如果数据库不存在，则会自动创建）
|
||
conn = get_cache_connection()
|
||
|
||
# 创建一个游标对象
|
||
cursor = conn.cursor()
|
||
|
||
cursor.execute(
|
||
"SELECT data FROM cache WHERE module=? AND key=?", (module, key))
|
||
result = cursor.fetchone()
|
||
if result:
|
||
cache_data = json.loads(result[0])
|
||
if isinstance(cache_data, dict):
|
||
cache_data.update(data)
|
||
else:
|
||
logger.error(
|
||
f"Cache data for module '{module}' and key '{key}' is not a dictionary.")
|
||
else:
|
||
cursor.execute(
|
||
"INSERT INTO cache (module, key, data) VALUES (?, ?, ?)", (module, key, json.dumps(data)))
|
||
|
||
conn.commit()
|
||
except:
|
||
logger.error('缓存写入遇到错误…')
|
||
logger.error(traceback.format_exc())
|
||
|
||
|
||
def resetRequestTime(ip):
|
||
config_data = load_data()
|
||
try:
|
||
try:
|
||
config_data['requestTime'][ip] = 0
|
||
except KeyError:
|
||
config_data['requestTime'] = {}
|
||
config_data['requestTime'][ip] = 0
|
||
save_data(config_data)
|
||
except:
|
||
logger.error('配置写入遇到错误…')
|
||
logger.error(traceback.format_exc())
|
||
|
||
|
||
def updateRequestTime(ip):
|
||
try:
|
||
config_data = load_data()
|
||
try:
|
||
config_data['requestTime'][ip] = time.time()
|
||
except KeyError:
|
||
config_data['requestTime'] = {}
|
||
config_data['requestTime'][ip] = time.time()
|
||
save_data(config_data)
|
||
except:
|
||
logger.error('配置写入遇到错误...')
|
||
logger.error(traceback.format_exc())
|
||
|
||
|
||
def getRequestTime(ip):
|
||
config_data = load_data()
|
||
try:
|
||
value = config_data['requestTime'][ip]
|
||
except:
|
||
value = 0
|
||
return value
|
||
|
||
|
||
def read_data(key):
|
||
config = load_data()
|
||
keys = key.split('.')
|
||
value = config
|
||
for k in keys:
|
||
if k not in value and keys.index(k) != len(keys) - 1:
|
||
value[k] = {}
|
||
elif k not in value and keys.index(k) == len(keys) - 1:
|
||
value = None
|
||
value = value[k]
|
||
|
||
return value
|
||
|
||
|
||
def write_data(key, value):
|
||
config = load_data()
|
||
|
||
keys = key.split('.')
|
||
current = config
|
||
for k in keys[:-1]:
|
||
if k not in current:
|
||
current[k] = {}
|
||
current = current[k]
|
||
|
||
current[keys[-1]] = value
|
||
|
||
save_data(config)
|
||
|
||
|
||
def push_to_list(key, obj):
|
||
config = load_data()
|
||
|
||
keys = key.split('.')
|
||
current = config
|
||
for k in keys[:-1]:
|
||
if k not in current:
|
||
current[k] = {}
|
||
current = current[k]
|
||
|
||
if keys[-1] not in current:
|
||
current[keys[-1]] = []
|
||
|
||
current[keys[-1]].append(obj)
|
||
|
||
save_data(config)
|
||
|
||
def write_config(key, value):
|
||
config = None
|
||
with open('config.json', 'r', encoding='utf-8') as f:
|
||
config = json.load(f)
|
||
|
||
keys = key.split('.')
|
||
current = config
|
||
for k in keys[:-1]:
|
||
if k not in current:
|
||
current[k] = {}
|
||
current = current[k]
|
||
|
||
current[keys[-1]] = value
|
||
variable.config = config
|
||
with open('config.json', 'w', encoding='utf-8') as f:
|
||
json.dump(config, f, indent=2, ensure_ascii=False, escape_forward_slashes=False)
|
||
f.close()
|
||
|
||
def read_default_config(key):
|
||
try:
|
||
config = default
|
||
keys = key.split('.')
|
||
value = config
|
||
for k in keys:
|
||
if isinstance(value, dict):
|
||
if k not in value and keys.index(k) != len(keys) - 1:
|
||
value[k] = {}
|
||
elif k not in value and keys.index(k) == len(keys) - 1:
|
||
value = None
|
||
value = value[k]
|
||
else:
|
||
value = None
|
||
break
|
||
|
||
return value
|
||
except:
|
||
return None
|
||
|
||
def _read_config(key):
|
||
try:
|
||
config = variable.config
|
||
keys = key.split('.')
|
||
value = config
|
||
for k in keys:
|
||
if isinstance(value, dict):
|
||
if k not in value and keys.index(k) != len(keys) - 1:
|
||
value[k] = None
|
||
elif k not in value and keys.index(k) == len(keys) - 1:
|
||
value = None
|
||
value = value[k]
|
||
else:
|
||
value = None
|
||
break
|
||
|
||
return value
|
||
except (KeyError, TypeError):
|
||
return None
|
||
|
||
def read_config(key):
|
||
try:
|
||
config = variable.config
|
||
keys = key.split('.')
|
||
value = config
|
||
for k in keys:
|
||
if isinstance(value, dict):
|
||
if k not in value and keys.index(k) != len(keys) - 1:
|
||
value[k] = {}
|
||
elif k not in value and keys.index(k) == len(keys) - 1:
|
||
value = None
|
||
value = value[k]
|
||
else:
|
||
value = None
|
||
break
|
||
|
||
return value
|
||
except:
|
||
default_value = read_default_config(key)
|
||
if (isinstance(default_value, type(None))):
|
||
logger.warning(f'配置文件{key}不存在')
|
||
else:
|
||
for i in range(len(keys)):
|
||
tk = '.'.join(keys[:(i + 1)])
|
||
tkvalue = _read_config(tk)
|
||
logger.debug(f'configfix: 读取配置文件{tk}的值：{tkvalue}')
|
||
if ((tkvalue is None) or (tkvalue == {})):
|
||
write_config(tk, read_default_config(tk))
|
||
logger.info(f'配置文件{tk}不存在，已创建')
|
||
return default_value
|
||
|
||
|
||
def write_data(key, value):
|
||
config = load_data()
|
||
|
||
keys = key.split('.')
|
||
current = config
|
||
for k in keys[:-1]:
|
||
if k not in current:
|
||
current[k] = {}
|
||
current = current[k]
|
||
|
||
current[keys[-1]] = value
|
||
|
||
save_data(config)
|
||
|
||
def initConfig():
|
||
try:
|
||
with open("./config.json", "r", encoding="utf-8") as f:
|
||
try:
|
||
variable.config = json.loads(f.read())
|
||
if (not isinstance(variable.config, dict)):
|
||
logger.warning('配置文件并不是一个有效的字典，使用默认值')
|
||
variable.config = default
|
||
with open("./config.json", "w", encoding="utf-8") as f:
|
||
f.write(json.dumps(variable.config, indent=2, ensure_ascii=False, escape_forward_slashes=False))
|
||
f.close()
|
||
except:
|
||
if os.path.getsize("./config.json") != 0:
|
||
logger.error("配置文件加载失败，请检查是否遵循JSON语法规范")
|
||
sys.exit(1)
|
||
else:
|
||
variable.config = handle_default_config()
|
||
except FileNotFoundError:
|
||
variable.config = handle_default_config()
|
||
# print(variable.config)
|
||
variable.log_length_limit = read_config('common.log_length_limit')
|
||
variable.debug_mode = read_config('common.debug_mode')
|
||
logger.debug("配置文件加载成功")
|
||
conn = sqlite3.connect('cache.db')
|
||
|
||
# 创建一个游标对象
|
||
cursor = conn.cursor()
|
||
|
||
# 创建一个表来存储缓存数据
|
||
cursor.execute('''CREATE TABLE IF NOT EXISTS cache
|
||
(id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
module TEXT NOT NULL,
|
||
key TEXT NOT NULL,
|
||
data TEXT NOT NULL)''')
|
||
|
||
conn.close()
|
||
|
||
conn2 = sqlite3.connect('data.db')
|
||
|
||
# 创建一个游标对象
|
||
cursor2 = conn2.cursor()
|
||
|
||
cursor2.execute('''CREATE TABLE IF NOT EXISTS data
|
||
(key TEXT PRIMARY KEY,
|
||
value TEXT)''')
|
||
|
||
conn2.close()
|
||
|
||
logger.debug('数据库初始化成功')
|
||
|
||
# handle data
|
||
all_data_keys = {'banList': [], 'requestTime': {}, 'banListRaw': []}
|
||
data = load_data()
|
||
if (data == {}):
|
||
write_data('banList', [])
|
||
write_data('requestTime', {})
|
||
logger.info('数据库内容为空，已写入默认值')
|
||
for k, v in all_data_keys.items():
|
||
if (k not in data):
|
||
write_data(k, v)
|
||
logger.info(f'数据库中不存在{k}，已创建')
|
||
|
||
# 处理代理配置
|
||
if (read_config('common.proxy.enable')):
|
||
if (read_config('common.proxy.http_value')):
|
||
os.environ['http_proxy'] = read_config('common.proxy.http_value')
|
||
logger.info('HTTP协议代理地址: ' + read_config('common.proxy.http_value'))
|
||
if (read_config('common.proxy.https_value')):
|
||
os.environ['https_proxy'] = read_config('common.proxy.https_value')
|
||
logger.info('HTTPS协议代理地址: ' + read_config('common.proxy.https_value'))
|
||
logger.info('代理功能已开启，请确保代理地址正确，否则无法连接网络')
|
||
|
||
# cookie池
|
||
if (read_config('common.cookiepool')):
|
||
logger.info('已启用cookie池功能，请确定配置的cookie都能正确获取链接')
|
||
logger.info('传统的源 - 单用户cookie配置将被忽略')
|
||
logger.info('所以即使某个源你只有一个cookie，也请填写到cookiepool对应的源中，否则将无法使用该cookie')
|
||
variable.use_cookie_pool = True
|
||
|
||
# 移除已经过期的封禁数据
|
||
banlist = read_data('banList')
|
||
banlistRaw = read_data('banListRaw')
|
||
count = 0
|
||
for b in banlist:
|
||
if (b['expire'] and (time.time() > b['expire_time'])):
|
||
count += 1
|
||
banlist.remove(b)
|
||
if (b['ip'] in banlistRaw):
|
||
banlistRaw.remove(b['ip'])
|
||
write_data('banList', banlist)
|
||
write_data('banListRaw', banlistRaw)
|
||
if (count != 0):
|
||
logger.info(f'已移除{count}条过期封禁数据')
|
||
|
||
# 处理旧版数据库的banListRaw
|
||
banlist = read_data('banList')
|
||
banlistRaw = read_data('banListRaw')
|
||
if (banlist != [] and banlistRaw == []):
|
||
for b in banlist:
|
||
banlistRaw.append(b['ip'])
|
||
return
|
||
|
||
def ban_ip(ip_addr, ban_time=-1):
|
||
if read_config('security.banlist.enable'):
|
||
banList = read_data('banList')
|
||
banList.append({
|
||
'ip': ip_addr,
|
||
'expire': read_config('security.banlist.expire.enable'),
|
||
'expire_time': read_config('security.banlist.expire.length') if (ban_time == -1) else ban_time,
|
||
})
|
||
write_data('banList', banList)
|
||
banListRaw = read_data('banListRaw')
|
||
if (ip_addr not in banListRaw):
|
||
banListRaw.append(ip_addr)
|
||
write_data('banListRaw', banListRaw)
|
||
else:
|
||
if (variable.banList_suggest < 10):
|
||
variable.banList_suggest += 1
|
||
logger.warning('黑名单功能已被关闭，我们墙裂建议你开启这个功能以防止恶意请求')
|
||
|
||
|
||
def check_ip_banned(ip_addr):
|
||
if read_config('security.banlist.enable'):
|
||
banList = read_data('banList')
|
||
banlistRaw = read_data('banListRaw')
|
||
if (ip_addr in banlistRaw):
|
||
for b in banList:
|
||
if (b['ip'] == ip_addr):
|
||
if (b['expire']):
|
||
if (b['expire_time'] > int(time.time())):
|
||
return True
|
||
else:
|
||
banList.remove(b)
|
||
banlistRaw.remove(b['ip'])
|
||
write_data('banListRaw', banlistRaw)
|
||
write_data('banList', banList)
|
||
return False
|
||
else:
|
||
return True
|
||
else:
|
||
return False
|
||
return False
|
||
else:
|
||
return False
|
||
else:
|
||
if (variable.banList_suggest <= 10):
|
||
variable.banList_suggest += 1
|
||
logger.warning('黑名单功能已被关闭，我们墙裂建议你开启这个功能以防止恶意请求')
|
||
return False
|
||
|
||
|
||
initConfig()
|