Skip to content

Commit afc17d7

Browse files
wwwjfyfantix
authored andcommitted
fix python-gino/gino#569, be compatible with Sanic 19.9.0
1 parent 17915fb commit afc17d7

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

gino/ext/sanic.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,11 +84,18 @@ def init_app(self, app):
8484
if app.config.setdefault('DB_USE_CONNECTION_FOR_REQUEST', True):
8585
@app.middleware('request')
8686
async def on_request(request):
87-
request['connection'] = await self.acquire(lazy=True)
87+
conn = await self.acquire(lazy=True)
88+
if hasattr(request, 'ctx'):
89+
request.ctx.connection = conn
90+
else:
91+
request['connection'] = conn
8892

8993
@app.middleware('response')
9094
async def on_response(request, _):
91-
conn = request.pop('connection', None)
95+
if hasattr(request, 'ctx'):
96+
conn = getattr(request.ctx, 'connection', None)
97+
else:
98+
conn = request.pop('connection', None)
9299
if conn is not None:
93100
await conn.release()
94101

0 commit comments

Comments
 (0)