I've opened an Issue/PR a while back and could use some help getting it out the door: https://github.com/apache/incubator-airflow/pull/3914.
It's a small change (I think), but I it would be useful for us, as we're currently reimplementing the same operator for this exact purpose. Some code reviews/feedback would be more than appreciated. Thanks! Szymon
