def __init__(self, xml_out=None, basedir=None): """ @param xml_out: Output filename, if None, will dump to tempfile @param basedir: working directory, where file will be created """ if xml_out: self.filename = xml_out fd = open(xml_out, 'w') else: try: (fdesc, self.filename) = tempfile.mkstemp('.xml', 'pyx12_') fd = os.fdopen(fdesc, 'w+b') #fd = tempfile.NamedTemporaryFile() #self.filename = fd.name except: #self.filename = '997.tmp.xml' (fdesc, self.filename) = tempfile.mkstemp(suffix='.xml', prefix='pyx12_', dir=basedir) fd = os.fdopen(fdesc, 'w+b') #fd = file(os.path.join(basedir, self.filename), 'w') self.cur_line = None self.errors = [] if not fd: raise EngineError('Could not open temp error xml file') self.writer = XMLWriter(fd) #self.writer.doctype( # u"x12simple", u"-//J Holland//DTD XML X12 Document # Conversion1.0//EN//XML", # u"%s" % (dtd_urn)) self.writer.push("x12err")
def __init__(self, fd, type, dtd_urn): self.writer = XMLWriter(fd) if dtd_urn: self.writer.doctype( type, "-//J Holland//DTD XML X12 Document Conversion1.0//EN//XML", "%s" % (dtd_urn)) self.writer.push(type) self.last_path = None
def test_write1(self): try: fd = StringIO(encoding='ascii') print('CASE 1:') except: fd = StringIO() print('CASE 2:') writer = XMLWriter(fd) writer.push("x12err") while len(writer) > 0: writer.pop() self.assertEqual(fd.getvalue(), self.res) fd.close()
def test_write_temp(self): (fdesc, filename) = tempfile.mkstemp('.xml', 'pyx12_') fd = os.fdopen(fdesc, 'w+b') #fd = file(filename, 'rw') writer = XMLWriter(fd) writer.push(u"x12err") while len(writer) > 0: writer.pop() fd.seek(0) self.assertEqual(fd.read(), self.res) fd.close() try: os.remove(filename) except: pass
def test_write1(self): try: fd = StringIO(encoding='ascii') except: fd = StringIO() writer = XMLWriter(fd) writer.push(u"x12err") while len(writer) > 0: writer.pop() self.assertEqual(fd.getvalue(), self.res) fd.close() try: os.remove(filename) except: pass
def test_write_temp(self): (fdesc, filename) = tempfile.mkstemp('.xml', 'pyx12_') with open(filename, 'w') as fd: writer = XMLWriter(fd) writer.push("x12err") while len(writer) > 0: writer.pop() with open(filename, 'r') as fd: self.assertEqual(fd.read(), self.res) try: os.remove(filename) except: pass