Fourier Optics in Python