Source code for juju.client.codegen

from collections import defaultdict
from io import StringIO
from textwrap import indent

[docs]class CodeWriter(StringIO): """ Blob of text that, when used in the context of, ends up holding the source code for a Python class and associated methods. """ INDENT = " " CLASS = 0 METHOD = 1
[docs] def write(self, msg, depth=0): if depth: prefix = self.INDENT * depth msg = indent(msg, prefix) return super(CodeWriter, self).write(msg)
def __str__(self): return super(CodeWriter, self).getvalue()
[docs]class Capture(defaultdict): """ A collection of CodeWriter objects, together representing a Python module. """ def __init__(self, default_factory=CodeWriter, *args, **kwargs): super(Capture, self).__init__(default_factory, *args, **kwargs)
[docs] def clear(self, name): """ Reset one of the keys in this class, if it exists. This is necessary, because we don't worry about de-duplicating the schemas for each version of juju up front, and this gives us a way to sort of de-duplicate on the fly, by resetting a specific CodeWriter instance before we start to write a class into it. """ try: del self[name] except KeyError: pass