From b5fe48746e80fedd6f083df88442e95d29f4c87b Mon Sep 17 00:00:00 2001 From: Nontre12 <62849884+Nontre12@users.noreply.github.com> Date: Fri, 9 Oct 2020 17:34:03 +0200 Subject: [PATCH] Added --clean-language option --- nhentai/cmdline.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/nhentai/cmdline.py b/nhentai/cmdline.py index 762a66d..1e13215 100644 --- a/nhentai/cmdline.py +++ b/nhentai/cmdline.py @@ -96,6 +96,8 @@ def cmd_parser(): help='set cookie of nhentai to bypass Google recaptcha') 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', @@ -163,6 +165,20 @@ def cmd_parser(): logger.info('Default language now is {}.'.format(args.language)) 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()