surok/surok.py

#!/usr/bin/python3

from time import sleep
import os
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

# Read config file
f = open(surok_conf, 'r')
conf = json.loads(f.read())
f.close()


# Get app configurations
# Return list of patches to app discovery configuration
def get_configs():
    confs = [f for f in listdir(conf['confd']) if isfile(
        join(conf['confd'], f))]
    return confs


# Get Surok App configuration
# Read app conf from file and return dict
def load_app_conf(app):
    # Load OS environment to app_conf
    f = open(conf['confd'] + '/' + app)
    c = json.loads(f.read())
    f.close()

    c['env'] = os.environ

    return c


# Main loop
###########

# Bad hack for detect first run
# On host system set it to False
# TODO: put it to config
first = True
while 1:
    confs = get_configs()
    for app in confs:
        app_conf = load_app_conf(app)

        # Will be removed later
        # For old configs
        try:
            loglevel = conf['loglevel']
        except:
            conf['loglevel'] = 'info'

        # Resolve services
        if resolve(app_conf, conf) != 404:
            app_hosts = resolve(app_conf, conf)

            # Populate my dictionary
            my = {"services": app_hosts,
                  "conf_name": app_conf['conf_name']}

            # Generate config from template
            service_conf = gen(my, app_conf['template'])

            first = reload_conf(service_conf, app_conf, first, conf)

    sleep(conf['wait_time'])