vtlengine 1.0.3rc3__py3-none-any.whl → 1.1rc1__py3-none-any.whl

vtlengine/AST/ASTConstructorModules/ExprComponents.py

@@ -31,7 +31,8 @@ class ExprComp(VtlVisitor):
 
                                 ExprComponent Definition.
 
-    _______________________________________________________________________________________"""
+    _______________________________________________________________________________________
+    """
 
     def visitExprComponent(self, ctx: Parser.ExprComponentContext):
         """
@@ -49,7 +50,7 @@ class ExprComp(VtlVisitor):
             | constant                                                                                              # constantExprComp
             | componentID                                                                                           # compId
         ;
-        """ # noqa E501
+        """  # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -280,7 +281,7 @@ class ExprComp(VtlVisitor):
     def visitCallComponent(self, ctx: Parser.CallComponentContext):
         """
         callFunction: operatorID LPAREN (parameterComponent (COMMA parameterComponent)*)? RPAREN    # callComponent
-        """ # noqa E501
+        """  # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -296,7 +297,7 @@ class ExprComp(VtlVisitor):
     def visitEvalAtomComponent(self, ctx: Parser.EvalAtomComponentContext):
         """
         | EVAL LPAREN routineName LPAREN (componentID|scalarItem)? (COMMA (componentID|scalarItem))* RPAREN (LANGUAGE STRING_CONSTANT)? (RETURNS outputParameterTypeComponent)? RPAREN      # evalAtomComponent
-        """ # noqa E501
+        """  # noqa E501
         ctx_list = list(ctx.getChildren())
 
         routine_name = Terminals().visitRoutineName(ctx_list[2])
@@ -347,7 +348,7 @@ class ExprComp(VtlVisitor):
     def visitCastExprComponent(self, ctx: Parser.CastExprComponentContext):
         """
         | CAST LPAREN exprComponent COMMA (basicScalarType|valueDomainName) (COMMA STRING_CONSTANT)? RPAREN         # castExprComponent
-        """ # noqa E501
+        """  # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -391,7 +392,6 @@ class ExprComp(VtlVisitor):
             raise NotImplementedError
 
     def visitParameterComponent(self, ctx: Parser.ParameterComponentContext):
-
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -582,16 +582,19 @@ class ExprComp(VtlVisitor):
             return self.visitDateDiffAtomComponent(ctx)
         elif isinstance(ctx, Parser.DateAddAtomComponentContext):
             return self.visitDateAddAtomComponentContext(ctx)
-        elif (isinstance(ctx, (
-            Parser.YearAtomComponentContext,
-            Parser.MonthAtomComponentContext,
-            Parser.DayOfMonthAtomComponentContext,
-            Parser.DayOfYearAtomComponentContext,
-            Parser.DayToYearAtomComponentContext,
-            Parser.DayToMonthAtomComponentContext,
-            Parser.YearToDayAtomComponentContext,
-            Parser.MonthToDayAtomComponentContext
-        ))):
+        elif isinstance(
+            ctx,
+            (
+                Parser.YearAtomComponentContext,
+                Parser.MonthAtomComponentContext,
+                Parser.DayOfMonthAtomComponentContext,
+                Parser.DayOfYearAtomComponentContext,
+                Parser.DayToYearAtomComponentContext,
+                Parser.DayToMonthAtomComponentContext,
+                Parser.YearToDayAtomComponentContext,
+                Parser.MonthToDayAtomComponentContext,
+            ),
+        ):
             return self.visitTimeUnaryAtomComponent(ctx)
         else:
             raise NotImplementedError
@@ -650,7 +653,7 @@ class ExprComp(VtlVisitor):
         """
         TIME_AGG LPAREN periodIndTo=STRING_CONSTANT (COMMA periodIndFrom=(STRING_CONSTANT| OPTIONAL ))?
         (COMMA op=optionalExprComponent)? (COMMA (FIRST|LAST))? RPAREN    # timeAggAtomComponent;
-        """ # noqa E501
+        """  # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -684,7 +687,11 @@ class ExprComp(VtlVisitor):
             # AST_ASTCONSTRUCTOR.17
             raise SemanticError("1-4-2-2")
         return TimeAggregation(
-            op=op, operand=operand_node, period_to=period_to, period_from=period_from, conf=conf
+            op=op,
+            operand=operand_node,
+            period_to=period_to,
+            period_from=period_from,
+            conf=conf,
         )
 
     def visitCurrentDateAtomComponent(self, ctx: Parser.CurrentDateAtomComponentContext):
@@ -825,7 +832,6 @@ class ExprComp(VtlVisitor):
     """
 
     def visitAnalyticFunctionsComponents(self, ctx: Parser.AnalyticFunctionsComponentsContext):
