fix unicode error on windows / python2

This commit is contained in:
Ricter Z 2018-08-12 23:11:01 +08:00
parent 16e8ce6f45
commit ef36e012ce
3 changed files with 12 additions and 5 deletions

View File

@ -13,8 +13,12 @@ from nhentai.utils import urlparse, generate_html
from nhentai.logger import logger
try:
reload(sys)
sys.setdefaultencoding(sys.stdin.encoding)
if sys.version_info < (3, 0, 0):
import codecs
import locale
sys.stdout = codecs.getwriter(locale.getpreferredencoding())(sys.stdout)
sys.stderr = codecs.getwriter(locale.getpreferredencoding())(sys.stderr)
except NameError:
# python3
pass

View File

@ -104,7 +104,10 @@ class ColorizingStreamHandler(logging.StreamHandler):
text = parts.pop(0)
if text:
write(text)
if sys.version_info < (3, 0, 0):
write(text.encode('utf-8'))
else:
write(text)
if parts:
params = parts.pop(0)

View File

@ -157,8 +157,8 @@ def print_doujinshi(doujinshi_list):
return
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'))
data = tabulate(tabular_data=doujinshi_list, headers=headers, tablefmt='rst')
logger.info('Search Result\n{}'.format(data))
def tag_parser(tag_id):