aboutsummaryrefslogtreecommitdiffstats
path: root/gunicorn_config.py
diff options
context:
space:
mode:
authorGravatar Christopher Baklid <[email protected]>2018-02-06 22:32:22 +0100
committerGravatar Christopher Baklid <[email protected]>2018-02-06 22:32:22 +0100
commit8436882783fb2fed76d9a2bde86e66fbe30ec30c (patch)
tree31941d8dbbbc1e3c76747274c369a54d663012a7 /gunicorn_config.py
parentFix template loading (diff)
adds gunicorn server_ready hook to set up database
Diffstat (limited to 'gunicorn_config.py')
-rw-r--r--gunicorn_config.py44
1 files changed, 44 insertions, 0 deletions
diff --git a/gunicorn_config.py b/gunicorn_config.py
new file mode 100644
index 00000000..c4869dd3
--- /dev/null
+++ b/gunicorn_config.py
@@ -0,0 +1,44 @@
+def when_ready(server):
+ """ server hook that only runs when the gunicorn master process loads """
+
+ import traceback
+ import rethinkdb as r
+
+ try:
+ server.log.info("rethinkdb initialising")
+
+ construct = [
+ {'DB':'test', 'TABLE':'test', 'INDEXES': ['test']},
+ ]
+ for struct in construct:
+ DB = struct['DB']
+ TABLE = struct['TABLE']
+ INDEXES = struct['INDEXES']
+
+ conn = r.connect(host='pdrdb', port=28016, db=DB)
+
+ # Check if database exists, if not create it
+ db_exists = r.db_list().contains(DB).run(conn)
+ if not db_exists:
+ log.info('adding database {0}'.format(DB))
+ r.db_create(DB).run(conn)
+
+ # Check if table exist, if not create it
+ table_exists = r.db(DB).table_list().contains(TABLE).run(conn)
+ if not table_exists:
+ log.info('adding table {0}'.format(TABLE))
+ result = r.db(DB).table_create(TABLE).run(conn)
+
+ # Check if index exists if not add it
+ rtable = r.db(DB).table(TABLE)
+ current_indexes = rtable.index_list().run(conn)
+ for index in INDEXES:
+ if index not in current_indexes:
+ log.info('adding index {0}'.format(index))
+ rtable.index_create(index).run(conn)
+
+ server.log.info("rethinkdb ready")
+
+ except:
+ server.log.error(traceback.format_exc())
+ server.log.error("rethinkdb failed to initialise")