67 lines
2.7 KiB
Python
67 lines
2.7 KiB
Python
import argparse
|
|
from pathlib import Path
|
|
from pytchat.util.extract_video_id import extract_video_id
|
|
from .arguments import Arguments
|
|
from .. exceptions import InvalidVideoIdException, NoContents
|
|
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('VideoID_or_URL', type=str, default='__NONE__',nargs='?',
|
|
# help='Video ID, or URL that includes id.\n'
|
|
# 'If ID starts with a hyphen (-), enclose the ID in square brackets.')
|
|
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:
|
|
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()}")
|
|
path = Path(Arguments().output + video_id + '.html')
|
|
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)
|
|
return
|
|
parser.print_help()
|
|
|
|
|
|
def _disp_progress(a, b):
|
|
print('.', end="", flush=True)
|