1#!/usr/bin/env python3
 2# SPDX-License-Identifier: GPL-2.0 OR BSD-3-Clause
 3
 4import argparse
 5import json
 6import pprint
 7import time
 8
 9from lib import YnlFamily, Netlink
10
11
12def main():
13    parser = argparse.ArgumentParser(description='YNL CLI sample')
14    parser.add_argument('--spec', dest='spec', type=str, required=True)
15    parser.add_argument('--schema', dest='schema', type=str)
16    parser.add_argument('--no-schema', action='store_true')
17    parser.add_argument('--json', dest='json_text', type=str)
18    parser.add_argument('--do', dest='do', type=str)
19    parser.add_argument('--dump', dest='dump', type=str)
20    parser.add_argument('--sleep', dest='sleep', type=int)
21    parser.add_argument('--subscribe', dest='ntf', type=str)
22    parser.add_argument('--replace', dest='flags', action='append_const',
23                        const=Netlink.NLM_F_REPLACE)
24    parser.add_argument('--excl', dest='flags', action='append_const',
25                        const=Netlink.NLM_F_EXCL)
26    parser.add_argument('--create', dest='flags', action='append_const',
27                        const=Netlink.NLM_F_CREATE)
28    parser.add_argument('--append', dest='flags', action='append_const',
29                        const=Netlink.NLM_F_APPEND)
30    parser.add_argument('--process-unknown', action=argparse.BooleanOptionalAction)
31    args = parser.parse_args()
32
33    if args.no_schema:
34        args.schema = ''
35
36    attrs = {}
37    if args.json_text:
38        attrs = json.loads(args.json_text)
39
40    ynl = YnlFamily(args.spec, args.schema, args.process_unknown)
41
42    if args.ntf:
43        ynl.ntf_subscribe(args.ntf)
44
45    if args.sleep:
46        time.sleep(args.sleep)
47
48    if args.do:
49        reply = ynl.do(args.do, attrs, args.flags)
50        pprint.PrettyPrinter().pprint(reply)
51    if args.dump:
52        reply = ynl.dump(args.dump, attrs)
53        pprint.PrettyPrinter().pprint(reply)
54
55    if args.ntf:
56        ynl.check_ntf()
57        pprint.PrettyPrinter().pprint(ynl.async_msg_queue)
58
59
60if __name__ == "__main__":
61    main()