Add readiness and liveness probe scripts.

.s2i/environment

@@ -2,3 +2,4 @@ S2I_SCRIPTS_PATH=/usr/libexec/s2i
 S2I_BASH_ENV=/opt/app-root/etc/scl_enable
 DISABLE_COLLECTSTATIC=1
 DISABLE_MIGRATE=1
+PYTHONUNBUFFERED=1

requirements.txt

@@ -3,3 +3,4 @@ dj_database_url
 mod_wsgi
 Pillow
 psycopg2
+requests

scripts/alive

@@ -0,0 +1,30 @@
+#!/bin/bash
+
+set -eo pipefail
+
+HERE=`dirname $0`
+
+ready() {
+    echo " -----> Starting liveness script."
+
+    # Check web server.
+
+    echo " -----> Checking server is working."
+
+    python $HERE/check-server.py
+
+    # Check database.
+
+    echo " -----> Checking database is working."
+
+    if [ x"$DATABASE_URL" != x"" ]; then
+        (cat - | python manage.py shell --plain) << !
+import runpy
+_ = runpy.run_path('$HERE/check-database.py')
+!
+    fi
+
+    echo " -----> Stopping liveness script."
+}
+
+ready 2>&1 | tee -a /proc/1/fd/1

scripts/check-database.py

@@ -0,0 +1,37 @@
+from __future__ import print_function
+
+import sys
+import time
+
+import django
+
+django.setup()
+
+from django.db import connection
+
+remaining = 600.0
+delay = 2.0
+
+success = False
+
+while remaining > 0.0:
+    try:
+        print('Check whether database is ready...')
+        cursor = connection.cursor()
+        with cursor:
+            cursor.execute('SELECT 1')
+            cursor.fetchall()
+
+        success = True
+
+        break
+
+    except Exception:
+        pass
+
+    time.sleep(delay)
+
+if not success:
+    print('Failed to connect to database')
+
+    sys.exit(1)

scripts/check-server.py

@@ -0,0 +1,7 @@
+import os
+import requests
+
+hostname = os.environ['HOSTNAME']
+
+response = requests.get('http://%s:8080/' % hostname)
+response.raise_for_status()

scripts/ready

@@ -0,0 +1,30 @@
+#!/bin/bash
+
+set -eo pipefail
+
+HERE=`dirname $0`
+
+ready() {
+    echo " -----> Starting readiness script."
+
+    # Check web server.
+
+    echo " -----> Checking server is working."
+
+    python $HERE/check-server.py
+
+    # Check database.
+
+    echo " -----> Checking database is working."
+
+    if [ x"$DATABASE_URL" != x"" ]; then
+        (cat - | python manage.py shell --plain) << !
+import runpy
+_ = runpy.run_path('$HERE/check-database.py')
+!
+    fi
+
+    echo " -----> Stopping readiness script."
+}
+
+ready 2>&1 | tee -a /proc/1/fd/1