controller/rootfs/api/admissions.py at 814912264c689c3ccbc1750c0edf4f701efd2943 · drycc/controller · 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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
from django.db.models import F, Func, Value, JSONField
from rest_framework.request import Request

from api import models


class BaseHandler(object):

    def detect(self, request: Request) -> bool:
        raise NotImplementedError()

    def handle(self, request: Request) -> bool:
        raise NotImplementedError()


class JobsStatusHandler(BaseHandler):

    def detect(self, request: Request) -> bool:
        group = request.get("resource", {}).get("group", None)
        resource = "/".join([
            request.get("resource", {}).get("resource", None),
            request.get("subResource", ""),
        ])
        if (group, resource) == ("batch", "jobs/status"):
            return True
        return False

    def handle(self, request: Request) -> bool:
        app_id = request["object"]["metadata"]["namespace"]
        app = models.app.App.objects.filter(id=app_id).first()
        ptype = request["object"]["metadata"].get("labels", {}).get("type", "")
        if app and ptype:
            status = request["object"]["status"]
            replicas = request["object"]["spec"].get("replicas", 0)
            if "active" in status:
                replicas += 1
            elif "succeeded" in status or "failed" in status:
                replicas -= 1
            replicas = 0 if replicas < 0 else replicas
            if app.structure.get(ptype, 0) != replicas:
                models.app.App.objects.filter(id=app.id).update(
                    structure=Func(
                        F("structure"),
                        Value([ptype]),
                        Value(replicas, JSONField()),
                        function="jsonb_set",
                    )
                )
        return True


class DeploymentsScaleHandler(BaseHandler):

    def detect(self, request: Request) -> bool:
        group = request.get("resource", {}).get("group", None)
        resource = "/".join([
            request.get("resource", {}).get("resource", None),
            request.get("subResource", ""),
        ])
        if (group, resource) == ("apps", "deployments/scale"):
            return True
        return False

    def handle(self, request: Request) -> bool:
        app_id = request["object"]["metadata"]["namespace"]
        app = models.app.App.objects.filter(id=app_id).first()
        ptype = None
        for item in request["object"]["status"]["selector"].split(","):
            key, value = item.split("=")
            if key == "type":
                ptype = value
        if app and ptype:
            replicas = request["object"]["spec"].get("replicas", 0)
            if app.structure.get(ptype, 0) != replicas:
                models.app.App.objects.filter(id=app.id).update(
                    structure=Func(
                        F("structure"),
                        Value([ptype]),
                        Value(replicas, JSONField()),
                        function="jsonb_set",
                    )
                )
        return True


class ServiceInstancesStatusHandler(BaseHandler):

    def detect(self, request: Request) -> bool:
        group = request.get("resource", {}).get("group", None)
        resource = "/".join([
            request.get("resource", {}).get("resource", None),
            request.get("subResource", ""),
        ])
        if (group, resource) == ("servicecatalog.k8s.io", "serviceinstances/status"):
            return True
        return False

    def handle(self, request: Request) -> bool:
        app_id = request["object"]["metadata"]["namespace"]
        name = request["object"]["metadata"]["name"]
        status = request["object"]["status"]["lastConditionState"]
        resource = models.resource.Resource.objects.filter(
            app__id=app_id, name=name).first()
        if resource and resource.status != status:
            resource.status = status
            resource.save(update_fields=["status"])
        return True


class ServicebindingsStatusHandler(BaseHandler):

    def detect(self, request: Request) -> bool:
        group = request.get("resource", {}).get("group", None)
        resource = "/".join([
            request.get("resource", {}).get("resource", None),
            request.get("subResource", ""),
        ])
        if (group, resource) == ("servicecatalog.k8s.io", "servicebindings/status"):
            return True
        return False

    def handle(self, request: Request) -> bool:
        app_id = request["object"]["metadata"]["namespace"]
        name = request["object"]["metadata"]["name"]
        binding = request["object"]["status"]["lastConditionState"]
        resource = models.resource.Resource.objects.filter(
            app__id=app_id, name=name).first()
        if resource and resource.binding != binding:
            resource.binding = binding
            resource.save(update_fields=["binding"])
        return True