A Python implementation of TraceRT