65 lines
1.3 KiB
Python
Executable File
65 lines
1.3 KiB
Python
Executable File
#!/usr/bin/python3
|
|
|
|
from time import sleep
|
|
from os import listdir
|
|
from os.path import isfile, join
|
|
import json
|
|
from surok.templates import gen
|
|
from surok.discovery import resolve
|
|
from surok.system import reload_conf
|
|
|
|
import argparse
|
|
|
|
# Load base configurations
|
|
surok_conf = '/etc/surok/conf/surok.json'
|
|
|
|
# Command line arguments
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-c', '--config')
|
|
args = parser.parse_args()
|
|
if args.config:
|
|
surok_conf = args.config
|
|
|
|
f = open(surok_conf, 'r')
|
|
conf = json.loads(f.read())
|
|
print(conf)
|
|
f.close()
|
|
|
|
|
|
# Get app configurations
|
|
def get_configs():
|
|
confs = [f for f in listdir(conf['confd']) if isfile( join(conf['confd'], f) )]
|
|
return confs
|
|
|
|
|
|
# Get Surok App configuration
|
|
def load_app_conf(app):
|
|
f = open( conf['confd'] + '/' + app )
|
|
c = json.loads( f.read() )
|
|
f.close()
|
|
|
|
return c
|
|
|
|
|
|
# Main loop
|
|
first = True
|
|
while 1:
|
|
confs = get_configs()
|
|
for app in confs:
|
|
app_conf = load_app_conf(app)
|
|
|
|
# Resolve services
|
|
app_hosts = resolve(app_conf, conf)
|
|
|
|
# Populate my dictionary
|
|
my = { "services": app_hosts }
|
|
|
|
service_conf = gen(my, app_conf['template'])
|
|
|
|
stdout, first = reload_conf(service_conf, app_conf, first)
|
|
print(stdout)
|
|
|
|
|
|
sleep( conf['wait_time'] )
|
|
|