Something like:

class _ModelInDocument(object):
    # 'models' can be a single Model, a single Document, or a list of either
    def __init__(self, models):
        from .document import Document
        self._to_remove_after = []
        self._theme = []
        if not isinstance(models, list):
            models = [models]

        self._doc = _find_some_document(models)
        if self._doc is None:
            # oh well - just make up a doc
            self._doc = Document()

        for model in models:
            if isinstance(model, Model):
                if model.document is None:
                    # Gets the model theme (or None if no theme)
                    self._theme.append(model.themed_values())
                    self._to_remove_after.append(model)

    def __exit__(self, type, value, traceback):
        for i, model in enumerate(self._to_remove_after):
            model.document.remove_root(model)
            if self._theme[i] is not None:
                # Applies the model theme after removing the document
                model.apply_theme(self._theme[i])

    def __enter__(self):
        for i, model in enumerate(self._to_remove_after):
            # This should preserve model theming here
            self._doc.add_root(model)
            if self._theme[i] is not None:
                # Applies the model theme after adding the document
                model.apply_theme(self._theme[i])

But I'm not sure if this would have repercussions elsewhere - otherwise this could be done around the with loop in embed.py.

Uh oh!

Theme doesn't apply when using components #4952

Description

Metadata

Metadata

Assignees

Labels

Type

Projects

Milestone

Relationships

Development

Issue actions