コード例 #1
0
    def _get_folder_files(self, folder, exclude_dotted=True):
        """
        Given a folder, recursively retrieves contained files
        returns a list of lists ([file extension, file path]).

        NB : Shall not be used independtly.

        folder                    String : path to retrieve files from.
        """
        listed_dirs = []

        for root, dirs, files in os.walk(folder):
            # If bool param is True, exclude dotted
            # dirs from computing.
            if exclude_dotted:
                dirs = utils.remove_dotted_path_elements(dirs)
                files = utils.remove_dotted_path_elements(files)
            for name in files:
                file_path = os.path.join(root, name)
                file_extension = utils.get_file_extension(file_path)

                # Stamper should not apply a header on file with
                # no language extension
                if self.license.is_valid_file_extension(file_extension):
                    listed_dirs.append([file_extension, file_path])

        return listed_dirs
コード例 #2
0
    def _get_path_elements(self, path, exclude_dotted=True):
        """
        Recursively retrieves a given path files.
        returns a list of tuples (file extension, file path).
        If given path is a folder, returns get_folder_files result,
        else uses the given path file.

        path                    String : path to retrieve files from.
        exclude_dotted          Boolean : Wheter to ignore paths beggining
                                with a dot while walking dirs.
                                Ex : .git .svn .emacs
        """
        listed_elems = []

        if os.path.isdir(path):
            listed_elems = self._get_folder_files(path, exclude_dotted)
        else:
            file_path = utils.remove_dotted_path_elements(path) if exclude_dotted \
                                                       else path
            if file_path:
                file_extension = utils.get_file_extension(file_path)
                if self.license.is_valid_file_extension(file_extension):
                    listed_elems = [[file_extension, file_path]]

        return(listed_elems)