156 lines
4.5 KiB
Python
156 lines
4.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
|
|
# 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; version 2 of the License.
|
|
#
|
|
# 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 Library 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 <https://gnu.org/licenses/>.
|
|
|
|
|
|
# TODO: remove all DNF hacks, possibly this whole file
|
|
|
|
|
|
import dnf
|
|
|
|
import dnf.conf
|
|
import dnf.repo
|
|
import dnf.sack
|
|
|
|
import pungi.arch
|
|
|
|
try:
|
|
import dnf.rpm as dnf_arch
|
|
except ImportError:
|
|
import dnf.arch as dnf_arch
|
|
|
|
|
|
class Conf(dnf.conf.Conf):
|
|
# This is only modified to get our custom Substitutions class in.
|
|
def __init__(self, arch, *args, **kwargs):
|
|
super(Conf, self).__init__(*args, **kwargs)
|
|
self.substitutions = Substitutions(arch)
|
|
|
|
|
|
class Substitutions(dict):
|
|
# DNF version of Substitutions detects host arch. We don't want that.
|
|
def __init__(self, arch):
|
|
super(Substitutions, self).__init__()
|
|
self["arch"] = arch
|
|
self["basearch"] = dnf_arch.basearch(arch)
|
|
|
|
|
|
class DnfWrapper(dnf.Base):
|
|
def __init__(self, *args, **kwargs):
|
|
super(DnfWrapper, self).__init__(*args, **kwargs)
|
|
self.arch_wrapper = ArchWrapper(self.conf.substitutions["arch"])
|
|
self.comps_wrapper = CompsWrapper(self)
|
|
|
|
def add_repo(
|
|
self, repoid, baseurl=None, enablegroups=True, lookaside=False, **kwargs
|
|
):
|
|
self.repos.add_new_repo(
|
|
repoid,
|
|
self.conf,
|
|
baseurl=[baseurl],
|
|
enabledgroups=enablegroups,
|
|
priority=10 if lookaside else 20,
|
|
module_hotfixes=True,
|
|
**kwargs
|
|
)
|
|
|
|
|
|
class CompsWrapper(object):
|
|
def __init__(self, dnf_obj):
|
|
self.dnf = dnf_obj
|
|
|
|
def __getitem__(self, name):
|
|
return self.groups[name]
|
|
|
|
@property
|
|
def comps(self):
|
|
return self.dnf.comps
|
|
|
|
@property
|
|
def groups(self):
|
|
result = {}
|
|
for i in self.comps.groups:
|
|
result[i.id] = i
|
|
return result
|
|
|
|
def get_packages_from_group(
|
|
self,
|
|
group_id,
|
|
include_default=True,
|
|
include_optional=True,
|
|
include_conditional=True,
|
|
):
|
|
packages = []
|
|
conditional = []
|
|
|
|
group = self.groups[group_id]
|
|
|
|
# add mandatory packages
|
|
packages.extend([i.name for i in group.mandatory_packages])
|
|
|
|
# add default packages
|
|
if include_default:
|
|
packages.extend([i.name for i in group.default_packages])
|
|
|
|
# add optional packages
|
|
if include_optional:
|
|
packages.extend([i.name for i in group.optional_packages])
|
|
|
|
for package in group.conditional_packages:
|
|
conditional.append({"name": package.requires, "install": package.name})
|
|
|
|
return packages, conditional
|
|
|
|
def get_comps_packages(self, groups, exclude_groups):
|
|
packages = set()
|
|
conditional = []
|
|
|
|
if isinstance(groups, list):
|
|
groups = dict([(i, 1) for i in groups])
|
|
|
|
for group_id, group_include in sorted(groups.items()):
|
|
if group_id in exclude_groups:
|
|
continue
|
|
|
|
include_default = group_include in (1, 2)
|
|
include_optional = group_include in (2,)
|
|
include_conditional = True
|
|
pkgs, cond = self.get_packages_from_group(
|
|
group_id, include_default, include_optional, include_conditional
|
|
)
|
|
packages.update(pkgs)
|
|
for i in cond:
|
|
if i not in conditional:
|
|
conditional.append(i)
|
|
return list(packages), conditional
|
|
|
|
def get_langpacks(self):
|
|
result = []
|
|
for name, install in self.comps._i.langpacks.items():
|
|
result.append({"name": name, "install": install})
|
|
return result
|
|
|
|
|
|
class ArchWrapper(object):
|
|
def __init__(self, arch):
|
|
self.base_arch = dnf_arch.basearch(arch)
|
|
self.all_arches = pungi.arch.get_valid_arches(
|
|
self.base_arch, multilib=True, add_noarch=True
|
|
)
|
|
self.native_arches = pungi.arch.get_valid_arches(
|
|
self.base_arch, multilib=False, add_noarch=True
|
|
)
|
|
self.multilib_arches = pungi.arch.get_valid_multilib_arches(self.base_arch)
|
|
self.source_arches = ["src", "nosrc"]
|