__main__.py

#!/usr/bin/env python
##
## This file is part of the OpenSIPS Python Package
## (see https://github.com/OpenSIPS/python-opensips).
##
## This program is free software: you can redistribute it and/or modify
## it under the terms of the GNU General Public License as published by
## the Free Software Foundation, either version 3 of the License, or
## (at your option) any later version.
##
## This program is distributed in the hope that it will be useful,
## but WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with this program. If not, see <http://www.gnu.org/licenses/>.
##

""" Script to run OpenSIPS MI commands """

import sys
import json
import argparse
from opensips.mi import OpenSIPSMI, OpenSIPSMIException

parser = argparse.ArgumentParser()

communication = parser.add_argument_group('communication')

communication.add_argument('-t', '--type',
                    type=str,
                    default='fifo',
                    choices=['fifo', 'http', 'datagram'],
                    help='OpenSIPS MI Communication Type')
communication.add_argument('-i', '--ip',
                    type=str,
                    help='OpenSIPS MI IP Address',
                    default='127.0.0.1')
communication.add_argument('-p', '--port',
                    type=int,
                    help='OpenSIPS MI Port',
                    default=8888)
communication.add_argument('-f', '--fifo-file',
                    type=str,
                    help='OpenSIPS MI FIFO File')
communication.add_argument('-fb', '--fifo-fallback',
                    type=str,
                    help='OpenSIPS MI Fallback FIFO File')
communication.add_argument('-fd', '--fifo-reply-dir',
                    type=str,
                    help='OpenSIPS MI FIFO Reply Directory')

group = parser.add_mutually_exclusive_group(required=True)

group.add_argument('-s', '--stats',
                    nargs='+',
                    default=[],
                    help='statistics')

group.add_argument('command',
                    nargs='?',
                    type=str,
                    help='command')

group = parser.add_mutually_exclusive_group(required=False)

group.add_argument('-j', '--json',
                    type=str,
                    help='json',
                    required=False)

group.add_argument('parameters',
                    nargs='*',
                    default=[],
                    help='cmd args')

def main():
    """ Main function of the opensips-mi script """
    args = parser.parse_args()

    if args.stats:
        print('Using get_statistics! Be careful not to use command after -s/--stats.')
        print(args.stats)
        args.command = 'get_statistics'

        if args.json:
            print('Cannot use -s/--stats with -j/--json!')
            sys.exit(1)

        args.parameters = {'statistics': args.stats}
    else:
        if args.json:
            try:
                args.parameters = json.loads(args.json)
                print(args.parameters)
            except json.JSONDecodeError as e:
                print('Invalid JSON: ', e)
                sys.exit(1)

    if args.type == 'fifo':
        fifo_args = {}
        if args.fifo_file:
            fifo_args['fifo_file'] = args.fifo_file
        if args.fifo_fallback:
            fifo_args['fifo_file_fallback'] = args.fifo_fallback
        if args.fifo_reply_dir:
            fifo_args['fifo_reply_dir'] = args.fifo_reply_dir
        mi = OpenSIPSMI('fifo', **fifo_args)
    elif args.type == 'http':
        mi = OpenSIPSMI('http', url=f'http://{args.ip}:{args.port}/mi')
    elif args.type == 'datagram':
        mi = OpenSIPSMI('datagram', datagram_ip=args.ip, datagram_port=args.port)
    else:
        print(f'Unknownt type: {args.type}')
        sys.exit(1)

    try:
        response = mi.execute(args.command, args.parameters)
        print(json.dumps(response, indent=4))
    except OpenSIPSMIException as e:
        print('Error: ', e)
        sys.exit(1)

if __name__ == "__main__":
    main()