mirror of
https://github.com/microsoft/debugpy.git
synced 2025-12-23 08:48:12 +00:00
40 lines
1 KiB
Python
40 lines
1 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
|
|
|
|
|
|
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
|
|
|
|
# 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
|
|
return lines
|