Minor fixes

kannibalox · Sep 12, 2018 · e618880 · e618880
1 parent c7e39b9
commit e618880
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 7 deletions.
diff --git a/docs/index.rst b/docs/index.rst
@@ -12,17 +12,20 @@ Welcome to PTPAPI's documentation!
 
 .. automodule:: ptpapi.api
                 :members:
-
+
+.. automodule:: ptpapi.user
+                :members:
+
 .. automodule:: ptpapi.movie
                 :members:
 
 .. automodule:: ptpapi.torrent
                 :members:
 
-.. automodule:: ptpapi.user
+.. automodule:: ptpapi.session
                 :members:
-                
-                    
+
+
 Indices and tables
 ==================
 

diff --git a/src/ptpapi/api.py b/src/ptpapi/api.py
@@ -2,7 +2,6 @@
 """The entrypoint module for access the API"""
 import re
 import os
-import json
 import pickle
 import logging
 
@@ -108,7 +107,7 @@ def search(self, filters):
 
     def need_for_seed(self, filters={}):
         """List torrents that need seeding"""
-        data = util.snarf_cover_view_data(session.base_get("needforseed.php", params=filters).content)
+        data = ptpapi.util.snarf_cover_view_data(session.base_get("needforseed.php", params=filters).content)
         torrents = []
         for m in data:
             torrent = m['GroupingQualities'][0]['Torrents'][0]

diff --git a/src/ptpapi/util.py b/src/ptpapi/util.py
@@ -1,4 +1,8 @@
+import re
+import json
+
 from bs4 import BeautifulSoup as bs4
+from six.moves import html_parser
 
 def raise_for_cloudflare(text):
     """Raises an exception if a CloudFlare error page is detected
@@ -19,7 +23,7 @@ def snarf_cover_view_data(text):
     for json_data in re.finditer(r'coverViewJsonData\[\s*\d+\s*\]\s*=\s*({.*});', text):
         data.extend(json.loads(json_data.group(1))['Movies'])
         for movie in data:
-            movie['Title'] = HTMLParser.HTMLParser().unescape(movie['Title'])
+            movie['Title'] = html_parser.HTMLParser().unescape(movie['Title'])
             movie['Torrents'] = []
             for group in movie['GroupingQualities']:
                 for torrent in group['Torrents']: