86 lines
3 KiB
Python
Executable file
86 lines
3 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
import os
|
|
import os.path
|
|
import json
|
|
import argparse
|
|
import dnf
|
|
import createrepo_c as cr
|
|
from common import *
|
|
|
|
# Source packages we do not ship or are rocky branded
|
|
IGNORES = [
|
|
'insights-client',
|
|
'redhat-cloud-client-configuration',
|
|
'rhc',
|
|
'rhc-worker-playbook',
|
|
'centos-indexhtml',
|
|
'centos-logos',
|
|
'centos-stream-release',
|
|
'redhat-indexhtml',
|
|
'redhat-logos',
|
|
'redhat-release'
|
|
]
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--version', type=str, required=True)
|
|
parser.add_argument("--pungi", help="local pungi is here", action='store_true')
|
|
parser.add_argument('--stream', action='store_true', help="Stream koji only")
|
|
parser.add_argument('--all', action='store_true', help="All repo")
|
|
|
|
results = parser.parse_args()
|
|
|
|
REPOS = switcher.rlver(results.version,
|
|
stream=results.stream, all_repo=results.all)
|
|
|
|
if results.pungi:
|
|
APPEND_TO_PATH = '/os'
|
|
else:
|
|
APPEND_TO_PATH = ''
|
|
|
|
def warningcb(warning_type, message):
|
|
print("WARNING: %s" % message)
|
|
return True
|
|
|
|
repo_prepop = {}
|
|
for k in REPOS:
|
|
repo_prepop[k] = {}
|
|
for arch in REPOS[k]:
|
|
PRIMARY_XML_PATH = None
|
|
FILELISTS_XML_PATH = None
|
|
OTHER_XML_PATH = None
|
|
REPO_PATH = k + '/' + arch + APPEND_TO_PATH
|
|
repomd = cr.Repomd()
|
|
cr.xml_parse_repomd(os.path.join(REPO_PATH, "repodata/repomd.xml"), repomd, warningcb)
|
|
for record in repomd.records:
|
|
if record.type == "primary":
|
|
PRIMARY_XML_PATH = os.path.join(REPO_PATH, record.location_href)
|
|
elif record.type == "filelists":
|
|
FILELISTS_XML_PATH = os.path.join(REPO_PATH, record.location_href)
|
|
elif record.type == "other":
|
|
OTHER_XML_PATH = os.path.join(REPO_PATH, record.location_href)
|
|
|
|
package_iterator = cr.PackageIterator(primary_path=PRIMARY_XML_PATH, filelists_path=FILELISTS_XML_PATH, other_path=OTHER_XML_PATH, warningcb=warningcb)
|
|
repo_prepop[k][arch] = {}
|
|
for pkg in package_iterator:
|
|
name = pkg.name + '.' + pkg.arch
|
|
subject = dnf.subject.Subject(pkg.rpm_sourcerpm)
|
|
possible_nevra = subject.get_nevra_possibilities()
|
|
srcname = possible_nevra[0].name
|
|
# Ignore packages (by source) that we do not ship
|
|
if srcname in IGNORES:
|
|
continue
|
|
|
|
# Create the initial list if the package (by source) does not exist
|
|
if srcname not in repo_prepop[k][arch]:
|
|
repo_prepop[k][arch][srcname] = []
|
|
|
|
# Avoids duplicate entries - This is especially helpful for modules
|
|
if name not in repo_prepop[k][arch][srcname]:
|
|
repo_prepop[k][arch][srcname].append(name)
|
|
|
|
# Sorts the list items
|
|
repo_prepop[k][arch][srcname].sort()
|
|
|
|
entry_point = open('/tmp/prepopulate.json', 'w+')
|
|
json.dump(repo_prepop, entry_point, indent=2, sort_keys=True)
|
|
entry_point.close()
|