anselm.py

import sys
import json
import argparse
import pika
from anselm.system import System

class Anselm(System):
    """
    https://chase-seibert.github.io/blog/2014/03/21/python-multilevel-argparse.html
    """
    def __init__(self):
        super().__init__()

        msg_dict = self.config['rabbitmq']
        host = msg_dict['host']
        self.msg_param = pika.ConnectionParameters(host=host)

        self.init_ltm_msg_prod()
        self.init_stm_msg_prod()

        parser = argparse.ArgumentParser(
            description='check systems',
            usage='''anselm <command> [<args>]''')

        parser.add_argument('command', help='Subcommand to run')
        args = parser.parse_args(sys.argv[1:2])

        if not hasattr(self, args.command):
            parser.print_help()
            exit(1)

        if len(args.command) > self.max_arg_len:
            print("command too long")
            exit(1)

        getattr(self, args.command)()

    def start(self):
        """
        """
        parser = argparse.ArgumentParser(
            description="sends a all to ltm exchange")

        self.ltm_pub(body_dict={
                    'do':'start',
                    'payload':{}
                    })
        self.ltm_conn.close()

    def clear_stm(self):
        """
        """
        parser = argparse.ArgumentParser(
            description="sends a clear.all to stm exchange")
        self.stm_pub(body_dict={
                    'do':'clear_all',
                    'payload':{}})
        self.ltm_conn.close()

    def mp_to_ltm(self):
        """
        """
        parser = argparse.ArgumentParser(
            description="sends a mp from stm back to ltm")
        parser.add_argument('mpid')
        arg = parser.parse_args(sys.argv[2:3])

        self.stm_pub(body_dict={
                    'do':'mp_to_ltm',
                    'payload':{'id': arg.mpid}})
        self.ltm_conn.close()

    def build_api_for(self):
        parser = argparse.ArgumentParser(
            description="checks if the systems are up")

        parser.add_argument('mpid')
        arg = parser.parse_args(sys.argv[2:3])

        if len(arg.mpid) < self.max_arg_len:
            self.stm_pub(body_dict={
                        'do':'build_api',
                        'payload':{"id": arg.mpid}
                        })

        self.stm_conn.close()

    def read_exchange(self):
        parser = argparse.ArgumentParser(description="read from exchange")

        self.stm_publish(body_dict={
                        'do':'read_exchange',
                        'payload':{"id":"mpd-ce3-calib", "find_set":{"StartTime.Type":"start"}}
                        })

        self.stm_conn.close()

if __name__ == '__main__':
    Anselm()