# Copyright (C) 2018 Red Hat, Inc.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
import logging
log = logging.getLogger("pylorax")
import os
import shutil
import libdnf5 as dnf5
from libdnf5.common import QueryCmp_GLOB as GLOB
from libdnf5.common import QueryCmp_EQ as EQ
from pylorax import DEFAULT_PLATFORM_ID, DEFAULT_RELEASEVER
from pylorax.sysutils import flatconfig
def _repo_onoff(dbo, repo_id, enabled):
"""Helper function for enabling/disabling repos"""
rq = dnf5.repo.RepoQuery(dbo)
if any(g for g in ['*', '?', '[', ']'] if g in repo_id):
rq.filter_id(repo_id, GLOB)
else:
rq.filter_id(repo_id, EQ)
if len(rq) == 0:
log.warning("%s is an unknown repo, not %s it", repo_id, "enabling" if enabled else "disabling")
return
for r in rq:
if enabled:
r.enable()
log.info("Enabled repo %s", r.get_id())
else:
r.disable()
log.info("Disabled repo %s", r.get_id())
[docs]
def get_dnf_base_object(installroot, sources, mirrorlists=None, repos=None,
enablerepos=None, disablerepos=None,
tempdir="/var/tmp", proxy=None, releasever=DEFAULT_RELEASEVER,
cachedir=None, logdir=None, sslverify=True, dnfplugins=None,
basearch=None):
""" Create a dnf Base object and setup the repositories and installroot
:param string installroot: Full path to the installroot
:param list sources: List of source repo urls to use for the installation
:param list enablerepos: List of repo names to enable
:param list disablerepos: List of repo names to disable
:param list mirrorlist: List of mirrors to use
:param string tempdir: Path of temporary directory
:param string proxy: http proxy to use when fetching packages
:param string releasever: Release version to pass to dnf
:param string cachedir: Directory to use for caching packages
:param bool noverifyssl: Set to True to ignore the CA of ssl certs. eg. use self-signed ssl for https repos.
:param string basearch: Architecture to use for $basearch substitution in repo urls
If tempdir is not set /var/tmp is used.
If cachedir is None a dnf.cache directory is created inside tmpdir
If basearch is not set it uses the host system's machine type.
"""
def sanitize_repo(repo):
"""Convert bare paths to file:/// URIs, and silently reject protocols unhandled by yum"""
if repo.startswith("/"):
return "file://{0}".format(repo)
elif any(repo.startswith(p) for p in ('http://', 'https://', 'ftp://', 'file://')):
return repo
else:
return None
mirrorlists = mirrorlists or []
enablerepos = enablerepos or []
disablerepos = disablerepos or []
# sanitize the repositories
sources = list(sanitize_repo(r) for r in sources)
mirrorlists = list(sanitize_repo(r) for r in mirrorlists)
# remove invalid repositories
sources = list(r for r in sources if r)
mirrorlists = list(r for r in mirrorlists if r)
if not cachedir:
cachedir = os.path.join(tempdir, "dnf.cache")
if not os.path.isdir(cachedir):
os.mkdir(cachedir)
if not logdir:
logdir = os.path.join(tempdir, "dnf.logs")
if not os.path.isdir(logdir):
os.mkdir(logdir)
# Used for url substitutions for $basearch
if not basearch:
basearch = os.uname().machine
dnfbase = dnf5.base.Base()
# TODO Add dnf5 plugin support
# Currently the documentation is not complete enough to use plugins with dnf5
# So print a warning and carry on if any have been selected
if dnfplugins:
log.warning("plugins are not yet supported with libdnf5, skipping: %s", dnfplugins)
# Enable DNF pluings
# NOTE: These come from the HOST system's environment
# XXX - dnfbase has add_plugin and load_plugins but neither seem to provide the ability to
# enable/disable based on glob. dnfbase.setup() already calls load_plugins()
# if dnfplugins:
# if dnfplugins[0] == "*":
# # Enable them all
# dnfbase.init_plugins()
# else:
# # Only enable the listed plugins
# dnfbase.init_plugins(disabled_glob=["*"], enable_plugins=dnfplugins)
conf = dnfbase.get_config()
conf.logdir = logdir
conf.cachedir = cachedir
conf.install_weak_deps = False
conf.installroot = installroot
# Load the file lists too
conf.optional_metadata_types = ['filelists']
conf.tsflags += ("nodocs",)
# Log details about the solver
conf.debug_solver = True
if proxy:
conf.proxy = proxy
if sslverify == False:
conf.sslverify = False
if not os.path.exists("/etc/os-release"):
log.warning("/etc/os-release is missing, cannot determine platform id, falling back to %s", DEFAULT_PLATFORM_ID)
platform_id = DEFAULT_PLATFORM_ID
else:
os_release = flatconfig("/etc/os-release")
platform_id = os_release.get("PLATFORM_ID", DEFAULT_PLATFORM_ID)
log.info("Using %s for module_platform_id", platform_id)
conf.module_platform_id = platform_id
# Set variables used for substitutions
dnfbase.get_vars().set("releasever", releasever)
dnfbase.get_vars().set("basearch", basearch)
# Add .repo files
if repos:
reposdir = os.path.join(tempdir, "dnf.repos")
if not os.path.isdir(reposdir):
os.mkdir(reposdir)
for r in repos:
shutil.copy2(r, reposdir)
conf.reposdir = reposdir
dnfbase.setup()
sack = dnfbase.get_repo_sack()
# add the sources
for i, r in enumerate(sources):
if "SRPM" in r or "srpm" in r:
log.info("Skipping source repo: %s", r)
continue
repo_name = "lorax-repo-%d" % i
repo = sack.create_repo(repo_name)
rc = repo.get_config()
rc.baseurl = r
if proxy:
rc.proxy = proxy
log.info("Added '%s': %s", repo_name, r)
# add the mirrorlists
for i, r in enumerate(mirrorlists):
if "SRPM" in r or "srpm" in r:
log.info("Skipping source repo: %s", r)
continue
repo_name = "lorax-mirrorlist-%d" % i
repo = sack.create_repo(repo_name)
rc = repo.get_config()
rc.mirrorlist = r
if proxy:
rc.proxy = proxy
log.info("Added '%s': %s", repo_name, r)
if repos:
sack.create_repos_from_reposdir()
# Enable repos listed on the cmdline
for r in enablerepos:
_repo_onoff(dnfbase, r, True)
# Disable repos listed on the cmdline
for r in disablerepos:
_repo_onoff(dnfbase, r, False)
# Make sure there are enabled repos
rq = dnf5.repo.RepoQuery(dnfbase)
rq.filter_enabled(True)
if len(rq) == 0:
log.error("No enabled repos")
return None
log.info("Using repos: %s", ", ".join(r.get_id() for r in rq))
log.info("Fetching metadata...")
try:
sack.update_and_load_enabled_repos(False)
except RuntimeError as e:
log.error("Problem fetching metadata: %s", e)
return None
return dnfbase