Skip database initialization when running as RQ worker

This commit is contained in:
aradhyacp
2026-01-04 01:27:12 +05:30
parent 370405ab24
commit 4faf7fd9b5
+11 -5
View File
@@ -4,6 +4,7 @@
"""This module defines the database models for the Aperi'Solve application.""" """This module defines the database models for the Aperi'Solve application."""
import itertools import itertools
import sys
from datetime import datetime, timezone from datetime import datetime, timezone
from os import getenv from os import getenv
from pathlib import Path from pathlib import Path
@@ -132,6 +133,13 @@ def init_db(app: Flask) -> None:
- Populates the IHDR CRC lookup table with initial data - Populates the IHDR CRC lookup table with initial data
- Prints status messages to console during initialization - Prints status messages to console during initialization
""" """
# Detect if running as RQ worker (not wsgi/web)
is_worker = len(sys.argv) > 0 and ("rq" in sys.argv[0] or "rq.worker" in sys.modules)
if is_worker:
print("Running as worker, skipping database initialization.")
return
with app.app_context(): with app.app_context():
if getenv("CLEAR_AT_RESTART", "0") == "1": # Force clear if CLEAR_AT_RESTART if getenv("CLEAR_AT_RESTART", "0") == "1": # Force clear if CLEAR_AT_RESTART
print("Clearing database and file system at restart...") print("Clearing database and file system at restart...")
@@ -142,8 +150,6 @@ def init_db(app: Flask) -> None:
db.create_all() db.create_all()
print("Database structure created successfully.") print("Database structure created successfully.")
if getenv("SKIP_IHDR_FILL", "0") == "1": print("Filling IHDR lookup table...")
print("Skipping IHDR lookup table fill as per configuration.") fill_ihdr_db()
else:
print("Filling IHDR lookup table...")
fill_ihdr_db()