Source code for s2d2.handler.files

#! /usr/bin/env python3
# -*- coding: utf-8 -*-

import os
import re



def get_list_files(dat_path, ending):
    """ generate a list with files, with a given ending/extension

    Parameters
    ----------
    dat_path : string
        path of interest.
    ending : string
        ending of a file.

    Returns
    -------
    list
        collection of file names.
    """
    if not os.path.isdir(dat_path):
        dat_path = os.path.dirname(dat_path)
    file_list = [x for x in os.listdir(dat_path) if x.endswith(ending)]
    return file_list