mirror of
https://github.com/microsoft/debugpy.git
synced 2025-12-23 08:48:12 +00:00
c03206972d
Fix Flask and Django multiprocess tests. Fix test logs not being captured by pytest. Fix "import debug_me" check improperly applied in tests where it is unnecessary. Fix some clarifying patterns not respecting the underlying pattern. Add pattern helpers for strings: starting_with, ending_with, containing. Move DAP test helpers to a separate module, and add a helper for frames. Add support for line markers when setting breakpoints and matching frames. Assorted test fixes around handling of Unicode and paths.
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License. See LICENSE in the project root
|
|
# for license information.
|
|
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
"""Helpers to work with Python code.
|
|
"""
|
|
|
|
import py.path
|
|
import re
|
|
|
|
from ptvsd.common import compat
|
|
|
|
_marked_line_numbers_cache = {}
|
|
|
|
|
|
def get_marked_line_numbers(path):
|
|
"""Given a path to a Python source file, extracts line numbers for all lines
|
|
that are marked with #@. For example, given this file::
|
|
|
|
print(1) # @foo
|
|
print(2)
|
|
print(3) # @bar
|
|
|
|
the function will return::
|
|
|
|
{"foo": 1, "bar": 3}
|
|
"""
|
|
|
|
if isinstance(path, py.path.local):
|
|
path = path.strpath
|
|
|
|
try:
|
|
return _marked_line_numbers_cache[path]
|
|
except KeyError:
|
|
pass
|
|
|
|
# Read as bytes, to avoid decoding errors on Python 3.
|
|
with open(path, "rb") as f:
|
|
lines = {}
|
|
for i, line in enumerate(f):
|
|
match = re.search(br"#\s*@\s*(.+?)\s*$", line)
|
|
if match:
|
|
marker = compat.force_unicode(match.group(1), "ascii")
|
|
lines[marker] = i + 1
|
|
|
|
_marked_line_numbers_cache[path] = lines
|
|
return lines
|