diff --git a/arkindex_worker/worker/element.py b/arkindex_worker/worker/element.py index f751b19b632416e4b86c695df86db5c2a555e126..747af95ba4cf492fe9981eb3e680a0f5e538bc85 100644 --- a/arkindex_worker/worker/element.py +++ b/arkindex_worker/worker/element.py @@ -1,5 +1,6 @@ # -*- coding: utf-8 -*- +import uuid from peewee import IntegrityError @@ -20,6 +21,9 @@ class ElementMixin(object): Check that a corpus has a list of required element types, and raise an exception if any of them are missing. """ + assert isinstance( + corpus_id, (uuid.UUID, str) + ), "Corpus ID should be a string or UUID" assert len(type_slugs), "At least one element type slug is required." assert all( isinstance(slug, str) for slug in type_slugs