250 lines
7.3 KiB
Python
250 lines
7.3 KiB
Python
import argparse
|
|
import sys
|
|
import traceback
|
|
|
|
from voussoirkit import betterhelp
|
|
from voussoirkit import interactive
|
|
from voussoirkit import pipeable
|
|
from voussoirkit import vlogging
|
|
from voussoirkit import operatornotify
|
|
|
|
import ycdl
|
|
|
|
log = vlogging.getLogger(__name__, 'ycdl_cli')
|
|
|
|
# HELPERS ##########################################################################################
|
|
|
|
def closest_db():
|
|
return ycdl.ycdldb.YCDLDB.closest_ycdldb()
|
|
|
|
# ARGPARSE #########################################################################################
|
|
|
|
def add_channel_argparse(args):
|
|
ycdldb = closest_db()
|
|
ycdldb.add_channel(
|
|
channel_id=args.channel_id,
|
|
automark=args.automark,
|
|
download_directory=args.download_directory,
|
|
get_videos=args.get_videos,
|
|
name=args.name,
|
|
queuefile_extension=args.queuefile_extension,
|
|
)
|
|
|
|
if args.autoyes or interactive.getpermission('Commit?'):
|
|
ycdldb.commit()
|
|
|
|
return 0
|
|
|
|
def channel_list_argparse(args):
|
|
ycdldb = closest_db()
|
|
channels = sorted(ycdldb.get_channels(), key=lambda c: c.name.lower())
|
|
for channel in channels:
|
|
line = args.format.format(
|
|
automark=channel.automark,
|
|
autorefresh=channel.autorefresh,
|
|
id=channel.id,
|
|
name=channel.name,
|
|
queuefile_extension=channel.queuefile_extension,
|
|
uploads_playlist=channel.uploads_playlist,
|
|
)
|
|
pipeable.stdout(line)
|
|
|
|
return 0
|
|
|
|
def delete_channel_argparse(args):
|
|
ycdldb = closest_db()
|
|
for channel_id in pipeable.input_many(args.channel_ids):
|
|
channel = ycdldb.get_channel(channel_id)
|
|
channel.delete()
|
|
|
|
if args.autoyes or interactive.getpermission('Commit?'):
|
|
ycdldb.commit()
|
|
|
|
return 0
|
|
|
|
def init_argparse(args):
|
|
ycdldb = ycdl.ycdldb.YCDLDB(create=True)
|
|
ycdldb.commit()
|
|
pipeable.stdout(ycdldb.data_directory.absolute_path)
|
|
return 0
|
|
|
|
def refresh_channels_argparse(args):
|
|
needs_commit = False
|
|
status = 0
|
|
|
|
ycdldb = closest_db()
|
|
if args.channels:
|
|
channels = [ycdldb.get_channel(c) for c in args.channels]
|
|
for channel in channels:
|
|
try:
|
|
channel.refresh(force=args.force)
|
|
needs_commit = True
|
|
except Exception as exc:
|
|
log.warning(traceback.format_exc())
|
|
status = 1
|
|
else:
|
|
excs = ycdldb.refresh_all_channels(force=args.force, skip_failures=True)
|
|
needs_commit = True
|
|
|
|
if not needs_commit:
|
|
return status
|
|
|
|
if args.autoyes or interactive.getpermission('Commit?'):
|
|
ycdldb.commit()
|
|
|
|
return status
|
|
|
|
DOCSTRING = '''
|
|
YCDL CLI
|
|
========
|
|
|
|
{add_channel}
|
|
|
|
{channel_list}
|
|
|
|
{delete_channel}
|
|
|
|
{init}
|
|
|
|
{refresh_channels}
|
|
|
|
TO SEE DETAILS ON EACH COMMAND, RUN
|
|
> ycdl_cli.py <command> --help
|
|
'''
|
|
|
|
SUB_DOCSTRINGS = dict(
|
|
add_channel='''
|
|
add_channel:
|
|
Add a channel to the database.
|
|
|
|
> ycdl_cli.py add_channel channel_id <flags>
|
|
|
|
flags:
|
|
--automark X:
|
|
Set the channel's automark to this value, which should be 'pending',
|
|
'downloaded', or 'ignored'.
|
|
|
|
--download_directory X:
|
|
Set the channel's download directory to this path, which must
|
|
be a directory.
|
|
|
|
--name X:
|
|
Override the channel's own name with a name of your choosing.
|
|
|
|
--no_videos:
|
|
By default, the channel's videos will be fetched right away.
|
|
Add this argument if you don't want to do that yet.
|
|
|
|
--queuefile_extension X:
|
|
Set the queuefile extension for all videos downloaded from this channel.
|
|
|
|
Examples:
|
|
> ycdl_cli.py add_channel UCFhXFikryT4aFcLkLw2LBLA
|
|
'''.strip(),
|
|
|
|
channel_list='''
|
|
channel_list:
|
|
Print all channels in the database.
|
|
|
|
> ycdl_cli.py channel_list <flags>
|
|
|
|
flags:
|
|
--format X:
|
|
A string like "{id}: {name}" to format the attributes of the channel.
|
|
The available attributes are id, name, automark, autorefresh,
|
|
uploads_playlist queuefile_extension.
|
|
|
|
> ycdl_cli.py channel_list
|
|
|
|
Example:
|
|
> ycdl_cli.py channel_list
|
|
> ycdl_cli.py channel_list --format "{id} automark={automark}"
|
|
'''.strip(),
|
|
|
|
delete_channel='''
|
|
delete_channel:
|
|
Delete a channel and all its videos from the database.
|
|
|
|
You can pass multiple channel IDs.
|
|
Uses pipeable to support !c clipboard, !i stdin.
|
|
|
|
> ycdl_cli.py delete_channel channel_id [channel_id channel_id...]
|
|
|
|
Examples:
|
|
> ycdl_cli.py delete_channel UCOYBuFGi8T3NM5fNAptCLCw
|
|
> ycdl_cli.py delete_channel UCOYBuFGi8T3NM5fNAptCLCw UCmu9PVIZBk-ZCi-Sk2F2utA
|
|
> ycdl_cli.py channel_list --format {id} | ycdl_cli.py delete_channel !i
|
|
'''.strip(),
|
|
|
|
init='''
|
|
init:
|
|
Create a new YCDL database in the current directory.
|
|
|
|
> ycdl_cli.py init
|
|
'''.strip(),
|
|
|
|
refresh_channels='''
|
|
refresh_channels:
|
|
Refresh some or all channels in the database.
|
|
|
|
> ycdl_cli.py refresh_channels <flags>
|
|
|
|
flags:
|
|
--channels X Y Z:
|
|
Any number of channel IDs.
|
|
|
|
--force:
|
|
If omitted, only new videos are downloaded.
|
|
If included, channels are refreshed completely.
|
|
|
|
Examples:
|
|
> ycdl_cli.py refresh_channels --force
|
|
> ycdl_cli.py refresh_channels --channels UC1_uAIS3r8Vu6JjXWvastJg
|
|
'''
|
|
)
|
|
|
|
DOCSTRING = betterhelp.add_previews(DOCSTRING, SUB_DOCSTRINGS)
|
|
|
|
@operatornotify.main_decorator(subject='ycdl_cli')
|
|
@vlogging.main_decorator
|
|
def main(argv):
|
|
parser = argparse.ArgumentParser(description=__doc__)
|
|
subparsers = parser.add_subparsers()
|
|
|
|
p_add_channel = subparsers.add_parser('add_channel', aliases=['add-channel'])
|
|
p_add_channel.add_argument('channel_id')
|
|
p_add_channel.add_argument('--automark', default='pending')
|
|
p_add_channel.add_argument('--download_directory', '--download-directory', default=None)
|
|
p_add_channel.add_argument('--name', default=None)
|
|
p_add_channel.add_argument('--no_videos', '--no-videos', dest='get_videos', action='store_false')
|
|
p_add_channel.add_argument('--queuefile_extension', '--queuefile-extension', default=None)
|
|
p_add_channel.add_argument('--yes', dest='autoyes', action='store_true')
|
|
p_add_channel.set_defaults(func=add_channel_argparse)
|
|
|
|
p_channel_list = subparsers.add_parser('channel_list', aliases=['channel-list'])
|
|
p_channel_list.add_argument('--format', default='{id}:{name}')
|
|
p_channel_list.set_defaults(func=channel_list_argparse)
|
|
|
|
p_delete_channel = subparsers.add_parser('delete_channel', aliases=['delete-channel'])
|
|
p_delete_channel.add_argument('channel_ids', nargs='+')
|
|
p_delete_channel.add_argument('--yes', dest='autoyes', action='store_true')
|
|
p_delete_channel.set_defaults(func=delete_channel_argparse)
|
|
|
|
p_init = subparsers.add_parser('init')
|
|
p_init.set_defaults(func=init_argparse)
|
|
|
|
p_refresh_channels = subparsers.add_parser('refresh_channels', aliases=['refresh-channels'])
|
|
p_refresh_channels.add_argument('--channels', nargs='*')
|
|
p_refresh_channels.add_argument('--force', action='store_true')
|
|
p_refresh_channels.add_argument('--yes', dest='autoyes', action='store_true')
|
|
p_refresh_channels.set_defaults(func=refresh_channels_argparse)
|
|
|
|
try:
|
|
return betterhelp.subparser_main(argv, parser, DOCSTRING, SUB_DOCSTRINGS)
|
|
except ycdl.exceptions.NoClosestYCDLDB as exc:
|
|
pipeable.stderr(exc.error_message)
|
|
pipeable.stderr('Try `ycdl_cli.py init` to create the database.')
|
|
return 1
|
|
|
|
if __name__ == '__main__':
|
|
raise SystemExit(main(sys.argv[1:]))
|