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
