Source code for flexget.components.backlog.cli

from flexget import options
from flexget.event import event
from flexget.manager import Session
from flexget.terminal import TerminalTable, console, table_parser

from . import db




def do_cli(manager, options):
    if options.action == 'clear':
        num = db.clear_entries(options.task, all=True)
        console(f'{num} entries cleared from backlog.')
    else:
        header = ['Title', 'Task', 'Expires']
        with Session() as session:
            entries = db.get_entries(options.task, session=session)
            table_data = [
                [entry.title, entry.task, entry.expire.strftime('%Y-%m-%d %H:%M')]
                for entry in entries
            ]
        table = TerminalTable(*header, table_type=options.table_type)
        for row in table_data:
            table.add_row(*row)
        console(table)





@event('options.register')
def register_options():
    parser = options.register_command(
        'backlog', do_cli, help='View or clear entries from backlog plugin', parents=[table_parser]
    )
    parser.add_argument(
        'action',
        choices=['list', 'clear'],
        help='Choose to show items in backlog, or clear all of them',
    )
    parser.add_argument('task', nargs='?', help='Limit to specific task (if supplied)')