Add adding parameter to is_movie

2014-11-09 14:10:23 +01:00
parent a12f049d14
commit 1087eb3a06
1 changed files with 5 additions and 2 deletions
@@ -69,12 +69,15 @@ class CouchPotatoApi(MovieProvider):
        name_enc = base64.b64encode(ss(name))
        return self.getJsonData(self.urls['validate'] % name_enc, headers = self.getRequestHeaders())

-    def isMovie(self, identifier = None):
+    def isMovie(self, identifier = None, adding = False):

        if not identifier:
            return

-        data = self.getJsonData(self.urls['is_movie'] % identifier, headers = self.getRequestHeaders())
+        url = self.urls['is_movie'] % identifier
+        url += '?adding=1' if adding else ''
+
+        data = self.getJsonData(url, headers = self.getRequestHeaders())
        if data:
            return data.get('is_movie', True)