Skip to content

Commit 90da0d7

Browse files
committed
Improve get_field_tokens in core parser
1 parent b40132a commit 90da0d7

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

uncoder-core/app/translator/core/parser.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from app.translator.core.models.platform_details import PlatformDetails
2929
from app.translator.core.models.query_container import RawQueryContainer, TokenizedQueryContainer
3030
from app.translator.core.models.query_tokens.field import Field
31+
from app.translator.core.models.query_tokens.field_field import FieldField
3132
from app.translator.core.models.query_tokens.field_value import FieldValue
3233
from app.translator.core.models.query_tokens.function_value import FunctionValue
3334
from app.translator.core.tokenizer import QueryTokenizer
@@ -68,6 +69,11 @@ def get_field_tokens(
6869
for token in query_tokens:
6970
if isinstance(token, FieldValue):
7071
field_tokens.append(token.field)
72+
elif isinstance(token, FieldField):
73+
if token.field_left:
74+
field_tokens.append(token.field_left)
75+
if token.field_right:
76+
field_tokens.append(token.field_right)
7177
elif isinstance(token, FunctionValue):
7278
field_tokens.extend(self.tokenizer.get_field_tokens_from_func_args([token.function]))
7379

0 commit comments

Comments
 (0)