I think the real limit is being very cautious on communal water (swimming pools, hot tubs) and infection.
Even with bathtubs in a private home it pays to be very cautious and obsessively clean.
If you have no complications you will never know exactly how over cautious you were.
Your surgeon will probably say two or three months wait for swimming.