[DayZ Parameters] DayZ parameter validation does not validate DAYZ_SPEC_VERSION (#2713)

Author: R-Palazzo

sdv/multi_table/_dayz_utils.py

@@ -1,9 +1,16 @@
 """Multi-Table DayZ parameter detection and creation."""
 
+import json
+
+import pandas as pd
+
 from sdv.cag._utils import _is_list_of_type
 from sdv.errors import SynthesizerInputError, SynthesizerProcessingError
-from sdv.multi_table._dayz_utils import create_parameters_multi_table
-from sdv.single_table.dayz import _validate_parameter_structure, _validate_tables_parameter
+from sdv.single_table.dayz import (
+    _validate_parameter_structure,
+    _validate_tables_parameter,
+    create_parameters,
+)
 
 REQUIRED_RELATIONSHIP_KEYS = [
     'parent_table_name',
@@ -19,6 +26,53 @@
 DEFAULT_NUM_ROWS = 1000
 
 
+def _detect_relationship_parameters(data, metadata):
+    """Detect all relationship-level for the DayZ parameters.
+
+    The relationship-level parameters are:
+    - The min and max cardinality
+
+    Args:
+        data (dict[str, pd.DataFrame]): The input data.
+        metadata (Metadata): The metadata object.
+
+    Returns:
+        dict: A list containing the detected parameters.
+    """
+    relationship_parameters = []
+    for relationship in metadata.relationships:
+        rel_tuple = (
+            relationship['parent_table_name'],
+            relationship['child_table_name'],
+            relationship['parent_primary_key'],
+            relationship['child_foreign_key'],
+        )
+        cardinality_table = pd.DataFrame(index=data[rel_tuple[0]][rel_tuple[2]].copy())
+        cardinality_table['cardinality'] = data[rel_tuple[1]][rel_tuple[3]].value_counts()
+        cardinality_table = cardinality_table.fillna(0)
+        relationship_parameters.append({
+            'parent_table_name': rel_tuple[0],
+            'child_table_name': rel_tuple[1],
+            'parent_primary_key': rel_tuple[2],
+            'child_foreign_key': rel_tuple[3],
+            'min_cardinality': cardinality_table['cardinality'].min(),
+            'max_cardinality': cardinality_table['cardinality'].max(),
+        })
+
+    return relationship_parameters
+
+
+def create_parameters_multi_table(data, metadata, output_filename):
+    """Create parameters for the DayZSynthesizer."""
+    parameters = create_parameters(data, metadata, None)
+    parameters['relationships'] = _detect_relationship_parameters(data, metadata)
+    if output_filename:
+        with open(output_filename, 'w') as f:
+            json.dump(parameters, f, indent=4)
+
+    return parameters
+
+
 def _validate_min_cardinality(relationship):
     min_cardinality = relationship['min_cardinality']
     if not isinstance(min_cardinality, int) or min_cardinality < 0: