Apologies for the long delay in response to this message - I have just started as a maintainer and developer of the Python SDK, so am still ramping up. It appears @rhytha has achieved something similar to what you seek to achieve here:
Does this answer your question, or you do require further help?