mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2025-08-03 10:50:23 +00:00
@@ -81,6 +81,17 @@ class MassFileListerCachedDir:
|
||||
self.files = {x[0].lower(): x for x in files}
|
||||
self.files_cased = {x[0]: x for x in files}
|
||||
|
||||
def update_entry(self, filename):
|
||||
"""Add a file to the cache"""
|
||||
file_path = os.path.join(self.dirname, filename)
|
||||
try:
|
||||
stat = os.stat(file_path)
|
||||
entry = (filename, stat.st_mtime, stat.st_ctime)
|
||||
self.files[filename.lower()] = entry
|
||||
self.files_cased[filename] = entry
|
||||
except FileNotFoundError as e:
|
||||
print(f'MassFileListerCachedDir.add_entry: "{file_path}" {e}')
|
||||
|
||||
|
||||
class MassFileLister:
|
||||
"""A class that provides a way to check for the existence and mtime/ctile of files without doing more than one stat call per file."""
|
||||
@@ -136,3 +147,9 @@ class MassFileLister:
|
||||
def reset(self):
|
||||
"""Clear the cache of all directories."""
|
||||
self.cached_dirs.clear()
|
||||
|
||||
def update_file_entry(self, path):
|
||||
"""Update the cache for a specific directory."""
|
||||
dirname, filename = os.path.split(path)
|
||||
if cached_dir := self.cached_dirs.get(dirname):
|
||||
cached_dir.update_entry(filename)
|
||||
|
Reference in New Issue
Block a user