OpenStack Study: types.py

**** CubicPower OpenStack Study ****

# Copyright 2011 OpenStack Foundation

# All Rights Reserved.

#

# 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.

"""Custom SQLAlchemy types."""

from sqlalchemy.dialects import postgresql

from sqlalchemy import types

from nova import utils

**** CubicPower OpenStack Study ****

class IPAddress(types.TypeDecorator):

"""An SQLAlchemy type representing an IP-address."""

impl = types.String

**** CubicPower OpenStack Study ****

    def load_dialect_impl(self, dialect):

        if dialect.name == 'postgresql':

            return dialect.type_descriptor(postgresql.INET())

        else:

            return dialect.type_descriptor(types.String(39))

**** CubicPower OpenStack Study ****

    def process_bind_param(self, value, dialect):

        """Process/Formats the value before insert it into the db."""

        if dialect.name == 'postgresql':

            return value

        # NOTE(maurosr): The purpose here is to convert ipv6 to the shortened

        # form, not validate it.

        elif utils.is_valid_ipv6(value):

            return utils.get_shortened_ipv6(value)

        return value

**** CubicPower OpenStack Study ****

class CIDR(types.TypeDecorator):

"""An SQLAlchemy type representing a CIDR

**** CubicPower OpenStack Study ****

    def load_dialect_impl(self, dialect):

        if dialect.name == 'postgresql':

            return dialect.type_descriptor(postgresql.INET())

        else:

            return dialect.type_descriptor(types.String(43))

**** CubicPower OpenStack Study ****

    def process_bind_param(self, value, dialect):

        """Process/Formats the value before insert it into the db."""

        # NOTE(sdague): normalize all the inserts

        if utils.is_valid_ipv6_cidr(value):

            return utils.get_shortened_ipv6_cidr(value)

        return value