| import os |
| import sys |
| import errno |
|
|
| try: |
| from lxml import etree |
| have_lxml = True |
| except ImportError: |
| have_lxml = False |
| from xml.etree import ElementTree as etree |
|
|
| 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: |
| """ |
| 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() |
| |
| 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) |
|
|
