cli.py: fix Pylint errors
[deliverable/barectf.git] / barectf / codegen.py
1 # The MIT License (MIT)
2 #
3 # Copyright (c) 2015-2020 Philippe Proulx <pproulx@efficios.com>
4 #
5 # Permission is hereby granted, free of charge, to any person obtaining
6 # a copy of this software and associated documentation files (the
7 # "Software"), to deal in the Software without restriction, including
8 # without limitation the rights to use, copy, modify, merge, publish,
9 # distribute, sublicense, and/or sell copies of the Software, and to
10 # permit persons to whom the Software is furnished to do so, subject to
11 # the following conditions:
12 #
13 # The above copyright notice and this permission notice shall be
14 # included in all copies or substantial portions of the Software.
15 #
16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
17 # EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
18 # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
19 # IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
20 # CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
21 # TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
22 # SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
23
24 class _CodeGenerator:
25 def __init__(self, indent_string):
26 self._indent_string = indent_string
27 self.reset()
28
29 @property
30 def code(self):
31 return '\n'.join(self._lines)
32
33 def reset(self):
34 self._lines = []
35 self._indent = 0
36 self._glue = False
37
38 def add_line(self, line):
39 if self._glue:
40 self.append_to_last_line(line)
41 self._glue = False
42 return
43
44 indent_string = self._get_indent_string()
45 self._lines.append(indent_string + str(line))
46
47 def add_lines(self, lines):
48 if type(lines) is str:
49 lines = lines.split('\n')
50
51 for line in lines:
52 self.add_line(line)
53
54 def add_glue(self):
55 self._glue = True
56
57 def append_to_last_line(self, s):
58 if self._lines:
59 self._lines[-1] += str(s)
60
61 def add_empty_line(self):
62 self._lines.append('')
63
64 def add_cc_line(self, comment):
65 self.add_line('/* {} */'.format(comment))
66
67 def append_cc_to_last_line(self, comment, with_space=True):
68 if with_space:
69 sp = ' '
70 else:
71 sp = ''
72
73 self.append_to_last_line('{}/* {} */'.format(sp, comment))
74
75 def indent(self):
76 self._indent += 1
77
78 def unindent(self):
79 self._indent = max(self._indent - 1, 0)
80
81 def _get_indent_string(self):
82 return self._indent_string * self._indent
This page took 0.055997 seconds and 4 git commands to generate.