Merge pull request #21 from mentaterasmus/master

[nhentai.git] / nhentai / constant.py

diff --git a/nhentai/constant.py b/nhentai/constant.py
index 54393841404cfda35af6ea70ccbc1e3606888d7b..55e2fa4ef5850796f67a6b11c3a58137de27f501 100644 (file)
--- a/nhentai/constant.py
+++ b/nhentai/constant.py
@@ -1,6 +1,16 @@
-SCHEMA = 'http://'
-URL = '%snhentai.net' % SCHEMA
-DETAIL_URL = '%s/g' % URL
-IMAGE_URL = '%si.nhentai.net/galleries' % SCHEMA
-SEARCH_URL = '%s/search/' % URL
+# coding: utf-8
+from __future__ import unicode_literals, print_function
+import os
+from nhentai.utils import urlparse
+
+BASE_URL = os.getenv('NHENTAI', 'https://nhentai.net')
+
+DETAIL_URL = '%s/api/gallery' % BASE_URL
+SEARCH_URL = '%s/api/galleries/search' % BASE_URL
+LOGIN_URL = '%s/login/' % BASE_URL
+FAV_URL = '%s/favorites/' % BASE_URL
+
+u = urlparse(BASE_URL)
+IMAGE_URL = '%s://i.%s/galleries' % (u.scheme, u.hostname)
+
 PROXY = {}