def po2tmx(self, posource, sourcelanguage='en', targetlanguage='af'):
     """helper that converts po source to tmx source without requiring files"""
     inputfile = wStringIO.StringIO(posource)
     outputfile = wStringIO.StringIO()
     outputfile.tmxfile = tmx.tmxfile(inputfile=None, sourcelanguage=sourcelanguage)
     po2tmx.convertpo(inputfile, outputfile, templatefile=None, sourcelanguage=sourcelanguage, targetlanguage=targetlanguage)
     return outputfile.tmxfile

 def test_addtranslation(self):
     """tests that addtranslation() stores strings correctly"""
     tmxfile = tmx.tmxfile()
     tmxfile.addtranslation("A string of characters", "en", "'n String karakters", "af")
     newfile = self.tmxparse(str(tmxfile))
     print(str(tmxfile))
     assert newfile.translate("A string of characters") == "'n String karakters"

 def test_withnewlines(self):
     """test addtranslation() with newlines"""
     tmxfile = tmx.tmxfile()
     tmxfile.addtranslation("First line\nSecond line", "en", "Eerste lyn\nTweede lyn", "af")
     newfile = self.tmxparse(str(tmxfile))
     print(str(tmxfile))
     assert newfile.translate("First line\nSecond line") == "Eerste lyn\nTweede lyn"

    def export(self, rotate=False):
        source_language = self.context.project.source_language.code
        target_language = self.context.language.code

        if not os.path.exists(self.directory):
            os.makedirs(self.directory)

        tmxfile = tmx.tmxfile()
        for store in self.context.stores.live().iterator():
            for unit in store.units.filter(state=TRANSLATED):
                tmxfile.addtranslation(unit.source, source_language,
                                       unit.target, target_language,
                                       unit.developer_comment)

        bs = BytesIO()
        tmxfile.serialize(bs)
        with open(self.abs_filepath, "wb") as f:
            with ZipFile(f, "w") as zf:
                zf.writestr(self.filename.rstrip('.zip'), bs.getvalue())

        last_exported_filepath = self.last_exported_file_path
        self.update_exported_revision()

        removed = []
        if rotate:
            for fn in os.listdir(self.directory):
                # Skip files from other projects.
                if not self.check_tp(fn):
                    continue
                filepath = os.path.join(self.directory, fn)
                if filepath not in [self.abs_filepath, last_exported_filepath]:
                    removed.append(filepath)
                    os.remove(filepath)

        return self.abs_filepath, removed

 def test_withcomment(self):
     """tests that addtranslation() stores string's comments correctly"""
     tmxfile = tmx.tmxfile()
     tmxfile.addtranslation("A string of chars",
                            "en", "'n String karakters", "af", "comment")
     newfile = self.tmxparse(str(tmxfile))
     print(str(tmxfile))
     assert newfile.findunit("A string of chars").getnotes() == "comment"

 def po2tmx(self, posource, sourcelanguage='en', targetlanguage='af',
            comment=None):
     """helper that converts po source to tmx source without requiring files"""
     inputfile = BytesIO(posource.encode('utf-8'))
     outputfile = BytesIO()
     outputfile.tmxfile = tmx.tmxfile(inputfile=None, sourcelanguage=sourcelanguage)
     po2tmx.convertpo(inputfile, outputfile, templatefile=None,
                      sourcelanguage=sourcelanguage,
                      targetlanguage=targetlanguage, comment=comment)
     return outputfile.tmxfile

    def __init__(self, filename, mode=None):
        """initialises tmxmultifile from a seekable inputfile or writable outputfile"""
        self.filename = filename
        if mode is None:
            if os.path.exists(filename):
                mode = 'r'
            else:
                mode = 'w'
        self.mode = mode
#        self.multifilestyle = multifilestyle
        self.multifilename = os.path.splitext(filename)[0]
#        self.multifile = open(filename, mode)
        self.tmxfile = tmx.tmxfile()

 def test_xmlentities(self):
     """Test that the xml entities '&' and '<'  are escaped correctly"""
     tmxfile = tmx.tmxfile()
     tmxfile.addtranslation("Mail & News", "en", "Nuus & pos", "af")
     tmxfile.addtranslation("Five < ten", "en", "Vyf < tien", "af")
     xmltext = str(tmxfile)
     print("The generated xml:")
     print(xmltext)
     assert tmxfile.translate('Mail & News') == 'Nuus & pos'
     assert xmltext.index('Mail &amp; News')
     assert xmltext.find('Mail & News') == -1
     assert tmxfile.translate('Five < ten') == 'Vyf < tien'
     assert xmltext.index('Five &lt; ten')
     assert xmltext.find('Five < ten') == -1

 def get_storage(self):
     return tmxfile()

def inittmx(inputfile, columnorder=None):
    return tmx.tmxfile(inputfile)

 def test_translate(self):
     tmxfile = tmx.tmxfile()
     assert tmxfile.translate("Anything") is None
     tmxfile.addtranslation("A string of characters", "en", "'n String karakters", "af")
     assert tmxfile.translate("A string of characters") == "'n String karakters"

 def tmxparse(self, tmxsource):
     """helper that parses tmx source without requiring files"""
     dummyfile = wStringIO.StringIO(tmxsource)
     print(tmxsource)
     tmxfile = tmx.tmxfile(dummyfile)
     return tmxfile