config: add go type of embeded enums defined in openconfig

these types are embeded enums of openconfig and were left uint32

Signed-off-by: ISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>

1 files changed, 13 insertions, 1 deletions

diff --git a/tools/pyang_plugins/bgpyang2golang.py b/tools/pyang_plugins/bgpyang2golang.py
index cc2b81f2..64d4d97c 100644
--- a/tools/pyang_plugins/bgpyang2golang.py
+++ b/tools/pyang_plugins/bgpyang2golang.py
@@ -166,6 +166,9 @@ def emit_class_def(ctx, yang_statement, struct_name, prefix):
                 else:
                     emit_type_name = t.arg
 
+            # case embeded enumeration
+            elif type_name == 'enumeration':
+                emit_type_name = val_name_go
 
             # case translation required
             elif is_translation_required(type_obj):
@@ -290,6 +293,16 @@ def visit_children(ctx, module, children):
 
         t = c.search_one('type')
 
+        # define container embeded enums
+        if is_leaf(c) and c.search_one('type').arg == 'enumeration':
+            prefix = module.i_prefix
+            c.path = get_path(c)
+            c.golang_name = convert_to_golang(c.arg)
+            if prefix in ctx.golang_typedef_map:
+                ctx.golang_typedef_map[prefix][c.arg] = c
+            else:
+                ctx.golang_typedef_map[prefix] = {c.arg: c}
+
         if is_list(c) or is_container(c) or is_choice(c):
             c.golang_name = convert_to_golang(c.uniq_name)
 
@@ -542,7 +555,6 @@ def is_translation_required(t):
 
 _type_translation_map = {
     'union': 'string',
-    'enumeration': 'uint32',
     'decimal64': 'float64',
     'boolean': 'bool',
     'empty': 'bool',