CAV_to_constraint_arr.py

import arcpy
import pandas as pd
import os

fc = r"C:\\temp\\GasPipelineEnterpriseDataManagement\\Databases\\UPDM_UtilityNetwork.gdb\\UtilityNetwork\\PipelineLine"
field_group_name = 'Limit Material By Asset Type'


def view_cav(table, subtype_field):
    index = ['fieldGroupName', 'subtype', 'isRetired', 'id']
    data = {}
    for cav in arcpy.da.ListContingentValues(table):
        contingent_value = {k: getattr(cav, k, None) for k in index}
        for field in cav.values:
            contingent_value[field.name] = dict(CODED_VALUE=field.code,
                                                RANGE=field.range,
                                                ANY='|ANY|',
                                                NULL='<NULL>')[field.type]
        data.setdefault(cav.fieldGroupName, []).append(contingent_value)
    return [pd.DataFrame(values).set_index(index).rename_axis(index={'subtype': subtype_field}).fillna('<NULL>') for
            values in data.values()]


desc = arcpy.Describe(fc)


for df in view_cav(fc, desc.subtypeFieldName):
    if field_group_name in df.index:
        subtypes = set()
        valid_combos = []
        df = df.reset_index().drop(['fieldGroupName', 'id'], axis=1)
        df = df[df['isRetired'] == False].drop(['isRetired'], axis=1)
        for row in df.itertuples(index=False):
            valid_combos.append("::".join(map(str, row)).replace('<NULL>', ''))
            subtypes.add(str(row[0]))
        subtypes = sorted(subtypes)

        field_list = [f'$feature.{fld}' for fld in df.columns]
        func = f'''
// Assigned To: {os.path.basename(fc)}
// Type: Constraint
// Name: {field_group_name}
// Description: Limit values combinations using the fields {', '.join(list(df.columns))}
// Subtypes: All
// Error: 5601
// Error Message: Incompatible types for {', '.join(list(df.columns))}
// Trigger: Insert, Update


// *************       User Variables       *************
// This section has the functions and variables that need to be adjusted based on your implementation

var valid_asset_groups = [{', '.join(subtypes)}];
if (indexof(valid_asset_groups, $feature.{desc.subtypeFieldName}) == -1) {{
    return true;
}}

var feature_fields = [{', '.join(field_list)}];

var valid_values = {valid_combos};

// ************* End User Variables Section *************
function splice(arr, start, end) {{
    var new_arr = [];
    var k = 0;
    for (var i = start; i < end; i++) {{
        new_arr[k++] = arr[i];
    }}
    return new_arr;
}}

function join_array(a, b) {{
    var new_arr = [];
    var k = 0;
    for (var i in a) {{
        new_arr[k++] = a[i];
    }}
    for (var i in b) {{
        new_arr[k++] = b[i];
    }}
    return new_arr;
}}
var feature_values = [Concatenate(feature_fields, '::')];
var any_sent = '|ANY|';
var fld_count = Count(feature_fields);
for (var i = 0; i < fld_count; i++) {{
    var start_arr = splice(feature_fields, 0, i)
    start_arr[Count(start_arr)] = any_sent;
    var end_arr = splice(feature_fields, i + 1, fld_count)
    feature_values[i + 1] = Concatenate(join_array(start_arr, end_arr), '::')

}}
var match_found = false;
for (var i = 0; i < Count(feature_values); i++){{
    if (IndexOf(valid_values, feature_values[i]) > -1) {{
        match_found = true;
        break;
    }}
}}
if (match_found == false) {{
    return {{"errorMessage": "The selected attributes for {', '.join(list(df.columns))} are not valid."}}
}}
return true;
'''
        print(func)
        break