def load_envs():
logger.info("Loading environment")
load_dotenv()
envs_correct = True
global protocol, endpoint, mode, record_name, records_path, socket_protocol, socket_rop, ignore_log, save_single
if (protocol := os.getenv("PROTOCOL")) is None:
logger.error("PROTOCOL variable not found. Check your env file")
envs_correct = False
def prepare(self):
try:
shutil.rmtree(utils.record_path)
except OSError as e:
logger.info(f"Nothing to delete at {utils.record_path}")
else:
logger.info(f"Removed existing record folder at {utils.record_path}")
try:
os.mkdir(utils.record_path)
if settings.save_single is not None:
os.mkdir(utils.singles_path)
if settings.socket_rop is not None:
os.mkdir(utils.sockets_path)
except OSError as e:
logger.error(f"Error creating record folder at {utils.record_path}: {str(e)}")
else:
logger.info(f"Created new record folder at {utils.record_path}")
ignore_log = None
save_single = None
def load_envs():
logger.info("Loading environment")
load_dotenv()
envs_correct = True
global protocol, endpoint, mode, record_name, records_path, socket_protocol, socket_rop, ignore_log, save_single
if (protocol := os.getenv("PROTOCOL")) is None:
logger.error("PROTOCOL variable not found. Check your env file")
envs_correct = False
if (endpoint := os.getenv("ENDPOINT")) is None:
logger.error("ENDPOINT variable not found. Check your env file")
envs_correct = False
if (records_path := os.getenv("RECORDS_PATH")) is None:
logger.error("RECORDS_PATH variable not found. Check your env file")
envs_correct = False
try:
mode = os.getenv("MODE")
except:
mode = "PROXY"
try:
record_name = os.getenv("RECORDING")
except:
mode = "PROXY"
def out_on_error(ws, error):
logger.error(f"Got error on OUT socket {error}")