Add context to logging in python