Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
S
Stable Diffusion Webui
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Locked Files
Issues
0
Issues
0
List
Boards
Labels
Service Desk
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Security & Compliance
Security & Compliance
Dependency List
License Compliance
Packages
Packages
List
Container Registry
Analytics
Analytics
CI / CD
Code Review
Insights
Issues
Repository
Value Stream
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
novelai-storage
Stable Diffusion Webui
Commits
fb62f1fb
Commit
fb62f1fb
authored
Mar 10, 2024
by
w-e-w
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add entry to MassFileLister after writing metadata
fix #15184
parent
6136db14
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
20 additions
and
4 deletions
+20
-4
modules/ui_extra_networks_user_metadata.py
modules/ui_extra_networks_user_metadata.py
+3
-4
modules/util.py
modules/util.py
+17
-0
No files found.
modules/ui_extra_networks_user_metadata.py
View file @
fb62f1fb
...
...
@@ -133,8 +133,10 @@ class UserMetadataEditor:
filename
=
item
.
get
(
"filename"
,
None
)
basename
,
ext
=
os
.
path
.
splitext
(
filename
)
with
open
(
basename
+
'.json'
,
"w"
,
encoding
=
"utf8"
)
as
file
:
metadata_path
=
basename
+
'.json'
with
open
(
metadata_path
,
"w"
,
encoding
=
"utf8"
)
as
file
:
json
.
dump
(
metadata
,
file
,
indent
=
4
,
ensure_ascii
=
False
)
self
.
page
.
lister
.
update_file_entry
(
metadata_path
)
def
save_user_metadata
(
self
,
name
,
desc
,
notes
):
user_metadata
=
self
.
get_user_metadata
(
name
)
...
...
@@ -200,6 +202,3 @@ class UserMetadataEditor:
inputs
=
[
self
.
edit_name_input
],
outputs
=
[]
)
modules/util.py
View file @
fb62f1fb
...
...
@@ -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
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment