Remove dependency on vendored libjuju
[osm/N2VC.git] / modules / libjuju / juju / client / codegen.py
1 from collections import defaultdict
2 from io import StringIO
3 from textwrap import indent
4
5
6 class CodeWriter(StringIO):
7 """
8 Blob of text that, when used in the context of facade.py, ends up
9 holding the source code for a Python class and associated methods.
10
11 """
12 INDENT = " "
13
14 CLASS = 0
15 METHOD = 1
16
17 def write(self, msg, depth=0):
18 if depth:
19 prefix = self.INDENT * depth
20 msg = indent(msg, prefix)
21
22 return super(CodeWriter, self).write(msg)
23
24 def __str__(self):
25 return super(CodeWriter, self).getvalue()
26
27
28 class Capture(defaultdict):
29 """
30 A collection of CodeWriter objects, together representing a Python
31 module.
32
33 """
34
35 def __init__(self, default_factory=CodeWriter, *args, **kwargs):
36 super(Capture, self).__init__(default_factory, *args, **kwargs)
37
38 def clear(self, name):
39 """
40 Reset one of the keys in this class, if it exists.
41
42 This is necessary, because we don't worry about de-duplicating
43 the schemas for each version of juju up front, and this gives
44 us a way to sort of de-duplicate on the fly, by resetting a
45 specific CodeWriter instance before we start to write a class
46 into it.
47
48 """
49 try:
50 del self[name]
51 except KeyError:
52 pass