summaryrefslogtreecommitdiff
path: root/tests/elements/filter/basic/element_plugins/dynamic.py
blob: 6cd6b10931c46e3274d25ba7ad1cb7327a063bb3 (plain)
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
from buildstream import Element, Scope


# Copies files from the dependent element but inserts split-rules using dynamic data
class DynamicElement(Element):
    def configure(self, node):
        node.validate_keys(["split-rules"])
        self.split_rules = {
            key: value.as_str_list()
            for key, value in node.get_mapping("split-rules").items()
        }

    def preflight(self):
        pass

    def get_unique_key(self):
        return {"split-rules": self.split_rules}

    def configure_sandbox(self, sandbox):
        pass

    def stage(self, sandbox):
        pass

    def assemble(self, sandbox):
        with self.timed_activity("Staging artifact", silent_nested=True):
            for dep in self.dependencies(Scope.BUILD):
                dep.stage_artifact(sandbox)

        bstdata = self.get_public_data("bst")
        bstdata["split-rules"] = self.split_rules
        self.set_public_data("bst", bstdata)

        return ""


def setup():
    return DynamicElement