pungi/pungi/scripts/gather_rpms.py

from argparse import ArgumentParser

import os
from glob import iglob
from typing import List
from pathlib import Path

from dataclasses import dataclass
from productmd.common import parse_nvra


@dataclass
class Package:
    nvra: str
    path: Path


def search_rpms(top_dir: Path) -> List[Package]:
    """
    Search for all *.rpm files recursively
    in given top directory
    Returns:
        list: list of paths
    """
    return [Package(
        nvra=Path(path).stem,
        path=Path(path),
    ) for path in iglob(str(top_dir.joinpath('**/*.rpm')), recursive=True)]


def copy_rpms(packages: List[Package], target_top_dir: Path):
    """
    Search synced repos for rpms and prepare
    koji-like structure for pungi

    Instead of repos, use following structure:
        # ls /mnt/koji/
            i686/  noarch/  x86_64/
    Returns:
        Nothing:
    """
    for package in packages:
        info = parse_nvra(package.nvra)
        target_arch_dir = target_top_dir.joinpath(info['arch'])
        target_file = target_arch_dir.joinpath(package.path.name)
        os.makedirs(target_arch_dir, exist_ok=True)

        if not target_file.exists():
            try:
                os.link(package.path, target_file)
            except OSError:
                # hardlink failed, try symlinking
                package.path.symlink_to(target_file)


def cli_main():
    parser = ArgumentParser()
    parser.add_argument('-p', '--path', required=True, type=Path)
    parser.add_argument('-t', '--target', required=True, type=Path)

    namespace = parser.parse_args()

    rpms = search_rpms(namespace.path)
    copy_rpms(rpms, namespace.target)


if __name__ == '__main__':
    cli_main()