99 lines
3.2 KiB
Python
99 lines
3.2 KiB
Python
# -*- coding:utf-8 -*-
|
||
# @name: config
|
||
# @auth: rainy-autumn@outlook.com
|
||
# @version:
|
||
import os
|
||
import random
|
||
import string
|
||
|
||
import yaml
|
||
|
||
VERSION = "1.3"
|
||
UPDATEURL = "http://update.scope-sentry.top"
|
||
REMOTE_REPO_URL = "https://github.com/Autumn-27/ScopeSentry.git"
|
||
SECRET_KEY = "ScopeSentry-15847412364125411"
|
||
MONGODB_IP = ""
|
||
MONGODB_PORT = 0
|
||
DATABASE_NAME = ""
|
||
DATABASE_USER = ''
|
||
DATABASE_PASSWORD = ''
|
||
REDIS_IP = ""
|
||
REDIS_PORT = ""
|
||
REDIS_PASSWORD = ""
|
||
TIMEZONE = 'Asia/Shanghai'
|
||
LOG_INFO = {}
|
||
GET_LOG_NAME = []
|
||
NODE_TIMEOUT = 50
|
||
TOTAL_LOGS = 1000
|
||
APP = {}
|
||
SensitiveRuleList = {}
|
||
POC_LIST = {}
|
||
Project_List = {}
|
||
def set_timezone(t):
|
||
global TIMEZONE
|
||
TIMEZONE = t
|
||
|
||
|
||
def get_timezone():
|
||
global TIMEZONE
|
||
return TIMEZONE
|
||
|
||
|
||
def generate_random_string(length):
|
||
# 生成随机字符串，包括大小写字母和数字
|
||
characters = string.ascii_letters + string.digits
|
||
random_string = ''.join(random.choice(characters) for _ in range(length))
|
||
return random_string
|
||
|
||
|
||
def set_config():
|
||
global MONGODB_IP, MONGODB_PORT, DATABASE_NAME, DATABASE_USER, DATABASE_PASSWORD, REDIS_IP, REDIS_PORT, REDIS_PASSWORD, SECRET_KEY, TOTAL_LOGS, TIMEZONE
|
||
SECRET_KEY = generate_random_string(16)
|
||
config_file_path = "config.yaml"
|
||
if os.path.exists(config_file_path):
|
||
with open(config_file_path, 'r') as file:
|
||
data = yaml.safe_load(file)
|
||
MONGODB_IP = data['mongodb']['ip']
|
||
MONGODB_PORT = data['mongodb']['port']
|
||
DATABASE_NAME = data['mongodb']['database_name']
|
||
DATABASE_USER = data['mongodb']['username']
|
||
DATABASE_PASSWORD = data['mongodb']['password']
|
||
REDIS_IP = data['redis']['ip']
|
||
REDIS_PORT = data['redis']['port']
|
||
REDIS_PASSWORD = data['redis']['password']
|
||
TOTAL_LOGS = data['logs']['total_logs']
|
||
TIMEZONE = data['system']['timezone']
|
||
else:
|
||
TIMEZONE = os.environ.get("TIMEZONE", default='Asia/Shanghai')
|
||
MONGODB_IP = os.environ.get("MONGODB_IP", default='127.0.0.1')
|
||
MONGODB_PORT = int(os.environ.get("MONGODB_PORT", default=27017))
|
||
DATABASE_NAME = os.environ.get("DATABASE_NAME", default='ScopeSentry')
|
||
DATABASE_USER = os.environ.get("DATABASE_USER", default='root')
|
||
DATABASE_PASSWORD = os.environ.get("DATABASE_PASSWORD", default='QckSdkg5CKvtxfec')
|
||
REDIS_IP = os.environ.get("REDIS_IP", default='127.0.0.1')
|
||
REDIS_PORT = os.environ.get("REDIS_PORT", default="6379")
|
||
REDIS_PASSWORD = os.environ.get("REDIS_PASSWORD", default='ScopeSentry')
|
||
TOTAL_LOGS = 1000
|
||
config_data = {
|
||
'system': {
|
||
'timezone': TIMEZONE
|
||
},
|
||
'mongodb': {
|
||
'ip': MONGODB_IP,
|
||
'port': int(MONGODB_PORT),
|
||
'database_name': DATABASE_NAME,
|
||
'username': DATABASE_USER,
|
||
'password': DATABASE_PASSWORD,
|
||
},
|
||
'redis': {
|
||
'ip': REDIS_IP,
|
||
'port': REDIS_PORT,
|
||
'password': REDIS_PASSWORD,
|
||
},
|
||
'logs': {
|
||
'total_logs': TOTAL_LOGS
|
||
}
|
||
}
|
||
with open(config_file_path, 'w') as file:
|
||
yaml.dump(config_data, file)
|