2525
2626from app .translator .core .exceptions .functions import NotSupportedFunctionException
2727from app .translator .core .mapping import SourceMapping
28- from app .translator .core .models .field import Alias ,Field
2928from app .translator .core .models .functions .base import Function ,ParsedFunctions ,RenderedFunctions
29+ from app .translator .core .models .query_tokens .field import Alias ,Field
3030from app .translator .tools .utils import execute_module
3131from settings import INIT_FUNCTIONS
3232
@@ -83,7 +83,6 @@ def parse(self, func_body: str, raw: str) -> Function:
8383class FunctionRender (ABC ):
8484function_names_map :ClassVar [dict [str ,str ]]= {}
8585order_to_render :int = 0
86- in_query_render :bool = False
8786render_to_prefix :bool = False
8887manager :PlatformFunctionsManager = None
8988
@@ -117,7 +116,6 @@ def __init__(self):
117116self ._parsers_map :dict [str ,FunctionParser ]= {}# {platform_func_name: FunctionParser}
118117
119118self ._renders_map :dict [str ,FunctionRender ]= {}# {generic_func_name: FunctionRender}
120- self ._in_query_renders_map :dict [str ,FunctionRender ]= {}# {generic_func_name: FunctionRender}
121119self ._order_to_render :dict [str ,int ]= {}# {generic_func_name: int}
122120
123121def register_render (self ,render_class :type [FunctionRender ])-> type [FunctionRender ]:
@@ -126,8 +124,6 @@ def register_render(self, render_class: type[FunctionRender]) -> type[FunctionRe
126124for generic_function_name in render .function_names_map :
127125self ._renders_map [generic_function_name ]= render
128126self ._order_to_render [generic_function_name ]= render .order_to_render
129- if render .in_query_render :
130- self ._in_query_renders_map [generic_function_name ]= render
131127
132128return render_class
133129
@@ -149,24 +145,16 @@ def get_hof_parser(self, platform_func_name: str) -> HigherOrderFunctionParser:
149145
150146raise NotSupportedFunctionException
151147
152- def get_parser (self ,platform_func_name :str )-> FunctionParser :
148+ def get_parser (self ,platform_func_name :str )-> Optional [ FunctionParser ] :
153149if INIT_FUNCTIONS and (parser := self ._parsers_map .get (platform_func_name )):
154150return parser
155151
156- raise NotSupportedFunctionException
157-
158152def get_render (self ,generic_func_name :str )-> FunctionRender :
159153if INIT_FUNCTIONS and (render := self ._renders_map .get (generic_func_name )):
160154return render
161155
162156raise NotSupportedFunctionException
163157
164- def get_in_query_render (self ,generic_func_name :str )-> FunctionRender :
165- if INIT_FUNCTIONS and (render := self ._in_query_renders_map .get (generic_func_name )):
166- return render
167-
168- raise NotSupportedFunctionException
169-
170158@property
171159def order_to_render (self )-> dict [str ,int ]:
172160if INIT_FUNCTIONS :