58 lines
2.4 KiB
Python
58 lines
2.4 KiB
Python
|
import argparse
|
||
|
|
from types import SimpleNamespace
|
||
|
|
|
||
|
|
hard_config = {
|
||
|
|
"local_ipv4_ranges": ["10.0.0.0/8","172.16.0.0/12","192.168.0.0/16","100.64.0.0/10"],
|
||
|
|
"clore_network_name_prefix":"clore-",
|
||
|
|
"clore_default_networks":[
|
||
|
|
{
|
||
|
|
"name": "clore-br0",
|
||
|
|
"subnet": "172.18.0.0/16",
|
||
|
|
"gateway": "172.18.0.1"
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"run_iptables_with_sudo":True,
|
||
|
|
"clore_iptables_rules":[
|
||
|
|
"-A INPUT -s <subnet> -j DROP",
|
||
|
|
"-I FORWARD -i <interface> -d <subnet> -j DROP"
|
||
|
|
],
|
||
|
|
"clore_br_first_allowed_octet":"172",
|
||
|
|
"ws_peers_recheck_interval": 300,
|
||
|
|
"max_ws_peer_heartbeat_interval":200, # seconds
|
||
|
|
"pull_log_streaming_interval": 450, # ms
|
||
|
|
"log_containers_strings": False, # For debinging only
|
||
|
|
"max_pull_logs_per_submit_run": {
|
||
|
|
"instances":10,
|
||
|
|
"size": 73728 # Characters
|
||
|
|
},
|
||
|
|
"max_remove_images_per_run": 2,
|
||
|
|
"delete_unused_containers": True,
|
||
|
|
"max_pull_log_size": 24576, # Characters
|
||
|
|
"max_container_log_size": 262144, # Characters
|
||
|
|
"container_log_streaming_interval": 2, # Seconds
|
||
|
|
}
|
||
|
|
|
||
|
|
parser = argparse.ArgumentParser(description='Example argparse usage')
|
||
|
|
|
||
|
|
# Add arguments
|
||
|
|
parser.add_argument('--service', action='store_true', help='Flag indicating service')
|
||
|
|
parser.add_argument('--reset', action='store_true', help='Reset init token')
|
||
|
|
parser.add_argument('--debug', action='store_true', help='Show Debug logs')
|
||
|
|
parser.add_argument('--init-token', type=str, help='Init token from clore.ai')
|
||
|
|
parser.add_argument('--auth-file', type=str, default="/opt/clore-hosting/client/auth", help='Auth file')
|
||
|
|
parser.add_argument('--startup-scripts-folder', type=str, default='/opt/clore-hosting/startup_scripts', help='Folder with startup scripts for containers')
|
||
|
|
parser.add_argument('--wireguard-config-folder', type=str, default='/opt/clore-hosting/wireguard/configs', help='Folder with wireguard configs')
|
||
|
|
parser.add_argument('--entrypoints-folder', type=str, default='/opt/clore-hosting/entrypoints', help='Folder with custom entrypoints')
|
||
|
|
parser.add_argument('--debug-ws-peer', type=str, help="Specific ws peer to connect to (for debugging only)")
|
||
|
|
|
||
|
|
# Parse arguments, ignoring any non-defined arguments
|
||
|
|
args, _ = parser.parse_known_args()
|
||
|
|
|
||
|
|
config={}
|
||
|
|
|
||
|
|
for key in vars(args).keys():
|
||
|
|
config[key]=vars(args)[key]
|
||
|
|
for key in hard_config.keys():
|
||
|
|
config[key]=hard_config[key]
|
||
|
|
|
||
|
|
config = SimpleNamespace(**config)
|