From 4222593a46a7583af1f5740bc7066c9389df9f2e Mon Sep 17 00:00:00 2001
From: Bastien Abadie <abadie@teklia.com>
Date: Mon, 23 Aug 2021 22:30:31 +0200
Subject: [PATCH] Explicit error message for corpus_id type in
check_required_types
---
arkindex_worker/worker/element.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/arkindex_worker/worker/element.py b/arkindex_worker/worker/element.py
index f751b19b..747af95b 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
--
GitLab