larigira/larigira/event_manage.py

from __future__ import print_function
import argparse
import json

from .event import EventModel
from .config import get_conf


def main_list(args):
    m = EventModel(args.file)
    for alarm, action in m.get_all_alarms_expanded():
        print(json.dumps(dict(alarm=alarm, action=action), indent=4))


def main_add(args):
    m = EventModel(args.file)
    m.add_event(dict(kind='frequency', interval=args.interval, start=1),
                [dict(kind='mpd', howmany=1)]
                )


def main():
    conf = get_conf()
    p = argparse.ArgumentParser()
    p.add_argument('-f', '--file', help="Filepath for DB", required=False,
                   default=conf['DB_URI'])
    sub = p.add_subparsers()
    sub_list = sub.add_parser('list')
    sub_list.set_defaults(func=main_list)
    sub_add = sub.add_parser('add')
    sub_add.add_argument('--interval', type=int, default=3600)
    sub_add.set_defaults(func=main_add)

    args = p.parse_args()
    args.func(args)

if __name__ == '__main__':
    main()