-
         if isinstance(ctx, Parser.AnSimpleFunctionComponentContext):
             return self.visitAnSimpleFunctionComponent(ctx)
         elif isinstance(ctx, Parser.LagOrLeadAnComponentContext):
@@ -861,7 +867,11 @@ class ExprComp(VtlVisitor):
                 raise NotImplementedError
 
         return Analytic(
-            op=op_node, operand=operand, partition_by=partition_by, order_by=order_by, window=params
+            op=op_node,
+            operand=operand,
+            partition_by=partition_by,
+            order_by=order_by,
+            window=params,
         )
 
     def visitLagOrLeadAnComponent(self, ctx: Parser.LagOrLeadAnComponentContext):
@@ -891,7 +901,11 @@ class ExprComp(VtlVisitor):
                 continue
 
         return Analytic(
-            op=op_node, operand=operand, partition_by=partition_by, order_by=order_by, params=params
+            op=op_node,
+            operand=operand,
+            partition_by=partition_by,
+            order_by=order_by,
+            params=params,
         )
 
     def visitRankAnComponent(self, ctx: Parser.RankAnComponentContext):
@@ -911,7 +925,11 @@ class ExprComp(VtlVisitor):
                 continue
 
         return Analytic(
-            op=op_node, operand=None, partition_by=partition_by, order_by=order_by, window=None
+            op=op_node,
+            operand=None,
+            partition_by=partition_by,
+            order_by=order_by,
+            window=None,
         )
 
     def visitRatioToReportAnComponent(self, ctx: Parser.RatioToReportAnComponentContext):
@@ -926,5 +944,9 @@ class ExprComp(VtlVisitor):
         partition_by = Terminals().visitPartitionByClause(ctx_list[5])
 
         return Analytic(
-            op=op_node, operand=operand, partition_by=partition_by, order_by=order_by, window=params
+            op=op_node,
+            operand=operand,
+            partition_by=partition_by,
+            order_by=order_by,
+            window=params,
         )

vtlengine/AST/ASTConstructorModules/Terminals.py

@@ -70,7 +70,6 @@ class Terminals(VtlVisitor):
         return var_id_node
 
     def visitVarIdExpr(self, ctx: Parser.VarIdExprContext):
-
         if isinstance(ctx.children[0], Parser.VarIDContext):
             return self.visitVarID(ctx.children[0])
 
@@ -127,7 +126,10 @@ class Terminals(VtlVisitor):
         valueDomainID: IDENTIFIER ;
         """
         return Collection(
-            name=ctx.children[0].getSymbol().text, children=[], kind="ValueDomain", type=""
+            name=ctx.children[0].getSymbol().text,
+            children=[],
+            kind="ValueDomain",
+            type="",
         )
 
     def visitRulesetID(self, ctx: Parser.RulesetIDContext):
@@ -381,7 +383,7 @@ class Terminals(VtlVisitor):
             | DATAPOINT_ON_VD  (GLPAREN  valueDomainName (MUL valueDomainName)*  GRPAREN )?         # dataPointVd
             | DATAPOINT_ON_VAR  (GLPAREN  varID (MUL varID)*  GRPAREN )?                            # dataPointVar
         ;
-        """ # noqa E501
+        """  # noqa E501
         # AST_ASTCONSTRUCTOR.54
         raise NotImplementedError
 
@@ -391,7 +393,7 @@ class Terminals(VtlVisitor):
             | HIERARCHICAL_ON_VD ( GLPAREN  vdName=IDENTIFIER (LPAREN valueDomainName (MUL valueDomainName)* RPAREN)?  GRPAREN )?   # hrRulesetVdType
             | HIERARCHICAL_ON_VAR ( GLPAREN  varName=varID (LPAREN  varID (MUL varID)* RPAREN)?  GRPAREN )?                         # hrRulesetVarType
         ;
-        """ # noqa E501
+        """  # noqa E501
         # AST_ASTCONSTRUCTOR.55
         raise NotImplementedError
 
@@ -483,7 +485,6 @@ class Terminals(VtlVisitor):
             raise NotImplementedError
 
     def visitScalarItem(self, ctx: Parser.ScalarItemContext):
-
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -497,7 +498,7 @@ class Terminals(VtlVisitor):
     def visitScalarWithCast(self, ctx: Parser.ScalarWithCastContext):
         """
         |  CAST LPAREN constant COMMA (basicScalarType) (COMMA STRING_CONSTANT)? RPAREN    #scalarWithCast  # noqa E501
