#162 catch IntegrityError for Wagtail5, too

ephes · Sep 8, 2024 · ee0d93a · ee0d93a
1 parent debce74
commit ee0d93a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cast/context_processors.py b/cast/context_processors.py
@@ -1,3 +1,4 @@
+from django.db import IntegrityError
 from django.http import HttpRequest
 
 from .models import TemplateBaseDirectory
@@ -15,8 +16,9 @@ def site_template_base_dir(request: HttpRequest) -> dict[str, str]:
     else:
         try:
             site_template_base_dir_name = TemplateBaseDirectory.for_request(request).name
-        except TemplateBaseDirectory.DoesNotExist:
+        except (TemplateBaseDirectory.DoesNotExist, IntegrityError):
             # If the site template base directory does not exist, use the default
+            # need to catch IntegrityError because of Wagtail5 support
             site_template_base_dir_name = DEFAULT_TEMPLATE_BASE_DIR
     return {
         "cast_site_template_base_dir": site_template_base_dir_name,