Include normalize function

php-tmdb · Aug 27, 2021 · 322a2a0 · 322a2a0
1 parent ddae50d
commit 322a2a0
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/lib/Tmdb/Model/Query/Discover/DiscoverTvQuery.php b/lib/Tmdb/Model/Query/Discover/DiscoverTvQuery.php
@@ -15,6 +15,7 @@
 namespace Tmdb\Model\Query\Discover;
 
 use DateTime;
+use Tmdb\Model\AbstractModel;
 use Tmdb\Model\Collection\QueryParametersCollection;
 
 /**
@@ -314,4 +315,19 @@ public function withNetworksAnd(array $networks = [])
             implode(',', $networks)
         );
     }
+
+    /**
+     * Extract object id's if an collection was passed on.
+     *
+     * @param $mixed
+     * @return mixed
+     */
+    protected function normalize($mixed)
+    {
+        if (is_object($mixed) && $mixed instanceof AbstractModel) {
+            return $mixed->getId();
+        }
+
+        return $mixed;
+    }
 }