puypuy.py

import glob
import os
import sys
import time
import threading
from lib.daemon import daemon
import lib.upload_cached
import lib.run_bash
import lib.pushdata
import lib.puylogger
import lib.getconfig
import lib.nag
import gc
import lib.webserver

sys.path.append(os.path.dirname(os.path.realpath("__file__"))+'/checks_enabled')
sys.path.append(os.path.dirname(os.path.realpath("__file__"))+'/lib')

cron_interval = int(lib.getconfig.getparam('SelfConfig', 'check_period_seconds'))
log_file = lib.getconfig.getparam('SelfConfig', 'log_file')
pid_file = lib.getconfig.getparam('SelfConfig', 'pid_file')
tsdb_type = lib.getconfig.getparam('TSDB', 'tsdtype')

tmpdir = lib.getconfig.getparam('SelfConfig', 'tmpdir')
if not os.path.exists(tmpdir):
    os.mkdir(tmpdir)

ws = lib.getconfig.getparam('WebServer', 'webserver').lower()

if ws == 'yes' or tsdb_type == "Prometheus":
    runweb = True
else:
    runweb = False

library_list = []

os.chdir("checks_enabled")

checklist = glob.glob("check_*.py")


module_names = []
for checks in checklist:
    module_names.append(checks.split('.')[0])
modules = list(map(__import__, module_names))


cluster_name = lib.getconfig.getparam('SelfConfig', 'cluster_name')
extra_tags = ('chart_type', 'check_type')
jsondata = lib.pushdata.JonSon()


def run_scripts():
    if len(modules) > 0:
        try:
            start_gtime = time.time()
            jsondata.prepare_data()
            for modol in modules:
                try:
                    # jsondata.prepare_data()
                    start_time = time.time()
                    aa = modol.Check().runcheck()
                    time_elapsed = "{:.9f}".format(time.time() - start_time) + " seconds"
                    message = time_elapsed + ' ' + str(modol).split("'")[1]
                    for b in aa:
                        jsondata.gen_data_json(b, b['host'], cluster_name)
                    lib.puylogger.print_message(message)
                    bb = lib.run_bash.run_shell_scripts()
                    try:
                        for c in bb:
                            jsondata.gen_data_json(c, c['host'], cluster_name)
                    except Exception as ss:
                        lib.puylogger.print_message(str(ss))
                except Exception as e:
                    lib.puylogger.print_message(str(e))
            jsondata.put_json()
            time_elapsed2 = '{:.9f}'.format(time.time() - start_gtime) + ' seconds '
            lib.puylogger.print_message('Spent ' + time_elapsed2 + 'to complete interation')
            return True
        except Exception as e:
            lib.puylogger.print_message(str(e))
    else:
        lib.puylogger.print_message(str('Please enable at least on python module'))
        return False

def upload_cache():
    lib.upload_cached.cache_uploader()


def rn(hast):
    backends = ('OddEye', 'InfluxDB', 'InfluxDB2', 'KairosDB', 'OpenTSDB', 'Prometheus')
    if tsdb_type in backends:
        def run_normal(hast):
            while True:
                run = run_scripts()
                if not run:
                    time.sleep(86400)
                if lib.puylogger.debug_log:
                    lib.puylogger.print_message(str(run_scripts))
                if lib.puylogger.debug_log:
                    lib.puylogger.print_message(str(lib.run_bash.run_shell_scripts()))
                if hast % 25 == 0:
                      gc.collect()
                      hast = 1
                else:
                    hast += 1
                time.sleep(cron_interval)
        def run_cache():
            while True:
                upload_cache()
                if lib.puylogger.debug_log:
                    lib.puylogger.print_message(str(upload_cache))
                time.sleep(cron_interval)

        cache = threading.Thread(target=run_cache, name='Run Cache')
        cache.daemon = True
        cache.start()

        if runweb:
            web = threading.Thread(target=lib.webserver.run_web, name='Run Web')
            web.daemon = True
            web.start()

        run_normal(hast)
    else:
        while True:
            run_scripts()
            lib.run_bash.run_shell_scripts()
            time.sleep(cron_interval)


class App(daemon):
    def run(self):
        rn(1)


def check_pid():
    try:
        with open(pid_file) as f:
            os.kill(int(f.read()), 0)
    except OSError:
        return False
    else:
        return True


def remove_orphan():
    if os.path.exists(pid_file):
        os.remove(pid_file)
        lib.puylogger.print_message('OE Agent is not running but pid file exists')
        lib.puylogger.print_message('Removing orphaned pid file')

if __name__ == "__main__":
    daemon = App(pid_file)
    if len(sys.argv) == 2:
        if 'start' == sys.argv[1]:
            remove_orphan()
            daemon.start()
        elif 'stop' == sys.argv[1]:
            daemon.stop()
        elif 'systemd' == sys.argv[1]:
            remove_orphan()
            rn(1)
        elif 'restart' == sys.argv[1]:
            daemon.restart()
        else:
            print("Unknown command")
            sys.exit(2)
        sys.exit(0)
    else:
            print(("usage: %s start|stop|restart" % sys.argv[0]))
            sys.exit(2)