Maternity leave, when is it good to go back to work

I’m wondering if 2 months if enough maternity leave before I go back to work. I feel I’m the sort who can’t be cooped up at home too long. Was thinking of breaking up my ML 2 months den work 2 months den another 2 months ML. Fully intend to try expressing milk even at work too