check user template dir when exporting single file

eylles · Jan 19, 2024 · 5adbd41 · 5adbd41
1 parent bf8e057
commit 5adbd41
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pywal/export.py b/pywal/export.py
@@ -131,7 +131,12 @@ def color(colors, export_type, output_file=None):
     all_colors = flatten_colors(colors)
 
     template_name = get_export_type(export_type)
-    template_file = os.path.join(MODULE_DIR, "templates", template_name)
+
+    if template_name == export_type:
+        template_file = os.path.join(CONF_DIR, "templates", template_name)
+    else:
+        template_file = os.path.join(MODULE_DIR, "templates", template_name)
+
     output_file = output_file or os.path.join(CACHE_DIR, template_name)
 
     if os.path.isfile(template_file):