]> git.lizzy.rs Git - nhentai.git/blobdiff - nhentai/parser.py
fix timeout
[nhentai.git] / nhentai / parser.py
index 02f976682bd1367ee70d033ee7b86860ff616522..deb40e2497f3496730fbcc27bb6be6ca25f4b7db 100644 (file)
@@ -1,5 +1,5 @@
 # coding: utf-8
-from __future__ import print_function
+from __future__ import unicode_literals, print_function
 
 from bs4 import BeautifulSoup
 import re
@@ -14,7 +14,7 @@ def request(method, url, **kwargs):
     if not hasattr(requests, method):
         raise AttributeError('\'requests\' object has no attribute \'{0}\''.format(method))
 
-    return requests.__dict__[method](url, proxies=constant.PROXY, **kwargs)
+    return requests.__dict__[method](url, proxies=constant.PROXY, verify=False, **kwargs)
 
 
 def doujinshi_parser(id_):
@@ -43,7 +43,7 @@ def doujinshi_parser(id_):
     doujinshi['subtitle'] = subtitle.text if subtitle else ''
 
     doujinshi_cover = html.find('div', attrs={'id': 'cover'})
-    img_id = re.search('/galleries/([\d]+)/cover\.(jpg|png)$', doujinshi_cover.a.img['src'])
+    img_id = re.search('/galleries/([\d]+)/cover\.(jpg|png)$', doujinshi_cover.a.img.attrs['data-src'])
     if not img_id:
         logger.critical('Tried yo get image id failed')
         exit(1)
@@ -99,7 +99,7 @@ def search_parser(keyword, page):
 def print_doujinshi(doujinshi_list):
     if not doujinshi_list:
         return
-    doujinshi_list = [i.values() for i in doujinshi_list]
+    doujinshi_list = [(i['id'], i['title']) for i in doujinshi_list]
     headers = ['id', 'doujinshi']
     logger.info('Search Result\n' +
                 tabulate(tabular_data=doujinshi_list, headers=headers, tablefmt='rst'))