stack_effect will never know about EXTENDED_ARG.  Instead, you simply pass the already-extended arg as the second argument. 

Making the second argument support a default of None will be slightly inconvenient, but I'll do it if you think it's a big improvement.

Considering how troublesome it was to recreate this information when I wrote my assembler, I definitely think this information should be exported out of the murky heart of Python.