Breast Pump

Hi, do you usually buy a breast pump before or after delivery? I bought one previously, but after two months of breastfeeding, I switched to formula due to low milk supply. Now, some parts of the pump are missing, and I'm considering buying a new one for my second baby. Should I wait until after delivery to get it, as I'm unsure how much milk I’ll be able to produce?