125 lines
2.7 KiB
Python
125 lines
2.7 KiB
Python
"""
|
|
.. module: lemur.elbs.service
|
|
:platform: Unix
|
|
:copyright: (c) 2015 by Netflix Inc., see AUTHORS for more
|
|
:license: Apache, see LICENSE for more details.
|
|
|
|
.. moduleauthor:: Kevin Glisson <kglisson@netflix.com>
|
|
|
|
"""
|
|
from sqlalchemy import func
|
|
from sqlalchemy.sql import and_
|
|
|
|
from lemur import database
|
|
from lemur.elbs.models import ELB
|
|
from lemur.listeners.models import Listener
|
|
|
|
|
|
def get_all(account_id, elb_name):
|
|
"""
|
|
Retrieves all ELBs in a given account
|
|
|
|
:param account_id:
|
|
:param elb_name:
|
|
:rtype : Elb
|
|
:return:
|
|
"""
|
|
query = database.session_query(ELB)
|
|
return query.filter(and_(ELB.name == elb_name, ELB.account_id == account_id)).all()
|
|
|
|
|
|
def get_by_region_and_account(region, account_id):
|
|
query = database.session_query(ELB)
|
|
return query.filter(and_(ELB.region == region, ELB.account_id == account_id)).all()
|
|
|
|
|
|
def get_all_elbs():
|
|
"""
|
|
Get all ELBs that Lemur knows about
|
|
|
|
:rtype : list
|
|
:return:
|
|
"""
|
|
return ELB.query.all()
|
|
|
|
|
|
def get(elb_id):
|
|
"""
|
|
Retrieve an ELB with a give ID
|
|
|
|
:rtype : Elb
|
|
:param elb_id:
|
|
:return:
|
|
"""
|
|
return database.get(ELB, elb_id)
|
|
|
|
|
|
def create(account, elb):
|
|
"""
|
|
Create a new ELB
|
|
|
|
:param account:
|
|
:param elb:
|
|
"""
|
|
elb = ELB(elb)
|
|
account.elbs.append(elb)
|
|
database.create(elb)
|
|
|
|
|
|
def delete(elb_id):
|
|
"""
|
|
Delete an ELB
|
|
|
|
:param elb_id:
|
|
"""
|
|
database.delete(get(elb_id))
|
|
|
|
|
|
def render(args):
|
|
query = database.session_query(ELB)
|
|
|
|
sort_by = args.pop('sort_by')
|
|
sort_dir = args.pop('sort_dir')
|
|
page = args.pop('page')
|
|
count = args.pop('count')
|
|
filt = args.pop('filter')
|
|
active = args.pop('active')
|
|
certificate_id = args.pop('certificate_id')
|
|
|
|
if certificate_id:
|
|
query.filter(ELB.listeners.any(Listener.certificate_id == certificate_id))
|
|
|
|
if active == 'true':
|
|
query = query.filter(ELB.listeners.any())
|
|
|
|
if filt:
|
|
terms = filt.split(';')
|
|
query = database.filter(query, ELB, terms)
|
|
|
|
query = database.find_all(query, ELB, args)
|
|
|
|
if sort_by and sort_dir:
|
|
query = database.sort(query, ELB, sort_by, sort_dir)
|
|
|
|
return database.paginate(query, page, count)
|
|
|
|
|
|
def stats(**kwargs):
|
|
attr = getattr(ELB, kwargs.get('metric'))
|
|
query = database.db.session.query(attr, func.count(attr))
|
|
|
|
if kwargs.get('account_id'):
|
|
query = query.filter(ELB.account_id == kwargs.get('account_id'))
|
|
|
|
if kwargs.get('active') == 'true':
|
|
query = query.join(ELB.listeners)
|
|
query = query.filter(Listener.certificate_id != None) # noqa
|
|
|
|
items = query.group_by(attr).all()
|
|
|
|
results = []
|
|
for key, count in items:
|
|
if key:
|
|
results.append({"key": key, "y": count})
|
|
return results
|