91 lines
2.4 KiB
Python
91 lines
2.4 KiB
Python
|
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
|
||
|
|
from ..Compiler.StringEncoding import EncodedString
|
||
|
|
|
||
|
|
|
||
|
|
def is_valid_tag(name):
|
||
|
|
"""
|
||
|
|
Names like '.0' are used internally for arguments
|
||
|
|
to functions creating generator expressions,
|
||
|
|
however they are not identifiers.
|
||
|
|
|
||
|
|
See https://github.com/cython/cython/issues/5552
|
||
|
|
"""
|
||
|
|
if isinstance(name, EncodedString):
|
||
|
|
if name.startswith(".") and name[1:].isdecimal():
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
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):
|
||
|
|
if is_valid_tag(name):
|
||
|
|
self.tb.start(name, attrs or {})
|
||
|
|
|
||
|
|
def end(self, name):
|
||
|
|
if is_valid_tag(name):
|
||
|
|
self.tb.end(name)
|
||
|
|
|
||
|
|
def add_entry(self, name, **attrs):
|
||
|
|
if is_valid_tag(name):
|
||
|
|
self.tb.start(name, attrs)
|
||
|
|
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)
|