Update config with arguments

This commit is contained in:
riot 2022-09-07 21:00:02 +03:00
parent 7840593370
commit e545e458af
2 changed files with 25 additions and 47 deletions

View file

@ -1,7 +1,6 @@
settings:
rps:
rps_max: 30
rps_min: 1
path_to_proxy: proxy_list.txt
proxy: proxy_list.txt

65
main.py
View file

@ -1,66 +1,45 @@
from loguru import logger
import pretty_errors
from colorama import Fore, Back, Style, init
import pathlib
import argparse
import yaml
import sys
from sys import platform
if platform != "linux":
logger.critical("""\nNot for windows, run only on GNU/Linux!\n""")
input()
sys.exit(1)
import pathlib
from loguru import logger
import pretty_errors
from colorama import Fore, Back, Style, init
import argparse
import yaml
from psycopg2 import OperationalError
parser = argparse.ArgumentParser(
description="List fish in aquarium.",
argument_default=argparse.SUPPRESS
)
# try:
# подключение модулей
# except ModuleNotFoundError:
# logger.critical('''\n[Error 2] Module not found! Сheck the integrity of the program!\n''')
# sys.exit(1)
# try:
# подключение модулей базы данных
# except OperationalError:
# logger.critical('''Type: "sudo systemctl start postgresql"''')
# sys.exit(1)
parser = argparse.ArgumentParser()
parser = argparse.ArgumentParser(description="List fish in aquarium.")
parser.add_argument(
"--config",
"-c",
"--config", "-c",
help="Path to the config file",
type=pathlib.Path,
default="config.yaml",
)
parser.add_argument("--proxy", "-p", help="Path to the proxy file", type=pathlib.Path)
parser.add_argument(
"--proxy", "-p",
help="Path to the proxy file",
type=pathlib.Path
)
parser.add_argument("--rtc_min", help="", type=int)
parser.add_argument("--rtc_max", help="", type=int)
args = parser.parse_args()
args = vars(parser.parse_args())
logger.add("project.log")
logger.info("Starting...")
with open(args.config) as f:
with open(args["config"]) as f:
config = yaml.safe_load(f)
if args.proxy is None:
path_to_proxy = config.path_to_proxy
else:
path_to_proxy = args.proxy
if args.rtc_min is None:
rtc_min = config.rtc_min
else:
rtc_min = args.rtc_min
if args.rtc_max is None:
rtc_max = config.rtc_max
else:
rtc_max = args.rtc_max
config["settings"].update(args)