¡@

Home 

OpenStack Study: sql.py

OpenStack Index

**** CubicPower OpenStack Study ****

# Copyright 2013 OpenStack 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 keystone.common import sql

from keystone.common.sql import migration_helpers

from keystone import credential

from keystone import exception

from keystone.openstack.common.db.sqlalchemy import migration

**** CubicPower OpenStack Study ****

class CredentialModel(sql.ModelBase, sql.DictBase):

__tablename__ = 'credential'

attributes = ['id', 'user_id', 'project_id', 'blob', 'type']

id = sql.Column(sql.String(64), primary_key=True)

user_id = sql.Column(sql.String(64),

nullable=False)

project_id = sql.Column(sql.String(64))

blob = sql.Column(sql.JsonBlob(), nullable=False)

type = sql.Column(sql.String(255), nullable=False)

extra = sql.Column(sql.JsonBlob())

**** CubicPower OpenStack Study ****

class Credential(credential.Driver):

# Internal interface to manage the database

**** CubicPower OpenStack Study ****

    def db_sync(self, version=None):

        migration.db_sync(

            sql.get_engine(), migration_helpers.find_migrate_repo(),

            version=version)

    # credential crud

    @sql.handle_conflicts(conflict_type='credential')

**** CubicPower OpenStack Study ****

    def create_credential(self, credential_id, credential):

        session = sql.get_session()

        with session.begin():

            ref = CredentialModel.from_dict(credential)

            session.add(ref)

        return ref.to_dict()

**** CubicPower OpenStack Study ****

    def list_credentials(self, **filters):

        session = sql.get_session()

        query = session.query(CredentialModel)

        if 'user_id' in filters:

            query = query.filter_by(user_id=filters.get('user_id'))

        refs = query.all()

        return [ref.to_dict() for ref in refs]

**** CubicPower OpenStack Study ****

    def _get_credential(self, session, credential_id):

        ref = session.query(CredentialModel).get(credential_id)

        if ref is None:

            raise exception.CredentialNotFound(credential_id=credential_id)

        return ref

**** CubicPower OpenStack Study ****

    def get_credential(self, credential_id):

        session = sql.get_session()

        return self._get_credential(session, credential_id).to_dict()

    @sql.handle_conflicts(conflict_type='credential')

**** CubicPower OpenStack Study ****

    def update_credential(self, credential_id, credential):

        session = sql.get_session()

        with session.begin():

            ref = self._get_credential(session, credential_id)

            old_dict = ref.to_dict()

            for k in credential:

                old_dict[k] = credential[k]

            new_credential = CredentialModel.from_dict(old_dict)

            for attr in CredentialModel.attributes:

                if attr != 'id':

                    setattr(ref, attr, getattr(new_credential, attr))

            ref.extra = new_credential.extra

        return ref.to_dict()

**** CubicPower OpenStack Study ****

    def delete_credential(self, credential_id):

        session = sql.get_session()

        with session.begin():

            ref = self._get_credential(session, credential_id)

            session.delete(ref)

**** CubicPower OpenStack Study ****

    def delete_credentials_for_project(self, project_id):

        session = sql.get_session()

        with session.begin():

            query = session.query(CredentialModel)

            query = query.filter_by(project_id=project_id)

            query.delete()

**** CubicPower OpenStack Study ****

    def delete_credentials_for_user(self, user_id):

        session = sql.get_session()

        with session.begin():

            query = session.query(CredentialModel)

            query = query.filter_by(user_id=user_id)

            query.delete()