def create_by_worker(self, *args, **kwargs): kwargs['url'] = sanitize_url(kwargs['url']) kwargs['source_type'] = SAMPLE_TAGASAURIS_WORKER kwargs['source_val'] = kwargs['worker_id'] del kwargs['worker_id'] try: kwargs['sample'] = Sample.objects.get( job=kwargs['job'], url=kwargs['url'] ) except Sample.DoesNotExist: pass btm_sample = self.create(**kwargs) # If sample exists, step immediately to classification if 'sample' in kwargs: send_event('EventNewBTMSample', sample_id=kwargs['sample'].id, job_id=kwargs['job'].id) else: Sample.objects.create_by_btm( job_id=kwargs['job'].id, url=kwargs['url'], source_val=kwargs['source_val'], create_classified=False, ) return btm_sample
def _sanitize(self, args, kwargs): """ Sanitizes information passed by users. """ kwargs['url'] = sanitize_url(kwargs['url'])
def _sanitize(self, args, kwargs): """ Sanitizes information passed by users. """ kwargs["url"] = sanitize_url(kwargs["url"])