from __future__ import absolute_import

from lib2to3.fixes import fix_unicode
from libmodernize import add_future

class FixUnicodeFuture(fix_unicode.FixUnicode):
    def transform(self, node, results):
        res = super(FixUnicodeFuture, self).transform(node, results)
        if res:
            add_future(node, 'unicode_literals')
        return res
