I don't kow if there is a function to what you want to do but I think that overloading draw_object would be a solution.

Best regards.