mirror of
https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
synced 2026-01-07 09:42:09 -08:00
parent
6136db1409
commit
fb62f1fb40
2 changed files with 20 additions and 4 deletions
|
|
@ -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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue