1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
#!/usr/bin/env python3
# Copyright (C) 2020 VyOS maintainers and contributors
#
# This library is free software; you can redistribute it and/or modify it under the terms of
# the GNU Lesser General Public License as published by the Free Software Foundation;
# either version 2.1 of the License, or (at your option) any later version.
#
# This library 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 Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License along with this library;
# if not, write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
import os
import sys
import pprint
import argparse
from vyos.xml import kw
from vyos.xml import load
# import json
# def save_json(fname, loaded):
# with open(fname, 'w') as w:
# print(f'saving {fname}')
# w.write(json.dumps(loaded))
def save_dict(fname, loaded):
with open(fname, 'w') as w:
print(f'saving {fname}')
w.write(f'# generated by {__file__}\n\n')
w.write('definition = ')
w.write(str(loaded))
def main():
parser = argparse.ArgumentParser(description='generate python file from xml defintions')
parser.add_argument('--conf-folder', type=str, default=load.configuration_definition, help='XML interface definition folder')
parser.add_argument('--conf-cache', type=str, default=load.configuration_cache, help='python file with the conf mode dict')
# parser.add_argument('--op-folder', type=str, default=load.operational_definition, help='XML interface definition folder')
# parser.add_argument('--op-cache', type=str, default=load.operational_cache, help='python file with the conf mode dict')
parser.add_argument('--dry', action='store_true', help='dry run, print to screen')
args = parser.parse_args()
if os.path.exists(load.configuration_cache):
os.remove(load.configuration_cache)
# if os.path.exists(load.operational_cache):
# os.remove(load.operational_cache)
conf = load.xml(args.conf_folder)
# op = load.xml(args.op_folder)
if args.dry:
pprint.pprint(conf)
return
save_dict(args.conf_cache, conf)
# save_dict(args.op_cache, op)
if __name__ == '__main__':
main()
|
