mirror of
https://github.com/RicterZ/nhentai.git
synced 2025-04-19 18:31:24 +02:00
237 lines
10 KiB
Python
237 lines
10 KiB
Python
# coding: utf-8
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import nhentai.constant as constant
|
|
|
|
from urllib.parse import urlparse
|
|
from optparse import OptionParser
|
|
|
|
from nhentai import __version__
|
|
from nhentai.utils import generate_html, generate_main_html, DB
|
|
from nhentai.logger import logger
|
|
|
|
|
|
def banner():
|
|
logger.debug(f'nHentai ver {__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.update(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 callback(option, _opt_str, _value, parser):
|
|
if option == '--id':
|
|
pass
|
|
value = []
|
|
|
|
for arg in parser.rargs:
|
|
if arg.isdigit():
|
|
value.append(int(arg))
|
|
elif arg.startswith('-'):
|
|
break
|
|
else:
|
|
logger.warning(f'Ignore invalid id {arg}')
|
|
|
|
setattr(parser.values, option.dest, value)
|
|
|
|
|
|
def cmd_parser():
|
|
load_config()
|
|
|
|
parser = OptionParser('\n nhentai --search [keyword] --download'
|
|
'\n NHENTAI=https://nhentai-mirror-url/ nhentai --id [ID ...]'
|
|
'\n nhentai --file [filename]'
|
|
'\n\nEnvironment Variable:\n'
|
|
' NHENTAI nhentai mirror url')
|
|
# operation options
|
|
parser.add_option('--download', '-D', dest='is_download', action='store_true',
|
|
help='download doujinshi (for search results)')
|
|
parser.add_option('--show', '-S', dest='is_show', action='store_true', help='just show the doujinshi information')
|
|
|
|
# doujinshi options
|
|
parser.add_option('--id', dest='id', action='callback', callback=callback,
|
|
help='doujinshi ids set, e.g. 167680 167681 167682')
|
|
parser.add_option('--search', '-s', type='string', dest='keyword', action='store',
|
|
help='search doujinshi by keyword')
|
|
parser.add_option('--favorites', '-F', action='store_true', dest='favorites',
|
|
help='list or download your favorites')
|
|
parser.add_option('--artist', '-a', action='store', dest='artist',
|
|
help='list doujinshi by artist name')
|
|
|
|
# page options
|
|
parser.add_option('--page-all', dest='page_all', action='store_true', default=False,
|
|
help='all search results')
|
|
parser.add_option('--page', '--page-range', type='string', dest='page', action='store',
|
|
help='page number of search results. e.g. 1,2-5,14')
|
|
parser.add_option('--sorting', '--sort', dest='sorting', action='store', default='popular',
|
|
help='sorting of doujinshi (recent / popular / popular-[today|week])',
|
|
choices=['recent', 'popular', 'popular-today', 'popular-week', 'date'])
|
|
|
|
# download options
|
|
parser.add_option('--output', '-o', type='string', dest='output_dir', action='store', default='./',
|
|
help='output dir')
|
|
parser.add_option('--threads', '-t', type='int', dest='threads', action='store', default=5,
|
|
help='thread count for downloading doujinshi')
|
|
parser.add_option('--timeout', '-T', type='int', dest='timeout', action='store', default=30,
|
|
help='timeout for downloading doujinshi')
|
|
parser.add_option('--delay', '-d', type='int', dest='delay', action='store', default=0,
|
|
help='slow down between downloading every doujinshi')
|
|
parser.add_option('--proxy', type='string', dest='proxy', action='store',
|
|
help='store a proxy, for example: -p "http://127.0.0.1:1080"')
|
|
parser.add_option('--file', '-f', type='string', dest='file', action='store', help='read gallery IDs from file.')
|
|
parser.add_option('--format', type='string', dest='name_format', action='store',
|
|
help='format the saved folder name', default='[%i][%a][%t]')
|
|
parser.add_option('--dry-run', action='store_true', dest='dryrun', help='Dry run, skip file download')
|
|
|
|
# generate options
|
|
parser.add_option('--html', dest='html_viewer', action='store_true',
|
|
help='generate a html viewer at current directory')
|
|
parser.add_option('--no-html', dest='is_nohtml', action='store_true',
|
|
help='don\'t generate HTML after downloading')
|
|
parser.add_option('--gen-main', dest='main_viewer', action='store_true',
|
|
help='generate a main viewer contain all the doujin in the folder')
|
|
parser.add_option('--cbz', '-C', dest='is_cbz', action='store_true',
|
|
help='generate Comic Book CBZ File')
|
|
parser.add_option('--pdf', '-P', dest='is_pdf', action='store_true',
|
|
help='generate PDF file')
|
|
parser.add_option('--rm-origin-dir', dest='rm_origin_dir', action='store_true', default=False,
|
|
help='remove downloaded doujinshi dir when generated CBZ or PDF file')
|
|
parser.add_option('--move-to-folder', dest='move_to_folder', action='store_true', default=False,
|
|
help='remove files in doujinshi dir then move new file to folder when generated CBZ or PDF file')
|
|
parser.add_option('--meta', dest='generate_metadata', action='store_true',
|
|
help='generate a metadata file in doujinshi format')
|
|
parser.add_option('--regenerate', dest='regenerate', action='store_true', default=False,
|
|
help='regenerate the cbz or pdf file if exists')
|
|
|
|
# nhentai options
|
|
parser.add_option('--cookie', type='str', dest='cookie', action='store',
|
|
help='set cookie of nhentai to bypass Cloudflare captcha')
|
|
parser.add_option('--useragent', '--user-agent', type='str', dest='useragent', action='store',
|
|
help='set useragent to bypass Cloudflare captcha')
|
|
parser.add_option('--language', type='str', dest='language', action='store',
|
|
help='set default language to parse doujinshis')
|
|
parser.add_option('--clean-language', dest='clean_language', action='store_true', default=False,
|
|
help='set DEFAULT as language to parse doujinshis')
|
|
parser.add_option('--save-download-history', dest='is_save_download_history', action='store_true',
|
|
default=False, help='save downloaded doujinshis, whose will be skipped if you re-download them')
|
|
parser.add_option('--clean-download-history', action='store_true', default=False, dest='clean_download_history',
|
|
help='clean download history')
|
|
parser.add_option('--template', dest='viewer_template', action='store',
|
|
help='set viewer template', default='')
|
|
parser.add_option('--legacy', dest='legacy', action='store_true', default=False,
|
|
help='use legacy searching method')
|
|
|
|
args, _ = parser.parse_args(sys.argv[1:])
|
|
|
|
if args.html_viewer:
|
|
generate_html(template=constant.CONFIG['template'])
|
|
sys.exit(0)
|
|
|
|
if args.main_viewer and not args.id and not args.keyword and not args.favorites:
|
|
generate_main_html()
|
|
sys.exit(0)
|
|
|
|
if args.clean_download_history:
|
|
with DB() as db:
|
|
db.clean_all()
|
|
|
|
logger.info('Download history cleaned.')
|
|
sys.exit(0)
|
|
|
|
# --- set config ---
|
|
if args.cookie is not None:
|
|
constant.CONFIG['cookie'] = args.cookie
|
|
write_config()
|
|
logger.info('Cookie saved.')
|
|
sys.exit(0)
|
|
elif args.useragent is not None:
|
|
constant.CONFIG['useragent'] = args.useragent
|
|
write_config()
|
|
logger.info('User-Agent saved.')
|
|
sys.exit(0)
|
|
elif args.language is not None:
|
|
constant.CONFIG['language'] = args.language
|
|
write_config()
|
|
logger.info(f'Default language now set to "{args.language}"')
|
|
sys.exit(0)
|
|
# TODO: search without language
|
|
|
|
if args.proxy is not None:
|
|
proxy_url = urlparse(args.proxy)
|
|
if not args.proxy == '' and proxy_url.scheme not in ('http', 'https', 'socks5', 'socks5h',
|
|
'socks4', 'socks4a'):
|
|
logger.error(f'Invalid protocol "{proxy_url.scheme}" of proxy, ignored')
|
|
sys.exit(0)
|
|
else:
|
|
constant.CONFIG['proxy'] = {
|
|
'http': args.proxy,
|
|
'https': args.proxy,
|
|
}
|
|
logger.info(f'Proxy now set to "{args.proxy}"')
|
|
write_config()
|
|
sys.exit(0)
|
|
|
|
if args.viewer_template is not None:
|
|
if not args.viewer_template:
|
|
args.viewer_template = 'default'
|
|
|
|
if not os.path.exists(os.path.join(os.path.dirname(__file__),
|
|
f'viewer/{args.viewer_template}/index.html')):
|
|
logger.error(f'Template "{args.viewer_template}" does not exists')
|
|
sys.exit(1)
|
|
else:
|
|
constant.CONFIG['template'] = args.viewer_template
|
|
write_config()
|
|
|
|
# --- end set config ---
|
|
|
|
if args.favorites:
|
|
if not constant.CONFIG['cookie']:
|
|
logger.warning('Cookie has not been set, please use `nhentai --cookie \'COOKIE\'` to set it.')
|
|
sys.exit(1)
|
|
|
|
if args.file:
|
|
with open(args.file, 'r') as f:
|
|
_ = [i.strip() for i in f.readlines()]
|
|
args.id = set(int(i) for i in _ if i.isdigit())
|
|
|
|
if (args.is_download or args.is_show) and not args.id and not args.keyword and not args.favorites and not args.artist:
|
|
logger.critical('Doujinshi id(s) are required for downloading')
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
|
|
if not args.keyword and not args.id and not args.favorites and not args.artist:
|
|
parser.print_help()
|
|
sys.exit(1)
|
|
|
|
if args.threads <= 0:
|
|
args.threads = 1
|
|
|
|
elif args.threads > 15:
|
|
logger.critical('Maximum number of used threads is 15')
|
|
sys.exit(1)
|
|
|
|
if args.dryrun and (args.is_cbz or args.is_pdf):
|
|
logger.critical('Cannot generate PDF or CBZ during dry-run')
|
|
sys.exit(1)
|
|
|
|
return args
|