# @author: Luiz Ozaki, Locaweb. from simplenet.common import event from simplenet.common.config import get_logger from simplenet.db.models import Switch, Interface, Router from simplenet.db import db_utils from simplenet.exceptions import ( FeatureNotAvailable, EntityNotFound, OperationNotPermited, DuplicatedEntryError ) from simplenet.network_appliances.base import SimpleNet from sqlalchemy.exc import IntegrityError logger = get_logger() session = db_utils.get_database_session() class Net(SimpleNet): def switch_list(self): return self._generic_list_("Switch") def switch_create(self, data): logger.debug("Creating device using data: %s" % data) session.begin(subtransactions=True) try: session.add(Switch(name=data['name'], mac=data['mac'], address=data['address'], model_type=data['model_type'])) session.commit() except IntegrityError:
def __init__(self): self.logger = get_logger() self.session = db_utils.get_database_session()