python save function variable to use in another function parameter