Map excercise

Author: alex-solovyev

maps/maps.py

@@ -4,28 +4,35 @@
 class MapExercise:
     @staticmethod
     def rating(list_of_movies: list[dict]) -> float:
-        """
-        !!Задание нужно решить используя map!!
-        Посчитать средний рейтинг фильмов (rating_kinopoisk) у которых две или больше стран.
-        Фильмы у которых рейтинг не задан или равен 0 не учитывать в расчете среднего.
+        MIN_NUMBER_OF_COUNTRIES = 2
 
-        :param list_of_movies: Список фильмов.
-        Ключи словаря: name, rating_kinopoisk, rating_imdb, genres, year, access_level, country
-        :return: Средний рейтинг фильмов у которых две или больше стран
-        """
-        pass
+        def is_suitable_movie(movie: dict) -> bool:
+            if len(movie["country"].split(",")) < MIN_NUMBER_OF_COUNTRIES:
+                return False
+            if movie["rating_kinopoisk"] == "" or float(movie["rating_kinopoisk"]) == 0:
+                return False
+            return True
+
+        ratings = list(
+            map(
+                lambda movie: float(movie["rating_kinopoisk"]),
+                filter(is_suitable_movie, list_of_movies),
+            )
+        )
+
+        return sum(ratings) / len(ratings)
 
     @staticmethod
     def chars_count(list_of_movies: list[dict], rating: Union[float, int]) -> int:
-        """
-        !!Задание нужно решить используя map!!
-        Посчитать количество букв 'и' в названиях всех фильмов с рейтингом (rating_kinopoisk) больше
-        или равным заданному значению
+        CHAR = "и"
 
-        :param list_of_movies: Список фильмов
-        Ключи словаря: name, rating_kinopoisk, rating_imdb, genres, year, access_level, country
-        :param rating: Заданный рейтинг
-        :return: Количество букв 'и' в названиях всех фильмов с рейтингом больше
-        или равным заданному значению
-        """
-        pass
+        return sum(
+            map(
+                lambda movie: movie["name"].count(CHAR),
+                filter(
+                    lambda movie: movie["rating_kinopoisk"]
+                    and float(movie["rating_kinopoisk"]) >= rating,
+                    list_of_movies,
+                ),
+            )
+        )

maps/test_maps.py

@@ -14,12 +14,10 @@ def list_of_movies(self) -> list[dict]:
             list_of_movies = list(DictReader(movies))
         return list_of_movies
 
-    @pytest.mark.skip(reason="MapExercise.rating is not implemented")
     def test_rating(self, list_of_movies: list[dict]) -> None:
         average_rating = MapExercise.rating(list_of_movies)
         assert round(average_rating, 15) == 6.809410385259628
 
-    @pytest.mark.skip(reason="MapExercise.chars_count is not implemented")
     def test_chars_count(self, list_of_movies: list[dict]) -> None:
         chars_count = MapExercise.chars_count(list_of_movies, 5)
         assert chars_count == 3850