You are right! I'm thinking to backport it with some minor syntax change for python2.