xos/tosca/resources/xossite.py

# Copyright 2017-present Open Networking Foundation
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.


from xosresource import XOSResource
from core.models import User,Site,Deployment,Controller,SiteDeployment

class XOSSite(XOSResource):
    provides = "tosca.nodes.Site"
    xos_model = Site

    def get_xos_args(self):
        display_name = self.get_property("display_name")
        if not display_name:
            display_name = self.obj_name

        args = {"login_base": self.obj_name,
                "name": display_name}

        # copy simple string properties from the template into the arguments
        for prop in ["site_url", ]:
            v = self.get_property(prop)
            if v:
                args[prop] = v

        return args

    def get_existing_objs(self):
        return self.xos_model.objects.filter(login_base = self.obj_name)

    def postprocess(self, obj):
        # The old way - DEPRECATED
        for reqs in self.nodetemplate.requirements:
            for (k,v) in reqs.items():
                if (v["relationship"] == "tosca.relationships.SiteDeployment"):
                    deployment_name = v["node"]
                    deployment = self.get_xos_object(Deployment, name=deployment_name)

                    controller_name = None
                    for sd_req in v.get("requirements", []):
                        for (sd_req_k, sd_req_v) in sd_req.items():
                            if sd_req_v["relationship"] == "tosca.relationships.UsesController":
                                controller_name = sd_req_v["node"]
                    if controller_name:
                        controller = self.get_xos_object(Controller, name=controller_name, throw_exception=True)
                    else:
                        controller = None
                        # raise Exception("Controller must be specified in SiteDeployment relationship")

                    existing_sitedeps = SiteDeployment.objects.filter(deployment=deployment, site=obj)
                    if existing_sitedeps:
                        sd = existing_sitedeps[0]
                        if (sd.controller != controller) and (controller != None):
                            sd.controller = controller
                            sd.save()
                            self.info("SiteDeployment from %s to %s updated controller" % (str(obj), str(deployment)))
                        else:
                            self.info("SiteDeployment from %s to %s already exists" % (str(obj), str(deployment)))
                    else:
                        sitedep = SiteDeployment(deployment=deployment, site=obj, controller=controller)
                        sitedep.save()
                        self.info("Created SiteDeployment from %s to %s" % (str(obj), str(deployment)))

        # The new way
        deployments = self.get_requirements("tosca.relationships.MemberOfDeployment")
        if deployments and len(deployments)>1:
            raise Exception("At most one deployment per site (%s)" % str(deployments))

        controllers = self.get_requirements("tosca.relationships.UsesController")
        if controllers and len(controllers)>1:
            raise Exception("At most one controller per site (%s)" % str(controllers))

        if controllers or deployments:
            if (not deployments):
                raise Exception("If you specify a controller, then you must specify a deployment")

            deployment = self.get_xos_object(Deployment, name = deployments[0], throw_exception = True)
            if controllers:
                controller = self.get_xos_object(Controller, name = controllers[0], throw_exception = True)
            else:
                controller = None
            existing_sitedeps = SiteDeployment.objects.filter(site=obj)
            if (existing_sitedeps):
                sd = existing_sitedeps[0]
                if (sd.controller != controller) or (sd.deployment != deployment):
                    sd.controller = controller
                    sd.deployment = deployment
                    sd.save()
                    self.info("SiteDeployment from %s to %s updated" % (str(obj), str(deployment)))
            else:
                sd = SiteDeployment(deployment=deployment, site=obj, controller=controller)
                sd.save()
                self.info("Created SiteDeployment from %s to %s" % (str(obj), str(deployment)))

    def delete(self, obj):
        if obj.slices.exists():
            self.info("Site %s has active slices; skipping delete" % obj.name)
            return
        if obj.users.exists():
            self.info("Site %s has active users; skipping delete" % obj.name)
            return
        if obj.nodes.exists():
            self.info("Site %s has active nodes; skipping delete" % obj.name)
            return
        super(XOSSite, self).delete(obj)