diff --git a/taiga/base/utils/db.py b/taiga/base/utils/db.py index 809c9d9c..cd4a7a15 100644 --- a/taiga/base/utils/db.py +++ b/taiga/base/utils/db.py @@ -16,11 +16,13 @@ # You should have received a copy of the GNU Affero General Public License # along with this program. If not, see . +from django.conf import settings from django.contrib.contenttypes.models import ContentType from django.db import connection from django.db import DatabaseError from django.db import transaction from django.shortcuts import _get_queryset +from taiga.celery import app from . import functions @@ -122,9 +124,9 @@ def update_in_bulk(instances, list_of_new_values, callback=None, precall=None): instance.save() callback(instance) - +@app.task @transaction.atomic -def update_attr_in_bulk_for_ids(values, attr, model): +def _update_attr_in_bulk_for_ids(values, attr, model): """Update a table using a list of ids. :params values: Dict of new values where the key is the pk of the element to update. @@ -162,6 +164,14 @@ def update_attr_in_bulk_for_ids(values, attr, model): transaction.on_commit(_run_sql) +@transaction.atomic +def update_attr_in_bulk_for_ids(values, attr, model): + if settings.CELERY_ENABLED: + _update_attr_in_bulk_for_ids.delay(values, attr, model) + else: + _update_attr_in_bulk_for_ids(values, attr, model) + + def to_tsquery(term): """ Based on: https://gist.github.com/wolever/1a5ccf6396f00229b2dc