carrotquest/django-clickhouse
1
1
Fork 0
mirror of https://github.com/carrotquest/django-clickhouse.git synced 2024-11-22 09:06:43 +03:00
Code Issues Packages Projects Releases Wiki Activity

Bugfix in QuerySetMixin inheritance

Browse Source
This commit is contained in:
M1ha 2019-09-11 15:35:43 +05:00
parent 0f1364b2b8
commit 6493373c8f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/django_clickhouse/models.py
View File

@ -75,14 +75,14 @@ class ClickHouseSyncBulkUpdateQuerySetMixin(ClickHouseSyncRegisterMixin, BulkUpd
original_returning = kwargs.pop('returning', None)
kwargs['returning'] = self._update_returning_param(original_returning)
result = super().bulk_update(*args, **kwargs)
self._register_ops(result)
self._register_ops('update', result)
return result.count() if original_returning is None else result
def bulk_update_or_create(self, *args, **kwargs):
original_returning = kwargs.pop('returning', None)
kwargs['returning'] = self._update_returning_param(original_returning)
result = super().bulk_update_or_create(*args, **kwargs)
self._register_ops(result)
self._register_ops('update', result)
return result.count() if original_returning is None else result