¡@

Home 

OpenStack Study: security_group.py

OpenStack Index

**** CubicPower OpenStack Study ****

# Copyright 2013 IBM Corp.

#

# 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 nova import db

from nova.objects import base

from nova.objects import fields

**** CubicPower OpenStack Study ****

class SecurityGroup(base.NovaPersistentObject, base.NovaObject):

# Version 1.0: Initial version

# Version 1.1: String attributes updated to support unicode

VERSION = '1.1'

fields = {

'id': fields.IntegerField(),

'name': fields.StringField(),

'description': fields.StringField(),

'user_id': fields.StringField(),

'project_id': fields.StringField(),

}

@staticmethod

**** CubicPower OpenStack Study ****

    def _from_db_object(context, secgroup, db_secgroup):

        # NOTE(danms): These are identical right now

        for field in secgroup.fields:

            secgroup[field] = db_secgroup[field]

        secgroup._context = context

        secgroup.obj_reset_changes()

        return secgroup

    @base.remotable_classmethod

**** CubicPower OpenStack Study ****

    def get(cls, context, secgroup_id):

        db_secgroup = db.security_group_get(context, secgroup_id)

        return cls._from_db_object(context, cls(), db_secgroup)

    @base.remotable_classmethod

**** CubicPower OpenStack Study ****

    def get_by_name(cls, context, project_id, group_name):

        db_secgroup = db.security_group_get_by_name(context,

                                                    project_id,

                                                    group_name)

        return cls._from_db_object(context, cls(), db_secgroup)

    @base.remotable

**** CubicPower OpenStack Study ****

    def in_use(self, context):

        return db.security_group_in_use(context, self.id)

    @base.remotable

**** CubicPower OpenStack Study ****

    def save(self, context):

        updates = self.obj_get_changes()

        if updates:

            db_secgroup = db.security_group_update(context, self.id, updates)

            SecurityGroup._from_db_object(context, self, db_secgroup)

        self.obj_reset_changes()

    @base.remotable

**** CubicPower OpenStack Study ****

    def refresh(self, context):

        SecurityGroup._from_db_object(context, self,

                                      db.security_group_get(context,

                                                            self.id))

**** CubicPower OpenStack Study ****

class SecurityGroupList(base.ObjectListBase, base.NovaObject):

# Version 1.0: Initial version

# SecurityGroup <= version 1.1

VERSION = '1.0'

fields = {

'objects': fields.ListOfObjectsField('SecurityGroup'),

}

child_versions = {

'1.0': '1.1',

# NOTE(danms): SecurityGroup was at 1.1 before we added this

}

**** CubicPower OpenStack Study ****

    def __init__(self):

        super(SecurityGroupList, self).__init__()

        self.objects = []

        self.obj_reset_changes()

    @base.remotable_classmethod

**** CubicPower OpenStack Study ****

    def get_all(cls, context):

        groups = db.security_group_get_all(context)

        return base.obj_make_list(context, SecurityGroupList(), SecurityGroup,

                                  groups)

    @base.remotable_classmethod

**** CubicPower OpenStack Study ****

    def get_by_project(cls, context, project_id):

        groups = db.security_group_get_by_project(context, project_id)

        return base.obj_make_list(context, SecurityGroupList(), SecurityGroup,

                                  groups)

    @base.remotable_classmethod

**** CubicPower OpenStack Study ****

    def get_by_instance(cls, context, instance):

        groups = db.security_group_get_by_instance(context, instance.uuid)

        return base.obj_make_list(context, SecurityGroupList(), SecurityGroup,

                                  groups)

def make_secgroup_list(security_groups):

    """A helper to make security group objects from a list of names.

    Note that this does not make them save-able or have the rest of the

    attributes they would normally have, but provides a quick way to fill,

    for example, an instance object during create.

    """

    secgroups = SecurityGroupList()

    secgroups.objects = []

    for name in security_groups:

        secgroup = SecurityGroup()

        secgroup.name = name

        secgroups.objects.append(secgroup)

    return secgroups

**** CubicPower OpenStack Study ****

def make_secgroup_list(security_groups):

    """A helper to make security group objects from a list of names.

    Note that this does not make them save-able or have the rest of the

    attributes they would normally have, but provides a quick way to fill,

    for example, an instance object during create.

    """

    secgroups = SecurityGroupList()

    secgroups.objects = []

    for name in security_groups:

        secgroup = SecurityGroup()

        secgroup.name = name

        secgroups.objects.append(secgroup)

    return secgroups