Sorting and filtering will use .txt as well

pull/13/head
Kyle 2023-01-26 22:04:35 -05:00
parent 263502565e
commit 974c5be722
1 changed files with 32 additions and 7 deletions

View File

@ -134,11 +134,22 @@ def cache_aes(fileinfos):
try:
image = PngImageFile(fi_info[0])
allExif = modules.extras.run_pnginfo(image)[1]
if allExif:
if allExif and allExif != "":
m = re.search("aesthetic_score: (\d+\.\d+)", allExif)
if m:
finfo_aes[fi_info[0]] = m.group(1)
aes_cache[fi_info[0]] = m.group(1)
else:
try:
filename = os.path.splitext(fi_info[0])[0] + ".txt"
geninfo = ""
with open(filename) as f:
for line in f:
geninfo += line
finfo_exif[fi_info[0]] = geninfo
exif_cache[fi_info[0]] = geninfo
except Exception:
print(f"No exif for {fi_info[0]}")
except SyntaxError:
print(f"Non-PNG file in directory when doing AES check: {fi_info[0]}")
@ -165,6 +176,17 @@ def cache_exif(fileinfos):
if allExif:
finfo_exif[fi_info[0]] = allExif
exif_cache[fi_info[0]] = allExif
else:
try:
filename = os.path.splitext(fi_info[0])[0] + ".txt"
geninfo = ""
with open(filename) as f:
for line in f:
geninfo += line
finfo_exif[fi_info[0]] = geninfo
exif_cache[fi_info[0]] = geninfo
except Exception:
print(f"No EXIF in PNG or txt file fpr {fi_info[0]}")
#print(f"{fi_info[0]} exif added: {allExif}!")
except SyntaxError:
print(f"Non-PNG file in directory when doing EXIF check: {fi_info[0]}")
@ -521,13 +543,16 @@ def run_pnginfo(image, image_path, image_file_name):
<p>{plaintext_to_html(str(text))}</p>
</div>
""".strip()+"\n"
if geninfo is None:
filename = os.path.splitext(image_file_name)[0] + ".txt"
geninfo = ""
with open(filename) as f:
for line in f:
geninfo += line
try:
filename = os.path.splitext(image_file_name)[0] + ".txt"
geninfo = ""
with open(filename) as f:
for line in f:
geninfo += line
except Exception:
print(f"No EXIF in PNG or txt file")
return '', geninfo, info