fsfwgen/fsfwgen/parserbase/file_list_parser.py

"""Generic File Parser class
Used by parse header files. Implemented as class in case header parser becomes more complex
"""
import os
import re
from typing import Union

from fsfwgen.core import get_console_logger

LOGGER = get_console_logger()


# pylint: disable=too-few-public-methods
class FileListParser:
    """Generic header parser which takes a directory name or directory name list
    and parses all included header files recursively.
    TODO: Filter functionality for each directory to filter out files or folders
    """

    def __init__(self, directory_list_or_name: Union[str, list]):
        self.directory_list = []
        if isinstance(directory_list_or_name, str):
            self.directory_list.append(directory_list_or_name)
        elif isinstance(directory_list_or_name, list):
            self.directory_list.extend(directory_list_or_name)
        else:
            LOGGER.warning(
                "Header Parser: Passed directory list is not a header name or list of header names"
            )
        self.header_files = []

    def parse_header_files(
        self,
        search_recursively: bool = False,
        printout_string: str = "Parsing header files: ",
        print_current_dir: bool = False,
    ):
        """This function is called to get a list of header files
        :param search_recursively:
        :param printout_string:
        :param print_current_dir:
        :return:
        """
        print(printout_string, end="")
        for directory in self.directory_list:
            self.__get_header_file_list(
                directory, search_recursively, print_current_dir
            )
        print(str(len(self.header_files)) + " header files were found.")
        # g.PP.pprint(self.header_files)
        return self.header_files

    def __get_header_file_list(
        self,
        base_directory: str,
        seach_recursively: bool = False,
        print_current_dir: bool = False,
    ):
        if base_directory[-1] != "/":
            base_directory += "/"
        local_header_files = []
        if print_current_dir:
            print("Parsing header files in: " + base_directory)
        base_list = os.listdir(base_directory)
        # g.PP.pprint(base_list)
        for entry in base_list:
            header_file_match = re.match(r"[_.]*.*\.h", entry)
            if header_file_match:
                if os.path.isfile(base_directory + entry):
                    match_string = header_file_match.group(0)
                    if match_string[0] == "." or match_string[0] == "_":
                        pass
                    else:
                        local_header_files.append(base_directory + entry)
            if seach_recursively:
                next_path = base_directory + entry
                if os.path.isdir(next_path):
                    self.__get_header_file_list(next_path, seach_recursively)
        # print("Files found in: " + base_directory)
        # g.PP.pprint(local_header_files)
        self.header_files.extend(local_header_files)