Python tools for magnetic equilibria in tokamak plasmas