Working after baby?

I currently don’t have a job, my first trimester and up until the middle of the second trimester was pretty rough, and because of that it made me quit my job I know many or most that have worked through their pregnancy but I did not. After I have the baby, I was wondering what the right time would be to get back to work or start looking for work. I was going to give myself 2 months but in those 2 months apply to jobs and tell them I would be available at the end of the 2 months. Do you guys think 2 months is appropriate? I would love to be a stay at home mom but financially I think even working part time would do wonders!! My significant other does work and makes a decent amount but I need my own money! I hate depending on someone else and feel I’m going to lose my mind not working!! Can anyone else relate