Fixing parser for nhentai site update

[nhentai.git] / setup.py

diff --git a/setup.py b/setup.py
index 59e3bbf16ac0a4162d41cb10338f2248a0d2a18a..7b87d5cc320f2ff02f860890e869baa9cf53cb3a 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -1,16 +1,20 @@
 # coding: utf-8
 from __future__ import print_function, unicode_literals
+import sys
 import codecs
 from setuptools import setup, find_packages
 from nhentai import __version__, __author__, __email__
 
+
 with open('requirements.txt') as f:
     requirements = [l for l in f.read().splitlines() if l]
 
 
 def long_description():
-    with codecs.open('README.md', 'rb') as f:
-        return f.read()
+    with codecs.open('README.rst', 'rb') as readme:
+        if not sys.version_info < (3, 0, 0):
+            return readme.read().decode('utf-8')
+
 
 setup(
     name='nhentai',