altlinux-autorepacked/autorepacked/base_provider.py

import os
import shutil
import subprocess
import tempfile
import re


def _get_file(directory):
    files_and_dirs = os.listdir(directory)
    files = [f for f in files_and_dirs if os.path.isfile(os.path.join(directory, f))]
    return os.path.join(directory, files[0]) if files else None

class BaseProvider:
    def __init__(self, config):
        self.config = config

    DOWNLOAD_URL = ''
    _name = ''

    def get_name(self):
        return self._name

    def get_version(self):
        url = subprocess.run([
            'epm',
            '--silent',
            'tool',
            'eget',
            '--get-real-url',
            self.get_download_url(),
        ], stdout=subprocess.PIPE).stdout.decode('utf-8')

        pattern = r'\d+\.\d+\.\d+'
        match = re.search(pattern, url)
        if match:
            return match.group()
        else:
            return "not found"

    def get_download_url(self):
        return self.DOWNLOAD_URL

    def download(self):
        download_directory = tempfile.mkdtemp()
        repacked_directory = tempfile.mkdtemp()

        subprocess.run(
            args=[
                'epm',
                '--silent',
                'tool',
                'eget',
                '-q',
                self.get_download_url(),
            ],
            stdout=subprocess.PIPE,
            cwd=download_directory,
        ).stdout.decode('utf-8')

        file = _get_file(download_directory)

        subprocess.run(
            args=[
                'epm',
                '--silent',
                '-y',
                'repack',
                file,
            ],
            stdout=subprocess.PIPE,
            cwd=repacked_directory,
        ).stdout.decode('utf-8')

        shutil.rmtree(download_directory)

        file = _get_file(repacked_directory)

        subprocess.run(
            args=[
                'epm',
                '--silent',
                'repo',
                'pkgadd',
                self.config.get('repo_path'),
                file,
            ],
            stdout=subprocess.PIPE,
            cwd=repacked_directory,
        ).stdout.decode('utf-8')

        shutil.rmtree(repacked_directory)