venus/aperisolve
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix invalid-context-manager

Browse Source
This commit is contained in:
Zeecka
2026-02-21 00:05:01 +01:00
parent 730ed50f70
commit 00c5ce4fe4
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
aperisolve/models.py
+5 -1
View File
@@ -136,7 +136,9 @@ def fill_ihdr_db() -> None:
combinations = itertools.product(resolutions, bit_color_pairs, interlace_methods) combinations = itertools.product(resolutions, bit_color_pairs, interlace_methods)
count = 0 count = 0
with db.session.no_autoflush: previous_autoflush = db.session.autoflush
db.session.autoflush = False
try:
for (width, height), (bit_depth, color_type), interlace in combinations: for (width, height), (bit_depth, color_type), interlace in combinations:
crc = IHDR.compute_crc(width, height, bit_depth, color_type, interlace) crc = IHDR.compute_crc(width, height, bit_depth, color_type, interlace)
db.session.add( db.session.add(
@@ -152,6 +154,8 @@ def fill_ihdr_db() -> None:
count += 1 count += 1
if count % 5000 == 0: if count % 5000 == 0:
db.session.commit() db.session.commit()
finally:
db.session.autoflush = previous_autoflush
except SQLAlchemyError: except SQLAlchemyError:
db.session.rollback() db.session.rollback()