"""Approximate stepped shaft with a broad axial key/wrench flat."""

from shaft_common import cut_side_flat, export_horizontal, stepped_body


SEGMENTS = [(16.0, 68.0), (24.0, 22.0)]


def gen_step():
    body, _ = stepped_body(SEGMENTS)
    body = cut_side_flat(body, diameter=16.0, z_start=0.0, length=30.0, flat_depth=2.7)
    return export_horizontal(body, "stepped_shaft_054964_like")