1580158
/
stable-diffusion-webui-images-browser
mirror of https://github.com/AlUlkesh/stable-diffusion-webui-images-browser.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remaining "conn" references changed to "cursor"

pull/223/head
AlUlkesh 2023-08-16 07:27:23 +02:00
parent 7bb07e0343
commit 51df4c1e5d
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
scripts/image_browser.py
View File

@ -495,7 +495,7 @@ def cache_exif(fileinfos):
raise raise
if allExif: if allExif:
exif_cache[fi_info[0]] = allExif exif_cache[fi_info[0]] = allExif
wib_db.update_exif_data(conn, fi_info[0], allExif) wib_db.update_exif_data(cursor, fi_info[0], allExif)
new_exif = new_exif + 1 new_exif = new_exif + 1
m = re.search("(?:aesthetic_score:|Score:) (\d+.\d+)", allExif, flags=re.IGNORECASE) m = re.search("(?:aesthetic_score:|Score:) (\d+.\d+)", allExif, flags=re.IGNORECASE)
@ -504,7 +504,7 @@ def cache_exif(fileinfos):
else: else:
aes_value = "0" aes_value = "0"
aes_cache[fi_info[0]] = aes_value aes_cache[fi_info[0]] = aes_value
wib_db.update_exif_data_by_key(conn, fi_info[0], "aesthetic_score", aes_value) wib_db.update_exif_data_by_key(cursor, fi_info[0], "aesthetic_score", aes_value)
new_aes = new_aes + 1 new_aes = new_aes + 1
else: else:
try: try:
@ -514,7 +514,7 @@ def cache_exif(fileinfos):
for line in f: for line in f:
geninfo += line geninfo += line
exif_cache[fi_info[0]] = geninfo exif_cache[fi_info[0]] = geninfo
wib_db.update_exif_data_by_key(conn, fi_info[0], geninfo) wib_db.update_exif_data_by_key(cursor, fi_info[0], geninfo)
new_exif = new_exif + 1 new_exif = new_exif + 1
m = re.search("(?:aesthetic_score:|Score:) (\d+.\d+)", geninfo, flags=re.IGNORECASE) m = re.search("(?:aesthetic_score:|Score:) (\d+.\d+)", geninfo, flags=re.IGNORECASE)
@ -523,19 +523,19 @@ def cache_exif(fileinfos):
else: else:
aes_value = "0" aes_value = "0"
aes_cache[fi_info[0]] = aes_value aes_cache[fi_info[0]] = aes_value
wib_db.update_exif_data_by_key(conn, fi_info[0], "aesthetic_score", aes_value) wib_db.update_exif_data_by_key(cursor, fi_info[0], "aesthetic_score", aes_value)
new_aes = new_aes + 1 new_aes = new_aes + 1
except Exception: except Exception:
logger.warning(f"cache_exif: No EXIF in image or txt file for {fi_info[0]}") logger.warning(f"cache_exif: No EXIF in image or txt file for {fi_info[0]}")
# Saved with defaults to not scan it again next time # Saved with defaults to not scan it again next time
exif_cache[fi_info[0]] = "0" exif_cache[fi_info[0]] = "0"
allExif = "0" allExif = "0"
wib_db.update_exif_data(conn, fi_info[0], allExif) wib_db.update_exif_data(cursor, fi_info[0], allExif)
new_exif = new_exif + 1 new_exif = new_exif + 1
aes_value = "0" aes_value = "0"
aes_cache[fi_info[0]] = aes_value aes_cache[fi_info[0]] = aes_value
wib_db.update_exif_data_by_key(conn, fi_info[0], "aesthetic_score", aes_value) wib_db.update_exif_data_by_key(cursor, fi_info[0], "aesthetic_score", aes_value)
new_aes = new_aes + 1 new_aes = new_aes + 1
if yappi_do: if yappi_do: