rocky-mirror.py

rocky-mirror.py

#!/usr/bin/env python3

import glob
import os

try:
    import in_place
except ImportError:
    print("Trying to Install required module: in-place")
    os.system('python3 -m pip install in-place')

BASE_DIR = "/etc/yum.repos.d/"
PARAM_COUNTRY = "&country=CN"

repo_files = glob.glob(f"{BASE_DIR}/*repo")

for repo in repo_files:
    with in_place.InPlace(repo, backup_ext='.bak') as fp:
        for line in fp:
            if line.startswith("mirrorlist=") and ("mirrors.rockylinux.org/mirrorlist" in line) and (not line.strip().endswith(PARAM_COUNTRY)):
                fp.write(line.strip() + PARAM_COUNTRY + "\n")
            else:
                fp.write(line)