diff --git a/horde/classes/stable/processing_generation.py b/horde/classes/stable/processing_generation.py index 94d9360a..13fe77d6 100644 --- a/horde/classes/stable/processing_generation.py +++ b/horde/classes/stable/processing_generation.py @@ -79,11 +79,15 @@ def log_aborted_generation(self): def set_generation(self, generation, things_per_sec, **kwargs): state = kwargs.get("state", "ok") - for metadata in kwargs.get("gen_metadata", []): - if metadata.get('value') == 'csam': - state = 'csam' + gen_metadata = kwargs.get("gen_metadata") or [] + for metadata in gen_metadata: + if metadata.get("type") != "censorship": + # this metadata isnt about censorship + continue + if metadata.get("value") == "csam": + state = "csam" else: - state = 'censored' + state = "censored" if state in ["censored", "csam"]: self.censored = True db.session.commit()