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
|
"""
Enums transformation
"""
import logging
import textwrap
from collections import namedtuple, OrderedDict
from model.enum import Enum
from model.enum_element import EnumElement
from transformers.common_producer import InterfaceProducerCommon
class EnumsProducer(InterfaceProducerCommon):
"""
Enums transformation
"""
def __init__(self, paths, key_words):
super(EnumsProducer, self).__init__(
container_name='elements',
enums_package=None,
structs_package=None,
package_name=paths.enums_package,
key_words=key_words)
self.logger = logging.getLogger('EnumsProducer')
self._params = namedtuple('params', 'origin name internal description since value deprecated')
@staticmethod
def converted(name):
if name[0].isdigit():
name = '_' + name
if '-' in name:
name = name.replace('-', '_')
return name
def transform(self, item: Enum) -> dict:
"""
Override
:param item: particular element from initial Model
:return: dictionary to be applied to jinja2 template
"""
imports = set()
params = OrderedDict()
if any(map(lambda l: l.name != self.converted(l.name), getattr(item, self.container_name).values())):
kind = 'custom'
imports.add('java.util.EnumSet')
else:
kind = 'simple'
for param in getattr(item, self.container_name).values():
p = self.extract_param(param, kind)
params[p.name] = p
render = OrderedDict()
render['kind'] = kind
render['package_name'] = self.package_name
render['class_name'] = item.name[:1].upper() + item.name[1:]
render['params'] = params
render['since'] = item.since
render['deprecated'] = item.deprecated
description = self.extract_description(item.description)
if description:
render['description'] = description
if imports:
render['imports'] = imports
render['params'] = tuple(render['params'].values())
if 'description' in render and isinstance(render['description'], str):
render['description'] = textwrap.wrap(render['description'], 90)
return render
def extract_param(self, param: EnumElement, kind):
d = {'origin': param.name, 'name': self.converted(param.name)}
if kind == 'custom':
d['internal'] = '"{}"'.format(param.name)
if getattr(param, 'since', None):
d['since'] = param.since
if getattr(param, 'deprecated', None):
d['deprecated'] = param.deprecated
if getattr(param, 'description', None):
d['description'] = textwrap.wrap(self.extract_description(param.description), 90)
Params = namedtuple('Params', sorted(d))
return Params(**d)
|
