Merge pull request #33 from dbatten5/cover-photo

Cover photo

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "phylm"
-version = "4.0.5"
+version = "4.1.0"
 description = "Phylm"
 authors = ["Dom Batten <[email protected]>"]
 license = "MIT"

src/phylm/tools.py

@@ -19,4 +19,13 @@ def search_movies(query: str) -> List[Dict[str, Union[str, int]]]:
     """
     results: List[Movie] = ia.search_movie(query)
 
-    return [{**r.data, "imdb_id": r.movieID} for r in results]
+    return [
+        {
+            "title": r.data.get("title"),
+            "kind": r.data.get("kind"),
+            "year": r.data.get("year"),
+            "imdb_id": r.movieID,
+            "cover_photo": r.data.get("cover url"),
+        }
+        for r in results
+    ]

tests/unit/tools/test_tools.py

@@ -39,4 +39,4 @@ def test_results(self, mock_ia: MagicMock, the_matrix: List[Movie]) -> None:
         assert result[0]["imdb_id"] == "0133093"
         assert result[0]["title"] == "The Matrix"
         assert result[0]["year"] == 1999
-        assert result[0]["cover url"]
+        assert result[0]["cover_photo"]