コード例 #1
0
ファイル: __init__.py プロジェクト: alexey-grom/fetcher
    def __init__(self, queue='memory', threads_count=20, **kwargs):
        '''
        Конструктор менеджера задач.
        Параметры:
            tasks_container - тип контейнера для хранения задач.
                Может принимать следующие значения: memory, mongo
        '''

        #self._queue = queue(**kwargs)

        if isinstance(queue, str):
            try:
                queue = import_module('fetcher.tasks.queues.%s' % queue).Queue
            except ImportError:
                raise Exception(u'Неудалось импортировать класс реализации очереди задач! Проверьте аргументы!')
        if queue:
            logger.info(u'Использование в качестве очереди задач %s' % queue)
            self._queue = queue(**kwargs)

        self._queue_size = threads_count * 2