contrib/schema: update generator to rewrite imports

Let's not make goimports a requirement if we can avoid it.