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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
|
# Copyright (C) 2013,2014 Nippon Telegraph and Telephone Corporation.
# Copyright (C) 2013,2014 YAMAMOTO Takashi <yamamoto at valinux co jp>
#
# 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.
# this is a pyang plugin to generate ryu/lib/of_config/generated_classes.py
# usage example:
# PYTHONPATH=. ./bin/pyang --plugindir ~/git/ryu/tools/pyang_plugins -f ryu ~/git/ryu/tools/of-config1.1.1.yang > ~/git/ryu/lib/of_config/generated_classes.py
_COPYRIGHT_NOTICE = """
// Copyright (C) 2013,2014 Nippon Telegraph and Telephone Corporation.
//
// 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 sys
import StringIO
import pyang
from pyang import plugin
struct_map = {}
def pyang_plugin_init():
plugin.register_plugin(GolangPlugin())
class GolangPlugin(plugin.PyangPlugin):
def add_output_format(self, fmts):
fmts['golang'] = self
def emit(self, ctx, modules, fd):
emit_golang(ctx, modules[0], fd)
def emit_golang(ctx, module, fd):
ctx.golang_struct_def = []
visit_children(ctx, module, fd, module.i_children)
ctx.golang_struct_def.reverse()
done = set()
generate_header(ctx)
for struct in ctx.golang_struct_def:
struct_name = struct.arg
if struct_name in done:
continue
emit_class_def(struct, struct_name)
done.add(struct_name)
# generate_header(ctx)
def emit_class_def(c, struct_name):
o = StringIO.StringIO()
struct_name_org = struct_name
struct_name = convert_to_gostruct(struct_name)
struct_map[struct_name_org] = struct_name
print >> o, '//struct for container %s' % struct_name_org
print >> o, 'type %s struct {' % struct_name
for child in c.i_children:
val_name_org = child.arg
val_name = convert_to_golang(child.arg)
if is_leaf(child):
type_obj = child.search_one('type')
#if type_obj.arg == 'leafref':
# print type_obj.search_one('path').arg
type_name = type_obj.arg if type_obj is not None else None
else:
if is_list(child):
assert val_name_org in struct_map
type_name = '[]'+ struct_map[val_name_org]
val_name = val_name + 'List'
if is_container(child):
type_name = struct_map[val_name_org]
val_name = val_name
print >> o, ' %s\t%s' % (val_name, translate_type(type_name))
print >> o, '}'
print o.getvalue()
def visit_children(ctx, module, fd, children, prefix=''):
for c in children:
t = c.search_one('type')
type_name = t.arg if t is not None else None
#print '%skeyword->%s, arg->%s, type->%s' % (prefix, c.keyword, c.arg, type_name)
if is_list(c) or is_container(c):
ctx.golang_struct_def.append(c)
if hasattr(c, 'i_children'):
visit_children(ctx, module, fd, c.i_children, prefix + ' ')
def is_leaf(s):
return s.keyword in ['leaf', 'leaf-list']
def is_list(s):
return s.keyword in ['list']
def is_container(s):
return s.keyword in ['container']
def generate_header(ctx):
print _COPYRIGHT_NOTICE
print 'package config'
print ''
_type_translation_map = {
'decimal64' : 'float64',
'inet:ip-address': '*net.IP',
'inet:ipv4-address': '*net.IP',
'inet:as-number' : 'uint32',
'rr-cluster-id-type' : 'uint32',
}
def translate_type(key):
if _type_translation_map.has_key(key):
return _type_translation_map[key]
else:
return key
def convert_to_gostruct(type_string):
return convert_to_golang(type_string) + 'Type'
# 'hoge-hoge' -> 'HogeHoge'
def convert_to_golang(type_string):
a = type_string.split('-')
a = map(lambda x: x.capitalize(), a) # XXX locale sensitive
return ''.join(a)
|
