One Hat Cyber Team

Your IP: 216.73.216.30
Server IP: 45.79.8.107
Server: Linux localhost 5.15.0-140-generic #150-Ubuntu SMP Sat Apr 12 06:00:09 UTC 2025 x86_64
Server Software: nginx/1.18.0
PHP Version: 8.1.2-1ubuntu2.21
Buat File | Buat Folder

Dir : ~/lib/python3/dist-packages/uaclient/timer/

View File Name : update_contract_info.py

import logging

from uaclient import contract, messages, system, util
from uaclient.api.u.pro.detach.v1 import detach
from uaclient.api.u.pro.status.is_attached.v1 import _is_attached
from uaclient.config import UAConfig
from uaclient.files import machine_token, notices
from uaclient.files.notices import Notice

LOG = logging.getLogger(util.replace_top_level_logger_name(__name__))


def update_contract_info(cfg: UAConfig) -> bool:
    if _is_attached(cfg).is_attached:
        try:
            if contract.is_contract_changed(cfg):
                notices.add(
                    Notice.CONTRACT_REFRESH_WARNING,
                )
            else:
                notices.remove(
                    Notice.CONTRACT_REFRESH_WARNING,
                )
        except Exception as e:
            LOG.warning(
                "Failed to check for change in machine contract. Reason: %s",
                str(e),
                exc_info=e,
            )
            return False
    return True


def validate_release_series(cfg: UAConfig, show_message=False):
    if not _is_attached(cfg).is_attached:
        return
    machine_token_file = machine_token.get_machine_token_file(cfg)
    current_series = system.get_release_info().series
    only_series = (
        machine_token_file.entitlements()
        .get("support", {})
        .get("entitlement", {})
        .get("affordances", {})
        .get("onlySeries", None)
    )
    if only_series and only_series != current_series:
        detach()
        allowed_release = system.get_distro_info(only_series)
        message = messages.PRO_ONLY_ALLOWED_FOR_RELEASE.format(
            release=allowed_release.release,
            series_codename=allowed_release.series_codename,
        )
        if show_message:
            print(message)
        LOG.warning(message)