Source code for flexget.plugins.input.generate

from loguru import logger

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

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




class Generate:
    """Generates n number of random entries. Used for debugging purposes."""

    schema = {'type': 'integer'}



    def on_task_input(self, task, config):
        amount = config or 0  # hackily makes sure it's an int value
        entries = []
        for i in range(amount):
            entry = Entry()
            import random
            import string

            entry['url'] = 'http://localhost/generate/{}/{}'.format(
                i,
                ''.join([
                    random.choice(string.ascii_letters + string.digits) for x in range(1, 30)
                ]),
            )
            entry['title'] = ''.join([
                random.choice(string.ascii_letters + string.digits) for x in range(1, 30)
            ])
            entry['description'] = ''.join([
                random.choice(string.ascii_letters + string.digits) for x in range(1, 1000)
            ])
            entries.append(entry)
        return entries






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