toolkit/mangle/rlutil.py
Neil Hanlon 1178e32e1c
Some checks failed
Build empanada images for imagefactory / buildx (push) Failing after 4s
Build empanada container images for lorax / buildx (push) Successful in 1s
fix: broke this by accident
2024-04-04 01:43:40 -04:00

163 lines
5.4 KiB
Python

#!/usr/bin/env python3
# This helps manage repo files and other little bits of Rocky Linux repos. Use
# at your own risk.
#
# WARNING: USING A RELEASE OLDER THAN THE CURRENT RELEASE IS NOT SUPPORTED.
# PLEASE SEE THE WIKI FOR MORE INFORMATION ON THE CURRENT RELEASES.
# https://wiki.rockylinux.org/rocky/version/
import os
import sys
import re
import argparse
import rpm
dist = rpm.expandMacro("%rocky")
regex = fr'^{dist}\.\d+$'
parser = argparse.ArgumentParser(description='Rocky Linux Repo Utility')
subparser = parser.add_subparsers(dest='cmd')
subparser.required = True
release_parser = subparser.add_parser('release', epilog='Use this to make general changes to the Rocky Linux release dnf repo configuration.')
release_parser.add_argument('--use-baseurl', action='store_true')
release_parser.add_argument('--use-mirrorlist', action='store_true')
release_parser.add_argument('--reset', action='store_true',
help='If this script was used to modify the config, this will reset everything back to the defaults.')
release_parser.add_argument('--contentdir',
choices=('pub', 'staging', 'stg', 'vault'))
release_parser.add_argument('--switch-to-vault', action='store_true', help='Switches to using the vault.')
release_parser.add_argument('--releasever', type=str, default='', help='Chooses a specific release. USE WITH CAUTION.')
results = parser.parse_args()
command = parser.parse_args().cmd
def all_rocky_files(
directory_path = '/etc/yum.repos.d',
filter_filename = lambda file: 'rocky' in file.lower()) -> list:
"""
Filter out specified files
"""
return_list = []
for file in os.scandir(directory_path):
if filter_filename(file.name):
return_list.append(os.path.join(directory_path, file.name))
return return_list
def switch_to_baseurl():
"""
Uncomments baseurl, removes mirrorlist
"""
repo_files = all_rocky_files()
print('Switching all rocky repo files to use baseurl')
for file in repo_files:
with open(file, "r") as f:
lines = f.readlines()
with open(file, "w") as f:
for line in lines:
ml_match = re.search(r'^mirrorlist', line)
bu_match = re.search(r'^#baseurl', line)
if ml_match:
f.write(re.sub(r'^mirrorlist', '#mirrorlist', line))
elif bu_match:
f.write(re.sub(r'^#baseurl', 'baseurl', line))
else:
f.write(line)
f.close()
def switch_to_mirrorlist():
"""
Uncomments mirrorlist, removes baseurl
"""
repo_files = all_rocky_files()
print('Switching all rocky repo files to use mirror list')
for file in repo_files:
with open(file, "r") as f:
lines = f.readlines()
with open(file, "w") as f:
for line in lines:
ml_match = re.search(r'^#mirrorlist', line)
bu_match = re.search(r'^baseurl', line)
if ml_match:
f.write(re.sub(r'^#mirrorlist', 'mirrorlist', line))
elif bu_match:
f.write(re.sub(r'^baseurl', '#baseurl', line))
else:
f.write(line)
f.close()
def set_releasever(releasever):
"""
Sets a release version
"""
check_for_rel = re.search(regex, releasever)
if not check_for_rel:
print(f'Release version does not conform to X.Y or it does not start with {dist}')
sys.exit(1)
print(f'Setting releasever to {releasever}')
with open('/etc/dnf/vars/releasever', 'w') as f:
f.write(releasever)
f.close()
def set_contentdir(contentdir):
"""
Sets the contentdir path to whatever/rocky
"""
print(f'Setting contentdir to {contentdir}/rocky')
with open('/etc/dnf/vars/contentdir', 'w') as f:
f.write(f'{contentdir}/rocky')
f.close()
def switch_to_vault(releasever):
"""
Switch to vault
"""
print('** Switching to use the vault')
switch_to_baseurl()
set_contentdir('vault')
set_releasever(releasever)
def full_reset():
"""
Resets everything to normal ONLY if this script was used to make changes.
"""
print('** Resetting mirrorlist')
switch_to_mirrorlist()
print('** Resetting contentdir')
set_contentdir('pub')
print('** Removing releasever var')
if os.path.exists('/etc/dnf/vars/releasever'):
os.remove('/etc/dnf/vars/releasever')
def repoutil(results):
if results.switch_to_vault:
if len(results.releasever) == 0 or not re.search(regex, results.releasever):
print(f'A release version was NOT specified nor correct for this release: {results.releasever}')
sys.exit(1)
elif re.search(regex, results.releasever):
switch_to_vault(results.releasever)
sys.exit(0)
elif results.reset:
full_reset()
else:
if results.use_baseurl and results.use_mirrorlist:
print('You cannot set both baseurl and mirrorlist.')
sys.exit(1)
if results.use_baseurl:
switch_to_baseurl()
elif results.use_mirrorlist:
switch_to_mirrorlist()
if len(results.releasever) > 0:
set_releasever(results.releasever)
if results.contentdir:
set_contentdir(results.contentdir)
def main():
if command == 'release':
repoutil(results)
if __name__ == '__main__':
main()