Dzigr/python-project-50

"""Get format of file."""
from os.path import splitext


def get_format(filepath):
    """Get the file format.

    Parameters:
        filepath: path to the file.

    Returns:
        extension: string, with specified the file format
    """
    _, extension = splitext(filepath)
    if extension == '.yaml':
        return 'yaml'
    elif extension == '.yml':
        return 'yaml'
    elif extension == '.json':
        return 'json'


def read_data(filepath):
    """Read the data from file.

    Parameters:
        filepath: path to the file.

    Returns:
        data: data from read file.
    """
    with open(filepath) as read_file:
        return read_file.read()