django-orchestra/orchestra/utils/db.py

16 lines
385 B
Python
Raw Normal View History

2015-05-03 17:44:46 +00:00
from django import db
def close_connection(execute):
""" Threads have their own connection pool, closing it when finishing """
def wrapper(*args, **kwargs):
try:
log = execute(*args, **kwargs)
except Exception as e:
pass
else:
wrapper.log = log
finally:
db.connection.close()
return wrapper