pydpm_xl 0.2.5rc1__py3-none-any.whl → 0.2.5rc2__py3-none-any.whl

py_dpm/__init__.py

@@ -41,7 +41,7 @@ Available packages:
 - pydpm.api: Main APIs for migration, syntax, and semantic analysis
 """
 
-__version__ = "0.2.5rc1"
+__version__ = "0.2.5rc2"
 __author__ = "MeaningfulData S.L."
 __email__ = "info@meaningfuldata.eu"
 __license__ = "GPL-3.0-or-later"

py_dpm/api/dpm_xl/ast_generator.py

@@ -1039,6 +1039,81 @@ class ASTGeneratorAPI:
         extract_from_node(ast_dict)
         return all_variables, variables_by_table
 
+    def _extract_time_shifts_by_table(self, expression: str) -> Dict[str, str]:
+        """
+        Extract time shift information for each table in the expression.
+
+        Uses the AST to properly parse the expression and find TimeShiftOp nodes
+        to determine the ref_period for each table reference.
+
+        Args:
+            expression: DPM-XL expression
+
+        Returns:
+            Dict mapping table codes to ref_period values (e.g., {"C_01.00": "T-1Q"})
+            Tables without time shifts default to "T".
+        """
+        from py_dpm.dpm_xl.ast.template import ASTTemplate
+
+        time_shifts = {}
+        current_period = ["t"]  # Use list to allow mutation in nested function
+
+        class TimeShiftExtractor(ASTTemplate):
+            """Lightweight AST visitor that extracts time shifts for each table."""
+
+            def visit_TimeShiftOp(self, node):
+                # Save current time period and compute new one
+                previous_period = current_period[0]
+
+                period_indicator = node.period_indicator
+                shift_number = node.shift_number
+
+                # Compute time period (same logic as ModuleDependencies)
+                if "-" in str(shift_number):
+                    current_period[0] = f"t+{period_indicator}{shift_number}"
+                else:
+                    current_period[0] = f"t-{period_indicator}{shift_number}"
+
+                # Visit operand (which contains the VarID)
+                self.visit(node.operand)
+
+                # Restore previous time period
+                current_period[0] = previous_period
+
+            def visit_VarID(self, node):
+                if node.table and current_period[0] != "t":
+                    time_shifts[node.table] = current_period[0]
+
+        def convert_to_ref_period(internal_period: str) -> str:
+            """Convert internal time period format to ref_period format.
+
+            Internal format: "t+Q-1" or "t-Q1"
+            Output format: "T-1Q" for one quarter back
+            """
+            if internal_period.startswith("t+"):
+                # e.g., "t+Q-1" -> "T-1Q"
+                indicator = internal_period[2]
+                number = internal_period[3:]
+                if number.startswith("-"):
+                    return f"T{number}{indicator}"
+                return f"T+{number}{indicator}"
+            elif internal_period.startswith("t-"):
+                # e.g., "t-Q1" -> "T-1Q"
+                indicator = internal_period[2]
+                number = internal_period[3:]
+                return f"T-{number}{indicator}"
+            return "T"
+
+        try:
+            ast = self.syntax_api.parse_expression(expression)
+            extractor = TimeShiftExtractor()
+            extractor.visit(ast)
+
+            return {table: convert_to_ref_period(period) for table, period in time_shifts.items()}
+
+        except Exception:
+            return {}
+
     def _detect_cross_module_dependencies(
         self,
         expression: str,
@@ -1075,7 +1150,7 @@ class ASTGeneratorAPI:
         )
 
         try:
-            # Get tables with module info
+            # Get tables with module info (includes module_version)
             tables_with_modules = scopes_api.get_tables_with_metadata_from_expression(
                 expression=expression,
                 release_id=release_id
@@ -1091,10 +1166,35 @@ class ASTGeneratorAPI:
             if scope_result.has_error or not scope_result.is_cross_module:
                 return {}, []
 
+            # Extract time shifts for each table from expression
+            time_shifts_by_table = self._extract_time_shifts_by_table(expression)
+
             # Determine primary module from first table if not provided
             if primary_module_vid is None and tables_with_modules:
                 primary_module_vid = tables_with_modules[0].get("module_vid")
 
+            # Helper to normalize table code (remove 't' prefix if present)
+            def normalize_table_code(code: str) -> str:
+                return code[1:] if code and code.startswith('t') else code
+
+            # Helper to lookup ref_period for a table
+            def get_ref_period(table_code: str) -> str:
+                if not table_code:
+                    return "T"
+                ref = time_shifts_by_table.get(table_code)
+                if not ref:
+                    ref = time_shifts_by_table.get(normalize_table_code(table_code))
+                return ref or "T"
+
+            # Helper to lookup variables for a table
+            def get_table_variables(table_code: str) -> dict:
+                if not table_code:
+                    return {}
+                variables = variables_by_table.get(table_code)
+                if not variables:
+                    variables = variables_by_table.get(f"t{table_code}", {})
+                return variables or {}
+
             # Group external tables by module
             external_modules = {}
             for table_info in tables_with_modules:
@@ -1106,36 +1206,38 @@ class ASTGeneratorAPI:
                 if not module_code:
                     continue
 
-                # Get module URI using existing ExplorerQuery
+                # Get module URI
                 try:
                     module_uri = ExplorerQuery.get_module_url(
                         scopes_api.session,
                         module_code=module_code,
                         release_id=release_id,
                     )
-                    # Remove .json suffix if present (for consistency with expected format)
                     if module_uri.endswith(".json"):
                         module_uri = module_uri[:-5]
                 except Exception:
                     continue
 
+                table_code = table_info.get("code")
+                ref_period = get_ref_period(table_code)
+
                 if module_uri not in external_modules:
                     external_modules[module_uri] = {
                         "module_vid": module_vid,
+                        "module_version": table_info.get("module_version"),  # Already in table_info
+                        "ref_period": ref_period,
                         "tables": {},
                         "variables": {},
                         "from_date": None,
                         "to_date": None
                     }
+                elif ref_period != "T":
+                    # Keep most specific ref_period (non-T takes precedence)
+                    external_modules[module_uri]["ref_period"] = ref_period
 
-                # Add table - get variables from variables_by_table
-                table_code = table_info.get("code")
+                # Add table and variables
                 if table_code:
-                    # Look up variables in variables_by_table (handles t prefix)
-                    table_variables = variables_by_table.get(table_code, {})
-                    if not table_variables:
-                        # Try with t prefix
-                        table_variables = variables_by_table.get(f"t{table_code}", {})
+                    table_variables = get_table_variables(table_code)
                     external_modules[module_uri]["tables"][table_code] = {
                         "variables": table_variables,
                         "open_keys": {}
@@ -1169,11 +1271,16 @@ class ASTGeneratorAPI:
                 # cross_instance_dependencies entry (one per external module)
                 from_date = data["from_date"]
                 to_date = data["to_date"]
+                module_entry = {
+                    "URI": uri,
+                    "ref_period": data["ref_period"]
+                }
+                # Add module_version if available
+                if data["module_version"]:
+                    module_entry["module_version"] = data["module_version"]
+
                 cross_instance_dependencies.append({
-                    "modules": [{
-                        "URI": uri,
-                        "ref_period": "T"
-                    }],
+                    "modules": [module_entry],
                     "affected_operations": [operation_code],
                     "from_reference_date": str(from_date) if from_date else "",
                     "to_reference_date": str(to_date) if to_date else ""

{pydpm_xl-0.2.5rc1.dist-info → pydpm_xl-0.2.5rc2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pydpm_xl
-Version: 0.2.5rc1
+Version: 0.2.5rc2
 Summary: Python library for DPM-XL data processing and analysis
 Author-email: "MeaningfulData S.L." <info@meaningfuldata.eu>
 License: GPL-3.0-or-later

{pydpm_xl-0.2.5rc1.dist-info → pydpm_xl-0.2.5rc2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-py_dpm/__init__.py,sha256=GmaGEIVu3IEt3i3UkrQJSpoLu_ENSpiSt_4rBceOjpc,1861
+py_dpm/__init__.py,sha256=u6lHTsVyMlEyPdrczeNUNlN8D7Dge-Cr0MPlZs8iIqo,1861
 py_dpm/api/__init__.py,sha256=n79vAD7qatlYaXaI2N5IAD9m_8Fgb00EOdapVXZYTpI,1081
 py_dpm/api/dpm/__init__.py,sha256=HQflgiRbs1eDi3KTadNhxS1NoaG6PGQDVMvFnuIEfXo,506
 py_dpm/api/dpm/data_dictionary.py,sha256=g0h6Yfschz7rboYly9LTbP-2SS5UxltU3AXu0v0tqrU,29457
@@ -7,7 +7,7 @@ py_dpm/api/dpm/hierarchical_queries.py,sha256=X4AbpsWy3iItOTVIdVbtaTmRgOHPf0Y64I
 py_dpm/api/dpm/instance.py,sha256=v3DWzdaM5gPCecLjwjZ49FGfqZzUR3dPC0U8zGwdttk,3795
 py_dpm/api/dpm/migration.py,sha256=9FT7zzz4QdUIRR6MD01gMODBtfq9HH_RF4hRgZqMcZc,2404
 py_dpm/api/dpm_xl/__init__.py,sha256=aRjaMAf_i2a33UAGTg-TF1BfO6miOOrbCydTUqAVvRU,910
-py_dpm/api/dpm_xl/ast_generator.py,sha256=_ir-LP3hgT8diUFt9sEcXsh3cQhA_YiSSFe4a03zjuA,51050
+py_dpm/api/dpm_xl/ast_generator.py,sha256=w7aeRDGJ-ggJoRXnZ-Tn1rEeb1zoSF21co9pSV_KU6A,55327
 py_dpm/api/dpm_xl/complete_ast.py,sha256=VkmcBatrydu97Inwp3pHjz93F38q2JRo-4Lohdu30RY,7684
 py_dpm/api/dpm_xl/operation_scopes.py,sha256=7AyOFAn9h012JPF9H5EtZ3sPzv6DOxkoinpj5ArzVOc,48492
 py_dpm/api/dpm_xl/semantic.py,sha256=Buo_t-sEv65r6RmYDy1xkCWGlU2pB2WQsDM-X-FX4cc,13629
@@ -76,9 +76,9 @@ py_dpm/exceptions/exceptions.py,sha256=6S3p-_i5O1oStvSMixt_JQG0xwTeSfBcdzrwL8yBy
 py_dpm/exceptions/messages.py,sha256=UwY6QIK8c-POcDCc9HYbZFGArCIYAanUGNh2LNKPx3U,7534
 py_dpm/instance/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 py_dpm/instance/instance.py,sha256=OPSEPgSYAxhgqhKuxbMpMPTfBnaFNzURTrUUT4kvGKc,10820
-pydpm_xl-0.2.5rc1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-pydpm_xl-0.2.5rc1.dist-info/METADATA,sha256=X7pPwj8tY9-p6ApsGXKW533zyekcZHHqRf9w1Ld9Ar4,9305
-pydpm_xl-0.2.5rc1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pydpm_xl-0.2.5rc1.dist-info/entry_points.txt,sha256=6DDmBfw-AjtgvMHgq_I730i_LAAs_7-N3C95HD_bRr4,47
-pydpm_xl-0.2.5rc1.dist-info/top_level.txt,sha256=495PvWZRoKl2NvbQU25W7dqWIBHqY-mFMPt83uxPpcM,7
-pydpm_xl-0.2.5rc1.dist-info/RECORD,,
+pydpm_xl-0.2.5rc2.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pydpm_xl-0.2.5rc2.dist-info/METADATA,sha256=wtWYaml5W9vzWfDQTiUI3x_EByIduVBZV6bo6fLu31A,9305
+pydpm_xl-0.2.5rc2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pydpm_xl-0.2.5rc2.dist-info/entry_points.txt,sha256=6DDmBfw-AjtgvMHgq_I730i_LAAs_7-N3C95HD_bRr4,47
+pydpm_xl-0.2.5rc2.dist-info/top_level.txt,sha256=495PvWZRoKl2NvbQU25W7dqWIBHqY-mFMPt83uxPpcM,7
+pydpm_xl-0.2.5rc2.dist-info/RECORD,,