Infinidat/infi.clickhouse_orm
1
1
Fork 0
mirror of https://github.com/Infinidat/infi.clickhouse_orm.git synced 2024-11-24 18:03:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

Support FINAL for ReplacingMergeTree

Browse Source
This commit is contained in:
Christian Pedersen 2020-06-23 09:35:13 +02:00
parent 4616f8cb0e
commit 45552fdb97
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/infi/clickhouse_orm/query.py
View File

@ -3,7 +3,7 @@ from __future__ import unicode_literals
import pytz
from copy import copy, deepcopy
from math import ceil
from .engines import CollapsingMergeTree
from .engines import CollapsingMergeTree, ReplacingMergeTree
from datetime import date, datetime
from .utils import comma_join, string_or_func
@ -540,8 +540,8 @@ class QuerySet(object):
Adds a FINAL modifier to table, meaning data will be collapsed to final version.
Can be used with `CollapsingMergeTree` engine only.
"""
if not isinstance(self._model_cls.engine, CollapsingMergeTree):
raise TypeError('final() method can be used only with CollapsingMergeTree engine')
if not isinstance(self._model_cls.engine, (CollapsingMergeTree, ReplacingMergeTree)):
raise TypeError('final() method can be used only with the CollapsingMergeTree and ReplacingMergeTree engines')
qs = copy(self)
qs._final = True