Merge pull request #4 from richardARPANET/py2-fix

python 2 setup.py compat fix

HISTORY.rst

@@ -6,7 +6,9 @@ Release History
 5.4.4 (unreleased)
 ++++++++++++++++++
 
-- Nothing changed yet.
+- Python 2.x setup.py bugfix.
+- Bugfix for ``title_exists`` method returning ``None``.
+- Bugfix for ``get_title`` raising an incorrect exception when redirection title.
 
 
 5.4.3 (2018-04-05)

setup.py

@@ -4,6 +4,7 @@
 from __future__ import absolute_import
 
 import os
+from codecs import open
 
 from setuptools import find_packages, setup
 

src/imdbpie/imdbpie.py

@@ -103,9 +103,9 @@ class Imdb(Auth):
 
     def title_exists(self, imdb_id):
         self.validate_imdb_id(imdb_id)
-        page_url = 'http://www.imdb.com/title/{0}/'.format(imdb_id)
+        page_url = 'https://www.imdb.com/title/{0}/'.format(imdb_id)
 
-        response = self.session.head(page_url)
+        response = self.session.get(page_url, allow_redirects=False)
 
         if response.status_code == httplib.OK:
             return True
@@ -298,8 +298,8 @@ class Imdb(Auth):
 
     def is_redirection_title(self, imdb_id):
         self.validate_imdb_id(imdb_id)
-        page_url = 'http://www.imdb.com/title/{0}/'.format(imdb_id)
-        response = self.session.head(page_url)
+        page_url = 'https://www.imdb.com/title/{0}/'.format(imdb_id)
+        response = self.session.get(page_url, allow_redirects=False)
         if response.status_code == httplib.MOVED_PERMANENTLY:
             return True
         else: