| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768 |
- from __future__ import absolute_import
- import os
- import sys
- import errno
- try:
- from lxml import etree
- have_lxml = True
- except ImportError:
- have_lxml = False
- try:
- from xml.etree import cElementTree as etree
- except ImportError:
- try:
- from xml.etree import ElementTree as etree
- except ImportError:
- etree = None
- from ..Compiler import Errors
- class CythonDebugWriter(object):
- """
- Class to output debugging information for cygdb
- It writes debug information to cython_debug/cython_debug_info_<modulename>
- in the build directory.
- """
- def __init__(self, output_dir):
- if etree is None:
- raise Errors.NoElementTreeInstalledException()
- self.output_dir = os.path.join(output_dir or os.curdir, 'cython_debug')
- self.tb = etree.TreeBuilder()
- # set by Cython.Compiler.ParseTreeTransforms.DebugTransform
- self.module_name = None
- self.start('cython_debug', attrs=dict(version='1.0'))
- def start(self, name, attrs=None):
- self.tb.start(name, attrs or {})
- def end(self, name):
- self.tb.end(name)
- def serialize(self):
- self.tb.end('Module')
- self.tb.end('cython_debug')
- xml_root_element = self.tb.close()
- try:
- os.makedirs(self.output_dir)
- except OSError as e:
- if e.errno != errno.EEXIST:
- raise
- et = etree.ElementTree(xml_root_element)
- kw = {}
- if have_lxml:
- kw['pretty_print'] = True
- fn = "cython_debug_info_" + self.module_name
- et.write(os.path.join(self.output_dir, fn), encoding="UTF-8", **kw)
- interpreter_path = os.path.join(self.output_dir, 'interpreter')
- with open(interpreter_path, 'w') as f:
- f.write(sys.executable)
|
