Esempio n. 1
0
    async def detect(self):
        self.context.request.prevent_result_storage = True
        try:
            if not Detector.detect_task:
                celery_tasks = CeleryTasks(
                    self.context.config.SQS_QUEUE_KEY_ID,
                    self.context.config.SQS_QUEUE_KEY_SECRET,
                    self.context.config.SQS_QUEUE_REGION,
                    None,
                )
                Detector.detect_task = celery_tasks.get_detect_task()

            Detector.detect_task.delay(
                "all",
                self.context.request.image_url,
                self.context.request.image_url,
            )
        except RuntimeError:
            self.context.request.detection_error = True
            Detector.detect_task = None
            logger.exception("Celery Error")

        # Error or not we return an empty list as detection
        # will be done later
        return []
Esempio n. 2
0
    def detect(self, callback):
        self.context.request.prevent_result_storage = True
        try:
            if not Detector.detect_task:
                celery_tasks = CeleryTasks(
                    self.context.config.SQS_QUEUE_KEY_ID,
                    self.context.config.SQS_QUEUE_KEY_SECRET,
                    self.context.config.SQS_QUEUE_REGION, None
                )
                Detector.detect_task = celery_tasks.get_detect_task()

            Detector.detect_task.delay('all', self.context.request.image_url, self.context.request.image_url)
        except RuntimeError:
            self.context.request.detection_error = True
            Detector.detect_task = None
            logger.exception('Celery Error')
        finally:
            callback([])