def set_gremlin_profile_metrics()

in src/graph_notebook/magics/metadata.py [0:0]

• 54 lines of code
• 15 McCabe index (conditional complexity)

def set_gremlin_profile_metrics(gremlin_metadata: Metadata, profile_str: str) -> Metadata:
    querytime_regex = re.search(r'Query Execution: (.*?)\n', profile_str)
    predicates_regex = re.search(r'# of predicates: (.*?)\n', profile_str)
    count_regex = re.search(r'Count: (.*?)\n', profile_str)
    serialization_regex = re.search(r'Serialization: (.*?)\n', profile_str)
    results_size_regex = re.search(r'Response size \(bytes\): (.*?)\n', profile_str)
    serializer_regex = re.search(r'Response serializer: (.*?)\n', profile_str)
    total_index_ops_regex = re.findall(r'# of statement index ops: (.*?)\n', profile_str)
    unique_index_ops_regex = re.findall(r'# of unique statement index ops: (.*?)\n', profile_str)
    duplication_ratio_regex = re.findall(r'Duplication ratio: (.*?)\n', profile_str)
    terms_materialized_regex = re.findall(r'# of terms materialized: (.*?)\n', profile_str)
    if querytime_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='query_time', metric_type='float',
                                 metric_value=querytime_regex.group(1))
    if predicates_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='predicates', metric_type='int',
                                 metric_value=predicates_regex.group(1))
    if count_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='results', metric_type='int',
                                 metric_value=count_regex.group(1))
    if serialization_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='seri_time', metric_type='float',
                                 metric_value=serialization_regex.group(1))
    if serializer_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='seri_type', metric_type='str',
                                 metric_value=serializer_regex.group(1))
    if results_size_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='results_size', metric_type='int',
                                 metric_value=results_size_regex.group(1))
    if total_index_ops_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='query_total_index_ops', metric_type='int',
                                 metric_value=total_index_ops_regex[0])
        if len(total_index_ops_regex) > 1:
            set_profile_metric_value(metadata=gremlin_metadata, metric_name='seri_total_index_ops', metric_type='int',
                                     metric_value=total_index_ops_regex[1])
    if unique_index_ops_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='query_unique_index_ops', metric_type='int',
                                 metric_value=unique_index_ops_regex[0])
        if len(unique_index_ops_regex) > 1:
            set_profile_metric_value(metadata=gremlin_metadata, metric_name='seri_unique_index_ops', metric_type='int',
                                     metric_value=unique_index_ops_regex[1])
    if duplication_ratio_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='query_duplication_ratio', metric_type='float',
                                 metric_value=duplication_ratio_regex[0])
        if len(duplication_ratio_regex) > 1:
            set_profile_metric_value(metadata=gremlin_metadata, metric_name='seri_duplication_ratio',
                                     metric_type='float', metric_value=duplication_ratio_regex[1])
    if terms_materialized_regex:
        set_profile_metric_value(metadata=gremlin_metadata, metric_name='query_terms_materialized', metric_type='int',
                                 metric_value=terms_materialized_regex[0])
        if len(terms_materialized_regex) > 1:
            set_profile_metric_value(metadata=gremlin_metadata, metric_name='seri_terms_materialized',
                                     metric_type='int', metric_value=terms_materialized_regex[1])
    return gremlin_metadata