calculate-utils-4-lib/calculate/utils/gentoo.py

import os
import re
from calculate.utils.files import read_file_lines


class ProfileWalker:
    '''Объект обходящий все директории профиля через parent файлы.'''
    def __init__(self, filename, repositories):
        self.repositories = repositories
        self.filename = filename
        self.re_reppath = re.compile("^({0})+:".format(
                                           "|".join(self.repositories.keys())))

    def interpolate(self, path):
        def subfunc(m):
            return "{0}/profiles/".format(self.repositories.get(m.group(1)))
        return self.re_reppath.sub(subfunc, path)

    def find(self, directory):
        '''Метод для поиска по профилю всех файлов с именем, указанным в
        self.filename.'''
        parent_file_path = os.path.join(directory, "parent")
        for line in (parent_line for parent_line in
                     read_file_lines(parent_file_path) if parent_line.strip()):
            parent_directory = os.path.normpath(
                               os.path.join(directory, self.interpolate(line)))
            if os.path.exists(parent_directory):
                yield from self.find(parent_directory)
        findfile = os.path.normpath(os.path.join(directory, self.filename))
        if os.path.exists(findfile):
            yield findfile