Add typing for context managers and methods returning Self in non-generated python code

Add typing for context managers and methods returning Self in non-generated python code #158