refactor icon installation by simplifying hook

westerveltco · Mar 23, 2024 · 26d9fdb · 26d9fdb
1 parent 7bf3c2f
commit 26d9fdb
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 58 deletions.
diff --git a/src/wagtail_heroicons/icons.py b/src/wagtail_heroicons/icons.py
diff --git a/src/wagtail_heroicons/wagtail_hooks.py b/src/wagtail_heroicons/wagtail_hooks.py
@@ -1,15 +1,12 @@
 from __future__ import annotations
 
-try:
-    from wagtail import hooks
-except ImportError:
-    from wagtail.core import hooks
+from pathlib import Path
 
-from .icons import Heroicon
+from wagtail import hooks
 
 
 @hooks.register("register_icons")
 def register_icons(icons):
-    for icon in Heroicon.get_icons():
-        icons.append(str(icon.path))
-    return icons
+    template_dir = Path(__file__).parent / "templates" / "heroicons"
+    heroicons = [file.stem for file in template_dir.glob("*.svg")]
+    return icons + heroicons
diff --git a/tests/test_icons.py b/tests/test_icons.py