As someone who had this exact surgery done and has read everything I could about multiple high level athletes who had it done, FOCUS ON YOUR RECOVERY by listening to your PT.
Seriously, it is going to be a up and down road for you to get from day 1 post op to fully recovered.
Best thing I can think of now since you're a couple months before the surgery is to work on dropping any bodyfat you have and getting your cardio up. General stuff that doesn't cause pain, both HIIT and steady state with stuff that doesn't hurt. Lift like normal.
Do not rush coming back.