Take a look at pylogctx maybe it'll do what you need.