cf-python-client/cloudfoundry_client/v3/buildpacks.py at df2620df7b20038fdcb6ea84b8045497a3401f4f · cloudfoundry-community/cf-python-client · GitHub

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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
from typing import Optional, TYPE_CHECKING

from cloudfoundry_client.v3.entities import EntityManager, Entity

if TYPE_CHECKING:
    from cloudfoundry_client.client import CloudFoundryClient


class BuildpackManager(EntityManager):
    def __init__(self, target_endpoint: str, client: "CloudFoundryClient"):
        super(BuildpackManager, self).__init__(target_endpoint, client, "/v3/buildpacks")

    def create(
        self,
        name: str,
        position: Optional[int] = 0,
        enabled: Optional[bool] = True,
        locked: Optional[bool] = False,
        stack: Optional[str] = None,
        meta_labels: Optional[dict] = None,
        meta_annotations: Optional[dict] = None,
    ) -> Entity:
        data = {
            "name": name,
            "position": position,
            "enabled": enabled,
            "locked": locked,
            "stack": stack,
            "metadata": {"labels": meta_labels, "annotations": meta_annotations},
        }
        return super(BuildpackManager, self)._create(data)

    def remove(self, buildpack_guid: str, asynchronous: bool = True) -> Optional[str]:
        return super(BuildpackManager, self)._remove(buildpack_guid, asynchronous)

    def update(
        self,
        buildpack_guid: str,
        name: str,
        position: Optional[int] = 0,
        enabled: Optional[bool] = True,
        locked: Optional[bool] = False,
        stack: Optional[str] = None,
        meta_labels: Optional[dict] = None,
        meta_annotations: Optional[dict] = None,
    ) -> Entity:
        data = {
            "name": name,
            "position": position,
            "enabled": enabled,
            "locked": locked,
            "stack": stack,
            "metadata": {"labels": meta_labels, "annotations": meta_annotations},
        }
        return super(BuildpackManager, self)._update(buildpack_guid, data)

    def upload(self, buildpack_guid: str, buildpack_zip: str, asynchronous: bool = False) -> Entity:
        buildpack = super(BuildpackManager, self)._upload_bits(buildpack_guid, buildpack_zip)
        if not asynchronous:
            self.client.v3.jobs.wait_for_job_completion(buildpack.job()["guid"])
            buildpack_after_job = super(BuildpackManager, self).get(buildpack["guid"])
            buildpack_after_job["links"]["job"] = buildpack["links"]["job"]
            buildpack_after_job.job = buildpack.job
            return buildpack_after_job
        return buildpack