75 lines
3.0 KiB
Python
75 lines
3.0 KiB
Python
import argparse
|
|
import os
|
|
from pathlib import Path
|
|
from pytchat.util.extract_video_id import extract_video_id
|
|
from .arguments import Arguments
|
|
from .. exceptions import InvalidVideoIdException, NoContents, VideoInfoParseException
|
|
from .. processors.html_archiver import HTMLArchiver
|
|
from .. tool.extract.extractor import Extractor
|
|
from .. tool.videoinfo import VideoInfo
|
|
from .. import __version__
|
|
|
|
'''
|
|
Most of CLI modules refer to
|
|
Petter Kraabøl's Twitch-Chat-Downloader
|
|
https://github.com/PetterKraabol/Twitch-Chat-Downloader
|
|
(MIT License)
|
|
|
|
'''
|
|
|
|
|
|
def main():
|
|
# Arguments
|
|
parser = argparse.ArgumentParser(description=f'pytchat v{__version__}')
|
|
parser.add_argument('-v', f'--{Arguments.Name.VIDEO_IDS}', type=str,
|
|
help='Video ID (or URL that includes Video ID). You can specify multiple video IDs by '
|
|
'separating them with commas without spaces.\n'
|
|
'If ID starts with a hyphen (-), enclose the ID in square brackets.')
|
|
parser.add_argument('-o', f'--{Arguments.Name.OUTPUT}', type=str,
|
|
help='Output directory (end with "/"). default="./"', default='./')
|
|
parser.add_argument(f'--{Arguments.Name.VERSION}', action='store_true',
|
|
help='Show version')
|
|
Arguments(parser.parse_args().__dict__)
|
|
if Arguments().print_version:
|
|
print(f'pytchat v{__version__} © 2019 taizan-hokuto')
|
|
return
|
|
|
|
# Extractor
|
|
if Arguments().video_ids:
|
|
for video_id in Arguments().video_ids:
|
|
if '[' in video_id:
|
|
video_id = video_id.replace('[', '').replace(']', '')
|
|
try:
|
|
video_id = extract_video_id(video_id)
|
|
if os.path.exists(Arguments().output):
|
|
path = Path(Arguments().output + video_id + '.html')
|
|
else:
|
|
raise FileNotFoundError
|
|
info = VideoInfo(video_id)
|
|
print(f"Extracting...\n"
|
|
f" video_id: {video_id}\n"
|
|
f" channel: {info.get_channel_name()}\n"
|
|
f" title: {info.get_title()}")
|
|
|
|
print(f" output path: {path.resolve()}")
|
|
Extractor(video_id,
|
|
processor=HTMLArchiver(
|
|
Arguments().output + video_id + '.html'),
|
|
callback=_disp_progress
|
|
).extract()
|
|
print("\nExtraction end.\n")
|
|
except InvalidVideoIdException:
|
|
print("Invalid Video ID or URL:", video_id)
|
|
except (TypeError, NoContents) as e:
|
|
print(e)
|
|
except FileNotFoundError:
|
|
print("The specified directory does not exist.:{}".format(Arguments().output))
|
|
except VideoInfoParseException:
|
|
print("Cannot parse video information.:{}".format(video_id))
|
|
return
|
|
parser.print_help()
|
|
|
|
|
|
def _disp_progress(a, b):
|
|
print('.', end="", flush=True)
|