blob: afbb3a1ea1ea01aad72c820f82c8e004777ac7dc (
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
|
# 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.
import os
import sys
import getopt
import argparse
from lxml import etree
def tosca_translator(file_name):
if os.path.exists(file_name):
var = file_name
file_format = var.split('.')
if len(file_format) == 2:
if file_format[1] == 'yaml' or file_format[1] == 'yang':
print("Its a yaml file")
os.system('pyang -f yin '+var + ' -o ' + file_format[0]+'.xml')
elif file_format[1] == 'xml':
print ("Its a "+file_format[1]+" file")
else:
print ("File format not supported exiting script.")
exit()
else:
print ("File format not supported exiting script.")
exit()
tree = etree.parse(file_format[0]+'.xml')
doc = tree.getroot()
xslt_root = etree.parse('tosca_transformer.xslt')
transform = etree.XSLT(xslt_root)
result_tree = transform(doc)
output_file = file_format[0]+'_tosca.yaml'
f = open(output_file, "w")
f.write(str(result_tree))
f.close()
print ("TOSCA file generated with name "+output_file)
else:
print ("File does not exist, exiting the script")
exit()
if __name__ == '__main__':
parser = argparse.ArgumentParser(add_help=True)
parser.add_argument('-n', '--filename', dest='filename', required=True,
help='Please enter the YANG file name: ')
args = parser.parse_args()
file_name = str(args.filename)
tosca_translator(file_name)
|
