from __future__ import print_function
import os
import sys
+import json
from optparse import OptionParser
+
try:
from itertools import ifilter as filter
except ImportError:
''' % __version__)
+def load_config():
+ if not os.path.exists(constant.NHENTAI_CONFIG_FILE):
+ return
+
+ try:
+ with open(constant.NHENTAI_CONFIG_FILE, 'r') as f:
+ constant.CONFIG = json.load(f)
+ except json.JSONDecodeError:
+ logger.error('Failed to load config file.')
+ write_config()
+
+
+def write_config():
+ if not os.path.exists(constant.NHENTAI_HOME):
+ os.mkdir(constant.NHENTAI_HOME)
+
+ with open(constant.NHENTAI_CONFIG_FILE, 'w') as f:
+ f.write(json.dumps(constant.CONFIG))
+
+
def cmd_parser():
+ load_config()
+
parser = OptionParser('\n nhentai --search [keyword] --download'
'\n NHENTAI=http://h.loli.club nhentai --id [ID ...]'
'\n nhentai --file [filename]'
logger.info('Download history cleaned.')
exit(0)
- if os.path.exists(constant.NHENTAI_COOKIE):
- with open(constant.NHENTAI_COOKIE, 'r') as f:
- constant.COOKIE = f.read()
-
+ # --- set config ---
if args.cookie:
- try:
- if not os.path.exists(constant.NHENTAI_HOME):
- os.mkdir(constant.NHENTAI_HOME)
-
- with open(constant.NHENTAI_COOKIE, 'w') as f:
- f.write(args.cookie)
- except Exception as e:
- logger.error('Cannot create NHENTAI_HOME: {}'.format(str(e)))
- exit(1)
-
+ constant.CONFIG['cookie'] = args.cookie
logger.info('Cookie saved.')
+ write_config()
exit(0)
- if os.path.exists(constant.NHENTAI_LANGUAGE) and not args.language:
- with open(constant.NHENTAI_LANGUAGE, 'r') as f:
- constant.LANGUAGE = f.read()
- args.language = f.read()
-
if args.language:
- try:
- if not os.path.exists(constant.NHENTAI_HOME):
- os.mkdir(constant.NHENTAI_HOME)
-
- with open(constant.NHENTAI_LANGUAGE, 'w') as f:
- f.write(args.language)
- except Exception as e:
- logger.error('Cannot create NHENTAI_HOME: {}'.format(str(e)))
- exit(1)
-
- logger.info('Default language now is {}.'.format(args.language))
+ constant.CONFIG['language'] = args.language
+ logger.info('LANGUAGE now set to \'{0}\''.format(args.language))
+ write_config()
exit(0)
-
- if args.clean_language:
- try:
- if not os.path.exists(constant.NHENTAI_HOME):
- os.mkdir(constant.NHENTAI_HOME)
-
- with open(constant.NHENTAI_LANGUAGE, 'w') as f:
- f.close()
- except Exception as e:
- logger.error('Cannot create NHENTAI_HOME: {}'.format(str(e)))
- exit(1)
-
- logger.info('Language now is DEFAULT')
- exit(0)
-
- if os.path.exists(constant.NHENTAI_PROXY):
- with open(constant.NHENTAI_PROXY, 'r') as f:
- link = f.read()
- constant.PROXY = {'http': link, 'https': link}
+ # TODO: search without language
if args.proxy:
- try:
- if not os.path.exists(constant.NHENTAI_HOME):
- os.mkdir(constant.NHENTAI_HOME)
-
- proxy_url = urlparse(args.proxy)
- if proxy_url.scheme not in ('http', 'https'):
- logger.error('Invalid protocol \'{0}\' of proxy, ignored'.format(proxy_url.scheme))
- else:
- with open(constant.NHENTAI_PROXY, 'w') as f:
- f.write(args.proxy)
-
- except Exception as e:
- logger.error('Cannot create NHENTAI_HOME: {}'.format(str(e)))
- exit(1)
-
+ proxy_url = urlparse(args.proxy)
+ if proxy_url.scheme not in ('http', 'https'):
+ logger.error('Invalid protocol \'{0}\' of proxy, ignored'.format(proxy_url.scheme))
+ constant.CONFIG['proxy'] = {
+ 'http': args.proxy,
+ 'https': args.proxy,
+ }
logger.info('Proxy \'{0}\' saved.'.format(args.proxy))
+ write_config()
exit(0)
+ # --- end set config ---
if args.favorites:
- if not constant.COOKIE:
+ if not constant.CONFIG['cookie']:
logger.warning('Cookie has not been set, please use `nhentai --cookie \'COOKIE\'` to set it.')
exit(1)
#!/usr/bin/env python2.7
# coding: utf-8
from __future__ import unicode_literals, print_function
+import json
+import os
import signal
import platform
import time
+from nhentai import constant
from nhentai.cmdline import cmd_parser, banner
from nhentai.parser import doujinshi_parser, search_parser, print_doujinshi, favorites_parser
from nhentai.doujinshi import Doujinshi
from nhentai.downloader import Downloader
from nhentai.logger import logger
-from nhentai.constant import BASE_URL, LANGUAGE
+from nhentai.constant import NHENTAI_CONFIG_FILE, BASE_URL
from nhentai.utils import generate_html, generate_cbz, generate_main_html, generate_pdf, \
paging, check_cookie, signal_handler, DB
options = cmd_parser()
logger.info('Using mirror: {0}'.format(BASE_URL))
- from nhentai.constant import PROXY
- # constant.PROXY will be changed after cmd_parser()
- if PROXY:
- logger.info('Using proxy: {0}'.format(PROXY))
+ # CONFIG['proxy'] will be changed after cmd_parser()
+ if constant.CONFIG['proxy']:
+ logger.info('Using proxy: {0}'.format(constant.CONFIG['proxy']))
# check your cookie
check_cookie()
doujinshis = favorites_parser(page=page_list)
elif options.keyword:
- if LANGUAGE:
- logger.info('Using default language: {0}'.format(LANGUAGE))
- options.keyword += ', language:{}'.format(LANGUAGE)
+ if constant.CONFIG['language']:
+ logger.info('Using default language: {0}'.format(constant.CONFIG['language']))
+ options.keyword += ' language:{}'.format(constant.CONFIG['language'])
doujinshis = search_parser(options.keyword, sorting=options.sorting, page=page_list,
is_page_all=options.page_all)