Blame libnmstate/nm/user.py

Packit Service 0535c1
#
Packit Service 0535c1
# Copyright (c) 2018-2019 Red Hat, Inc.
Packit Service 0535c1
#
Packit Service 0535c1
# This file is part of nmstate
Packit Service 0535c1
#
Packit Service 0535c1
# This program is free software: you can redistribute it and/or modify
Packit Service 0535c1
# it under the terms of the GNU Lesser General Public License as published by
Packit Service 0535c1
# the Free Software Foundation, either version 2.1 of the License, or
Packit Service 0535c1
# (at your option) any later version.
Packit Service 0535c1
#
Packit Service 0535c1
# This program is distributed in the hope that it will be useful,
Packit Service 0535c1
# but WITHOUT ANY WARRANTY; without even the implied warranty of
Packit Service 0535c1
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
Packit Service 0535c1
# GNU Lesser General Public License for more details.
Packit Service 0535c1
#
Packit Service 0535c1
# You should have received a copy of the GNU Lesser General Public License
Packit Service 0535c1
# along with this program. If not, see <https://www.gnu.org/licenses/>.
Packit Service 0535c1
#
Packit Service 0535c1
""" Use a Network Manager User Setting to store and retrieve information that
Packit Service 0535c1
does not fit somewhere else such as an interface's description.
Packit Service 0535c1
Packit Service 0535c1
https://lazka.github.io/pgi-docs/#NM-1.0/classes/SettingUser.html
Packit Service 0535c1
"""
Packit Service 0535c1
Packit Service 0535c1
from libnmstate.error import NmstateValueError
Packit Service 0535c1
from libnmstate.nm import connection as nm_connection
Packit Service 0535c1
from .common import NM
Packit Service 0535c1
Packit Service 0535c1
NMSTATE_DESCRIPTION = "nmstate.interface.description"
Packit Service 0535c1
Packit Service 0535c1
Packit Service 0535c1
def create_setting(iface_state, base_con_profile):
Packit Service 0535c1
    description = iface_state.get("description")
Packit Service 0535c1
Packit Service 0535c1
    if not description:
Packit Service 0535c1
        return None
Packit Service 0535c1
Packit Service 0535c1
    if not NM.SettingUser.check_val(description):
Packit Service 0535c1
        raise NmstateValueError("Invalid description")
Packit Service 0535c1
Packit Service 0535c1
    user_setting = None
Packit Service 0535c1
    if base_con_profile:
Packit Service 0535c1
        user_setting = base_con_profile.get_setting_by_name(
Packit Service 0535c1
            NM.SETTING_USER_SETTING_NAME
Packit Service 0535c1
        )
Packit Service 0535c1
        if user_setting:
Packit Service 0535c1
            user_setting = user_setting.duplicate()
Packit Service 0535c1
Packit Service 0535c1
    if not user_setting:
Packit Service 0535c1
        user_setting = NM.SettingUser.new()
Packit Service 0535c1
Packit Service 0535c1
    user_setting.set_data(NMSTATE_DESCRIPTION, description)
Packit Service 0535c1
    return user_setting
Packit Service 0535c1
Packit Service 0535c1
Packit Service 0535c1
def get_info(context, device):
Packit Service 0535c1
    """
Packit Service 0535c1
    Get description from user settings for a connection
Packit Service 0535c1
    """
Packit Service 0535c1
    info = {}
Packit Service 0535c1
Packit Service 0535c1
    connection = nm_connection.ConnectionProfile(context)
Packit Service 0535c1
    connection.import_by_device(device)
Packit Service 0535c1
    if not connection.profile:
Packit Service 0535c1
        return info
Packit Service 0535c1
Packit Service 0535c1
    try:
Packit Service 0535c1
        user_setting = connection.profile.get_setting_by_name(
Packit Service 0535c1
            NM.SETTING_USER_SETTING_NAME
Packit Service 0535c1
        )
Packit Service 0535c1
        description = user_setting.get_data(NMSTATE_DESCRIPTION)
Packit Service 0535c1
        if description:
Packit Service 0535c1
            info["description"] = description
Packit Service 0535c1
    except AttributeError:
Packit Service 0535c1
        pass
Packit Service 0535c1
Packit Service 0535c1
    return info