-        """ # noqa E501
+        """  # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -593,7 +594,7 @@ class Terminals(VtlVisitor):
 
     def visitSignature(self, ctx: Parser.SignatureContext, kind="ComponentID"):
         """
-        varID (AS alias)?
+        VarID (AS alias)?
         """
 
         ctx_list = list(ctx.getChildren())
@@ -672,8 +673,12 @@ class Terminals(VtlVisitor):
         first = num_rows_1  # unbounded (default value)
         second = num_rows_2  # current data point (default value)
 
-        if (mode_2 == "preceding" and mode_1 == "preceding" and num_rows_1 == -1
-                and num_rows_2 == -1):  # preceding and preceding (error)
+        if (
+            mode_2 == "preceding"
+            and mode_1 == "preceding"
+            and num_rows_1 == -1
+            and num_rows_2 == -1
+        ):  # preceding and preceding (error)
             raise Exception(
                 f"Cannot have 2 preceding clauses with unbounded in analytic clause, "
                 f"line {ctx_list[3].start.line}"
@@ -706,7 +711,8 @@ class Terminals(VtlVisitor):
             return OrderBy(component=self.visitComponentID(ctx_list[0]).value, order="asc")
 
         return OrderBy(
-            component=self.visitComponentID(ctx_list[0]).value, order=ctx_list[1].getSymbol().text
+            component=self.visitComponentID(ctx_list[0]).value,
+            order=ctx_list[1].getSymbol().text,
         )
 
     def visitLimitClauseItem(self, ctx: Parser.LimitClauseItemContext):
@@ -735,5 +741,9 @@ def create_windowing(win_mode, values, modes):
             values[e] = "CURRENT ROW"
 
     return Windowing(
-        type_=win_mode, start=values[0], stop=values[1], start_mode=modes[0], stop_mode=modes[1]
+        type_=win_mode,
+        start=values[0],
+        stop=values[1],
+        start_mode=modes[0],
+        stop_mode=modes[1],
     )

vtlengine/AST/ASTEncoders.py

@@ -1,6 +1,6 @@
 import json
 
-import AST
+from vtlengine import AST
 
 
 class ComplexEncoder(json.JSONEncoder):

vtlengine/AST/DAG/__init__.py

@@ -300,7 +300,6 @@ class DAGAnalyzer(ASTTemplate):
         self.visit(node.right)
 
     def visit_RegularAggregation(self, node: RegularAggregation) -> None:
-
         self.visit(node.dataset)
         for child in node.children:
             self.isFromRegularAggregation = True
@@ -329,7 +328,6 @@ class DAGAnalyzer(ASTTemplate):
             self.inputs.append(node.value)
 
     def visit_ParamOp(self, node: ParamOp) -> None:
-
         if self.udos and node.op in self.udos:
             DO_AST: Operator = self.udos[node.op]
 
@@ -426,7 +424,6 @@ class HRDAGAnalyzer(DAGAnalyzer):
         # def visit_Identifier(self, node: Identifier) -> None:
         if node.kind == "CodeItemID":  # and node.value not in self.alias:
             if self.isFirstAssignment:
-
                 self.isFirstAssignment = False
                 self.outputs.append(node.value)
             else:

vtlengine/AST/Grammar/lexer.py

@@ -1116,7 +1116,6 @@ def serializedATN():
 
 
 class Lexer(Lexer):
-
     atn = ATNDeserializer().deserialize(serializedATN())
 
     decisionsToDFA = [DFA(ds, i) for i, ds in enumerate(atn.decisionToState)]