1
0
Fork 0
mirror of https://github.com/anyproto/any-sync-dockercompose.git synced 2025-06-08 05:47:03 +09:00
any-sync-dockercompose/docker-generateconfig/setListenIp.py

23 lines
585 B
Python
Executable file

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import sys
import yaml
arguments = sys.argv[1:]
yamlFile = arguments[0]
listenHosts = arguments[1:]
with open(yamlFile, 'r') as file:
config = yaml.load(file,Loader=yaml.Loader)
for index, nodes in enumerate(config['nodes']):
addresses = nodes['addresses']
port = addresses[0].split(':')[1]
for listenHost in listenHosts:
listenAddress = listenHost +':'+ port
if listenAddress not in addresses:
addresses.append(listenAddress)
with open(yamlFile, 'w') as file:
yaml.dump(config, file)