Source code for flexget.components.series.gen_series

import random
import string

from loguru import logger

from flexget import plugin
from flexget.entry import Entry
from flexget.event import event

logger = logger.bind(name='gen_series')

PER_RUN = 50




class GenSeries:
    """Purely for debugging purposes. Not great quality :).

    ::

        gen_series_data:
            series: NUM
            seasons: NUM
            episodes: NUM
            qualities:
              - LIST

    This will also auto configure series plugin for testing
    """

    def __init__(self):
        self.entries = []

    schema = {'type': 'object', 'minProperties': 1}



    @plugin.priority(200)
    def on_task_start(self, task, config):
        logger.info('Generating test data ...')
        series = []
        for num in range(config['series']):
            series.append(f'series {num} name')
            for season in range(int(config['seasons'])):
                for episode in range(int(config['episodes'])):
                    for quality in config['qualities']:
                        entry = Entry()
                        entry['title'] = (
                            f'series {num} name - S{season + 1:02d}E{episode + 1:02d} - {quality}'
                        )
                        entry['url'] = 'http://localhost/mock/{}'.format(
                            ''.join([
                                random.choice(string.ascii_letters + string.digits)
                                for x in range(1, 30)
                            ])
                        )
                        self.entries.append(entry)
        logger.info('Generated {} entries', len(self.entries))

        # configure series plugin, bad way but this is debug shit
        task.config['series'] = series




    def on_task_input(self, task, config):
        entries = []
        for num, entry in enumerate(self.entries):
            entries.append(entry)
            if num == PER_RUN - 1:
                break
        self.entries = self.entries[len(entries) :]
        return entries




    def on_task_exit(self, task, config):
        if self.entries:
            logger.info('There are still {} left to be processed!', len(self.entries))
            # rerun ad infinitum, also commits session between them
            task._rerun = True
            task._rerun_count = 0






@event('plugin.register')
def register_plugin():
    plugin.register(GenSeries, 'gen_series_data', api_ver=2, debug=True)