a4e583f1c6
* API now executes (not fully working but a good start) * Load balancers list now works * Load balancer detail now works * Fix pep8, pyflakes + Shrews' comments * Temporarily make test cases pass Change-Id: Ia082b8bb60a95abee086073908256649e4ebca23
110 lines
4.1 KiB
Python
110 lines
4.1 KiB
Python
# vim: tabstop=4 shiftwidth=4 softtabstop=4
|
|
# Copyright 2013 Hewlett-Packard Development Company, L.P.
|
|
#
|
|
# 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 sqlalchemy import Table, Column, Integer, ForeignKey, create_engine
|
|
from sqlalchemy import INTEGER, VARCHAR, TIMESTAMP, BIGINT
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.orm import relationship, backref, sessionmaker
|
|
from pecan import conf
|
|
|
|
# TODO replace this with something better
|
|
conn_string = '''mysql://%s:%s@%s/%s''' % (
|
|
conf.database.username,
|
|
conf.database.password,
|
|
conf.database.host,
|
|
conf.database.schema
|
|
)
|
|
|
|
engine = create_engine(conn_string)
|
|
DeclarativeBase = declarative_base()
|
|
metadata = DeclarativeBase.metadata
|
|
metadata.bind = engine
|
|
|
|
loadbalancers_devices = Table(
|
|
'loadbalancers_devices',
|
|
metadata,
|
|
Column('loadbalancer', Integer, ForeignKey('loadbalancers.id')),
|
|
Column('device', Integer, ForeignKey('devices.id'))
|
|
)
|
|
|
|
|
|
class Device(DeclarativeBase):
|
|
"""device model"""
|
|
__tablename__ = 'devices'
|
|
#column definitions
|
|
az = Column(u'az', INTEGER(), nullable=False)
|
|
created = Column(u'created', TIMESTAMP(), nullable=False)
|
|
floatingIpAddr = Column(
|
|
u'floatingIpAddr', VARCHAR(length=128), nullable=False
|
|
)
|
|
id = Column(u'id', BIGINT(), primary_key=True, nullable=False)
|
|
name = Column(u'name', VARCHAR(length=128), nullable=False)
|
|
publicIpAddr = Column(u'publicIpAddr', VARCHAR(length=128), nullable=False)
|
|
status = Column(u'status', VARCHAR(length=128), nullable=False)
|
|
type = Column(u'type', VARCHAR(length=128), nullable=False)
|
|
updated = Column(u'updated', TIMESTAMP(), nullable=False)
|
|
|
|
def find_free_device(self):
|
|
"""queries for free and clear device
|
|
|
|
sql form java api
|
|
SELECT * FROM devices WHERE loadbalancers = " + EMPTY_LBIDS + " AND
|
|
status = '" + Device.STATUS_OFFLINE + "'" ;
|
|
"""
|
|
return session.query(Device).\
|
|
filter_by(loadbalancers="", status="OFFLINE").\
|
|
first()
|
|
|
|
|
|
class LoadBalancer(DeclarativeBase):
|
|
"""load balancer model"""
|
|
__tablename__ = 'loadbalancers'
|
|
#column definitions
|
|
algorithm = Column(u'algorithm', VARCHAR(length=80), nullable=False)
|
|
created = Column(u'created', TIMESTAMP(), nullable=False)
|
|
errmsg = Column(u'errmsg', VARCHAR(length=128))
|
|
id = Column(u'id', BIGINT(), primary_key=True, nullable=False)
|
|
name = Column(u'name', VARCHAR(length=128), nullable=False)
|
|
port = Column(u'port', INTEGER(), nullable=False)
|
|
protocol = Column(u'protocol', VARCHAR(length=128), nullable=False)
|
|
status = Column(u'status', VARCHAR(length=50), nullable=False)
|
|
tenantid = Column(u'tenantid', VARCHAR(length=128), nullable=False)
|
|
updated = Column(u'updated', TIMESTAMP(), nullable=False)
|
|
nodes = relationship(
|
|
'Node', backref=backref('loadbalancers', order_by='Node.id')
|
|
)
|
|
devices = relationship(
|
|
'Device', secondary=loadbalancers_devices, backref='loadbalancers',
|
|
lazy='joined'
|
|
)
|
|
|
|
|
|
class Node(DeclarativeBase):
|
|
"""node model"""
|
|
__tablename__ = 'nodes'
|
|
#column definitions
|
|
address = Column(u'address', VARCHAR(length=128), nullable=False)
|
|
enabled = Column(u'enabled', Integer(), nullable=False)
|
|
id = Column(u'id', BIGINT(), primary_key=True, nullable=False)
|
|
lbid = Column(
|
|
u'lbid', BIGINT(), ForeignKey('loadbalancers.id'), nullable=False
|
|
)
|
|
port = Column(u'port', INTEGER(), nullable=False)
|
|
status = Column(u'status', VARCHAR(length=128), nullable=False)
|
|
weight = Column(u'weight', INTEGER(), nullable=False)
|
|
|
|
"""session"""
|
|
session = sessionmaker(bind=engine)()
|