some new features
This commit is contained in:
@ -0,0 +1,90 @@
|
||||
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)
|
||||
Reference in New Issue
Block a user