import os
import sys
import itertools

def get_extension(filename):
    """ Вернуть расширение файла """
    return os.path.splitext(filename)[1]


def is_right_ext(extensions):

    def function(filenames):
        ext = get_extension(filename)
        return ext in extensions
    return function


def with_extensions(extensions, filenames):
    """
    Оставить из итератора ``filenames`` только
    имена файлов, у которых расширение - одно из ``extensions``.
    """
    ext = get_extension()
    filter(is_right_ext(extensions),filenames)


def project_stats(path, extensions):
    """
    Вернуть число строк в исходниках проекта.

    Файлами, входящими в проект, считаются все файлы
    в папке ``path`` (и подпапках), имеющие расширение
    из множества ``extensions``.
    """
    pass


def total_number_of_lines(filenames):
    """
    Вернуть общее число строк в файлах ``filenames``.
    """
    pass


def number_of_lines(filename):
    """
    Вернуть число строк в файле.
    """
    pass


def iter_filenames(path):
    """
    Итератор по именам файлов в дереве.
    """
    pass




def print_usage():
    print("Usage: python project_sourse_stats_3.py <project_path>")


if __name__ == '__main__':
    if len(sys.argv) != 2:
        print_usage()
        sys.exit(1)

    project_path = sys.argv[1]
    print(project_stats(project_path, {'.cs'}))