Function delete_segment

api/segments/services.py:13–79

Delete a segment using optimized bulk operations. Uses bulk UPDATE/DELETE operations instead of individual soft-deletes, reducing the number of database queries from O(n) to O(1) where n is the number of rules and conditions. Note: This is a temporary solution until we redesig

(
    segment: "Segment",
    author: AuthorData,
)

def delete_segment(
    segment: "Segment",
    author: AuthorData,
) -> None:
    """
    Delete a segment using optimized bulk operations.

    Uses bulk UPDATE/DELETE operations instead of individual soft-deletes,
    reducing the number of database queries from O(n) to O(1) where n is
    the number of rules and conditions.

    Note: This is a temporary solution until we redesign the segment data model.
    """
    from features.models import FeatureSegment
    from segments.models import Condition, Segment, SegmentRule
    from segments.tasks import create_segment_deleted_audit_log

    now = timezone.now()

    segment_name = segment.name
    segment_uuid = str(segment.uuid)
    segment_id = segment.id
    project_id = segment.project_id

    segment_ids = list(
        Segment.objects.filter(
            models.Q(id=segment.id) | models.Q(version_of_id=segment.id)
        ).values_list("id", flat=True)
    )

    top_level_rule_ids = list(
        SegmentRule.objects.filter(segment_id__in=segment_ids).values_list(
            "id", flat=True
        )
    )

    all_rule_ids = set(top_level_rule_ids)
    current_level_ids = top_level_rule_ids

    while current_level_ids:
        nested_rule_ids = list(
            SegmentRule.objects.filter(rule_id__in=current_level_ids).values_list(
                "id", flat=True
            )
        )
        all_rule_ids.update(nested_rule_ids)
        current_level_ids = nested_rule_ids

    all_rule_ids_list = list(all_rule_ids)

    with transaction.atomic():
        FeatureSegment.objects.filter(segment_id__in=segment_ids).delete()
        Condition.objects.filter(rule_id__in=all_rule_ids_list).update(deleted_at=now)
        SegmentRule.objects.filter(id__in=all_rule_ids_list).update(deleted_at=now)
        Segment.objects.filter(id__in=segment_ids).update(deleted_at=now)

    create_segment_deleted_audit_log.delay(
        args=(