68 lines
1.6 KiB
Python
68 lines
1.6 KiB
Python
from argparse import ArgumentParser
|
|
|
|
import os
|
|
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.stem,
|
|
path=path,
|
|
) for path in top_dir.rglob('*.rpm')]
|
|
|
|
|
|
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()
|