From 083dc3c76ab7dbc7b2b04f3396d4f5280b002906 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Mon, 8 May 2023 11:33:45 +0300
Subject: directory hiding for extra networks: dirs starting with . will hide
 their cards on extra network tabs unless specifically searched for create
 HTML for extra network pages only on demand allow directories starting with .
 to still list their models for lora, checkpoints, etc keep "search" filter
 for extra networks when user refreshes the page

---
 modules/shared.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

(limited to 'modules/shared.py')

diff --git a/modules/shared.py b/modules/shared.py
index 91aac1a3..dd374713 100644
--- a/modules/shared.py
+++ b/modules/shared.py
@@ -726,3 +726,20 @@ def html(filename):
             return file.read()
 
     return ""
+
+
+def walk_files(path, allowed_extensions=None):
+    if not os.path.exists(path):
+        return
+
+    if allowed_extensions is not None:
+        allowed_extensions = set(allowed_extensions)
+
+    for root, dirs, files in os.walk(path):
+        for filename in files:
+            if allowed_extensions is not None:
+                _, ext = os.path.splitext(filename)
+                if ext not in allowed_extensions:
+                    continue
+
+            yield os.path.join(root, filename)
-- 
cgit v1.2.3