darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 07:35:34 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #2242 from alphagov/enable-sa-preping

Browse Source 
Enable pessimistic DB connection disconnect handling
This commit is contained in:
Alexey Bezhan
2018-11-30 16:18:31 +00:00
committed by GitHub
parent 15849b421b 614a2dae2c
commit 32f0236ff2
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
app/__init__.py
View File

@@ -4,7 +4,7 @@ import string
import uuid import uuid
from flask import _request_ctx_stack, request, g, jsonify from flask import _request_ctx_stack, request, g, jsonify
from flask_sqlalchemy import SQLAlchemy from flask_sqlalchemy import SQLAlchemy as _SQLAlchemy
from flask_marshmallow import Marshmallow from flask_marshmallow import Marshmallow
from flask_migrate import Migrate from flask_migrate import Migrate
from time import monotonic from time import monotonic
@@ -27,6 +27,13 @@ from app.encryption import Encryption
DATETIME_FORMAT = "%Y-%m-%dT%H:%M:%S.%fZ" DATETIME_FORMAT = "%Y-%m-%dT%H:%M:%S.%fZ"
DATE_FORMAT = "%Y-%m-%d" DATE_FORMAT = "%Y-%m-%d"
class SQLAlchemy(_SQLAlchemy):
def apply_pool_defaults(self, app, options):
_SQLAlchemy.apply_pool_defaults(self, app, options)
options["pool_pre_ping"] = True
db = SQLAlchemy() db = SQLAlchemy()
migrate = Migrate() migrate = Migrate()
ma = Marshmallow() ma = Marshmallow()