File size: 2,604 Bytes
9dce458 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
import os
import asyncio
import logging
from argparse import Namespace
from .manga_translator import (
MangaTranslator,
MangaTranslatorWeb,
MangaTranslatorWS,
MangaTranslatorAPI,
set_main_logger,
)
from .args import parser
from .utils import (
BASE_PATH,
init_logging,
get_logger,
set_log_level,
natural_sort,
)
# TODO: Dynamic imports to reduce ram usage in web(-server) mode. Will require dealing with args.py imports.
async def dispatch(args: Namespace):
args_dict = vars(args)
logger.info(f'Running in {args.mode} mode')
if args.mode in ('demo', 'batch'):
if not args.input:
raise Exception('No input image was supplied. Use -i <image_path>')
translator = MangaTranslator(args_dict)
if args.mode == 'demo':
if len(args.input) != 1 or not os.path.isfile(args.input[0]):
raise FileNotFoundError(f'Invalid single image file path for demo mode: "{" ".join(args.input)}". Use `-m batch`.')
dest = os.path.join(BASE_PATH, 'result/final.png')
args.overwrite = True # Do overwrite result/final.png file
await translator.translate_path(args.input[0], dest, args_dict)
else: # batch
dest = args.dest
for path in natural_sort(args.input):
await translator.translate_path(path, dest, args_dict)
elif args.mode == 'web':
from .server.web_main import dispatch
await dispatch(args.host, args.port, translation_params=args_dict)
elif args.mode == 'web_client':
translator = MangaTranslatorWeb(args_dict)
await translator.listen(args_dict)
elif args.mode == 'ws':
translator = MangaTranslatorWS(args_dict)
await translator.listen(args_dict)
elif args.mode == 'api':
translator = MangaTranslatorAPI(args_dict)
await translator.listen(args_dict)
if __name__ == '__main__':
args = None
init_logging()
try:
args = parser.parse_args()
set_log_level(level=logging.DEBUG if args.verbose else logging.INFO)
logger = get_logger(args.mode)
set_main_logger(logger)
if args.mode != 'web':
logger.debug(args)
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
loop.run_until_complete(dispatch(args))
except KeyboardInterrupt:
if not args or args.mode != 'web':
print()
except Exception as e:
logger.error(f'{e.__class__.__name__}: {e}',
exc_info=e if args and args.verbose else None)
|