tmdbId is available in movie data

danielx · Oct 20, 2024 · 2d9b67a · 2d9b67a
1 parent 8c9dc9f
commit 2d9b67a
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 9 deletions.
diff --git a/app/api_v1/endpoints/radarr.py b/app/api_v1/endpoints/radarr.py
@@ -29,7 +29,7 @@ async def radarr(event: schemas.RadarrEvent):
         MESSAGE.substitute(
             title=event.movie.title.strip(),
             year=event.movie.year,
-            id=event.remoteMovie.tmdbId,
+            id=event.movie.tmdbId,
         )
     )
 

diff --git a/app/schemas/_events.py b/app/schemas/_events.py
@@ -4,17 +4,13 @@
 class Movie(BaseModel):
     title: str
     year: int
-
-
-class RemoteMovie(BaseModel):
     tmdbId: int
 
 
 class RadarrEvent(BaseModel):
     eventType: str
 
     movie: Movie
-    remoteMovie: RemoteMovie
 
 
 class Series(BaseModel):

diff --git a/tests/test_radarr.py b/tests/test_radarr.py
@@ -23,8 +23,7 @@ class Payload(NamedTuple):
     )
     json: dict | None = {
         "eventType": "Grab",
-        "movie": {"title": "string", "year": 0},
-        "remoteMovie": {"tmdbId": 0},
+        "movie": {"title": "string", "year": 0, "tmdbId": 0},
     }
 
 
@@ -48,8 +47,7 @@ class Result(NamedTuple):
             Payload(
                 json={
                     "eventType": "Download",
-                    "movie": {"title": "string", "year": 0},
-                    "remoteMovie": {"tmdbId": 0},
+                    "movie": {"title": "string", "year": 0, "tmdbId": 0},
                 }
             ),
             Result(