summaryrefslogtreecommitdiffstats
path: root/plugins/grToolkit/model/scripts/python/yang2props.py
blob: 6a0b33660489bc83eafad5b5e92110f58703cf2f (plain)
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
71
72
73
74
75
76
77
78
#!/usr/bin/python

###
# ============LICENSE_START=======================================================
# openECOMP : SDN-C
# ================================================================================
# Copyright (C) 2018 AT&T Intellectual Property. All rights
# 			reserved.
# ================================================================================
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============LICENSE_END=========================================================
###

import re
import sys


# Convert word from foo-bar to FooBar
# words begining with a digit will be converted to _digit
def to_enum(s):
    if s[0].isdigit():
        s = "_" + s
    else:
        s = s[0].upper() + s[1:]
    return re.sub(r'(?!^)-([a-zA-Z])', lambda m: m.group(1).upper(), s)

leaf = ""
val = ""
li = []

if len(sys.argv) < 3:
     print('yang2props.py <input yang> <output properties>')
     sys.exit(2)

with open(sys.argv[1], "r") as ins:
    for line in ins:
        # if we see a leaf save the name for later
        if "leaf " in line:
	    match = re.search(r'leaf (\S+)', line)
	    if match:
                 leaf = match.group(1)
      
        # if we see enum convert the value to enum format and see if it changed
        # if the value is different write a property entry
        if "enum " in line:
	    match = re.search(r'enum "(\S+)";', line)
	    if match:
                val = match.group(1)
            	enum = to_enum(val)

                # see if converting to enum changed the string
		if val != enum:
                    property = "yang."+leaf+"."+enum+"="+val
		    if property not in li:
		        li.append( property)


# Open output file
fo = open(sys.argv[2], "wb")
fo.write("# yang conversion properties \n")
fo.write("# used to convert Enum back to the original yang value \n")
fo.write("\n".join(li))
fo.write("\n")

# Close opend file
fo.close()