Bryan Van Deusen
55 lines
1.6 KiB
Bash
55 lines
1.6 KiB
Bash
#!/usr/bin/env sh
|
|
set -e
|
|
|
|
# Ensure Flask CLI finds your app
|
|
export FLASK_APP=run.py
|
|
export FLASK_ENV=production
|
|
|
|
# If arguments are passed and first arg is NOT "web", execute them directly
|
|
# This allows: docker run <image> celery -A app.celery_app:celery worker ...
|
|
if [ $# -gt 0 ] && [ "$1" != "web" ]; then
|
|
exec "$@"
|
|
fi
|
|
|
|
# --- Web service mode (default or explicit "web" argument) ---
|
|
|
|
# --- Migrations with simple retry (no pg_isready needed) ---
|
|
if [ "${RUN_DB_MIGRATIONS:-1}" = "1" ]; then
|
|
ATTEMPTS="${MIGRATION_MAX_ATTEMPTS:-30}" # ~90s default with 3s sleep
|
|
SLEEP_SECS="${MIGRATION_RETRY_SLEEP:-3}"
|
|
i=1
|
|
echo "Applying migrations (up to $ATTEMPTS attempts)..."
|
|
until flask db upgrade; do
|
|
if [ "$i" -ge "$ATTEMPTS" ]; then
|
|
echo "ERROR: Migrations failed after $ATTEMPTS attempts."
|
|
exit 1
|
|
fi
|
|
echo "DB not ready yet; retrying in ${SLEEP_SECS}s ($i/$ATTEMPTS)…"
|
|
sleep "$SLEEP_SECS"
|
|
i=$((i+1))
|
|
done
|
|
|
|
# Run version check and data migrations (e.g., phash recomputation)
|
|
echo "Checking app version and running data migrations..."
|
|
flask version-check
|
|
fi
|
|
|
|
# --- Start Gunicorn ---
|
|
GUNICORN_BIND="${GUNICORN_BIND:-0.0.0.0:5000}"
|
|
GUNICORN_WORKERS="${GUNICORN_WORKERS:-8}"
|
|
GUNICORN_THREADS="${GUNICORN_THREADS:-4}"
|
|
GUNICORN_TIMEOUT="${GUNICORN_TIMEOUT:-180}"
|
|
GUNICORN_OPTS="${GUNICORN_OPTS:-}"
|
|
|
|
echo "Starting Gunicorn on ${GUNICORN_BIND} …"
|
|
exec gunicorn \
|
|
--bind "${GUNICORN_BIND}" \
|
|
--workers "${GUNICORN_WORKERS}" \
|
|
--worker-class gthread \
|
|
--threads "${GUNICORN_THREADS}" \
|
|
--timeout "${GUNICORN_TIMEOUT}" \
|
|
--access-logfile - \
|
|
--error-logfile - \
|
|
${GUNICORN_OPTS} \
|
|
run:app
|