Ejemplo n.º 1
0
def _get_switch_machine(switch_id, machine_id, session=None, **kwargs):
    """Get switch machine by switch id and machine id."""
    switch = _get_switch(switch_id, session=session)
    from compass.db.api import machine as machine_api
    machine = machine_api.get_machine_internal(machine_id, session=session)
    return utils.get_db_object(
        session, models.SwitchMachine,
        switch_id=switch.id, machine_id=machine.id, **kwargs
    )
Ejemplo n.º 2
0
def _get_switch_machine(switch_id, machine_id, session=None, **kwargs):
    """Get switch machine by switch id and machine id."""
    switch = _get_switch(switch_id, session=session)
    from compass.db.api import machine as machine_api
    machine = machine_api.get_machine_internal(machine_id, session=session)
    return utils.get_db_object(
        session, models.SwitchMachine,
        switch_id=switch.id, machine_id=machine.id, **kwargs
    )
Ejemplo n.º 3
0
def _add_machine_to_switch(switch_id, machine_id, session=None, **kwargs):
    """Add machine to switch."""
    switch = _get_switch(switch_id, session=session)
    from compass.db.api import machine as machine_api
    machine = machine_api.get_machine_internal(machine_id, session=session)
    _add_switch_machine_only(switch,
                             machine,
                             False,
                             owner_id=machine.owner_id,
                             **kwargs)
Ejemplo n.º 4
0
def _add_machine_to_switch(
    switch_id, machine_id, session=None, **kwargs
):
    """Add machine to switch."""
    switch = _get_switch(switch_id, session=session)
    from compass.db.api import machine as machine_api
    machine = machine_api.get_machine_internal(
        machine_id, session=session
    )
    _add_switch_machine_only(
        switch, machine, False, **kwargs
    )
Ejemplo n.º 5
0
def get_machine_or_host(host_id,
                        exception_when_missing=True,
                        user=None,
                        session=None,
                        **kwargs):
    """get machine or host if possible."""
    from compass.db.api import machine as machine_api
    machine = machine_api.get_machine_internal(
        host_id,
        exception_when_missing=exception_when_missing,
        session=session)
    if machine.host:
        return machine.host
    else:
        return machine
Ejemplo n.º 6
0
def get_machine_or_host(
    host_id, exception_when_missing=True,
    user=None, session=None, **kwargs
):
    """get machine or host if possible."""
    from compass.db.api import machine as machine_api
    machine = machine_api.get_machine_internal(
        host_id,
        exception_when_missing=exception_when_missing,
        session=session
    )
    if machine.host:
        return machine.host
    else:
        return machine