Получение номера строки в Python (line number) для файловых операций

Подробные примеры

Пример 1: Чтение файла с пропуском комментариев и сохранением реальных номеров строк

Генератор, который выдает только непустые и некомментарные строки вместе с их номерами в исходном файле.

def read_non_comment_lines(filepath):
    with open(filepath) as f:
        for lineno, line in enumerate(f, 1):
            stripped = line.strip()
            if stripped and not stripped.startswith('#'):
                yield lineno, stripped

for num, text in read_non_comment_lines('config.txt'):
    print(f'{num}: {text}')

1: server=localhost
3: port=8080
5: debug=True

Пример 2: Быстрый доступ к случайным строкам большого файла через linecache

При случайном доступе к строкам без необходимости чтения всего файла linecache удобен, но требует осторожности с памятью.

import linecache
import random

filename = 'large_log.txt'
num_lines = 1000000  # необходимо знать или оценить
rand_indices = [random.randint(1, num_lines) for _ in range(5)]
for idx in rand_indices:
    content = linecache.getline(filename, idx)
    print(f'Строка {idx}: {content.strip()}')
linecache.clearcache()

Строка 456789: INFO 2023-10-01 ...
Строка 12: ERROR ...
...

Пример 3: Декоратор для логирования места вызова функции

Декоратор, записывающий номер строки, из которой была вызвана декорированная функция.

import inspect
import functools

def log_caller(func):
    @functools.wraps(func)
    def wrapper(*args, **kwargs):
        frame = inspect.currentframe().f_back
        print(f'Вызвана функция {func.__name__} из строки {frame.f_lineno}')
        return func(*args, **kwargs)
    return wrapper

@log_caller
def process_file(filename):
    with open(filename) as f:
        return f.read()

process_file('test.txt')

Вызвана функция process_file из строки 13

Пример 4: Кастомная константа __LINE__ с помощью sys._getframe

Аналог макроса __LINE__ из C, позволяющий получать номер строки в месте вызова.

import sys

def __LINE__():
    return sys._getframe(1).f_lineno

print('Текущая строка', __LINE__())
a = 1
print('А теперь строка', __LINE__())

Текущая строка 8
А теперь строка 10

Пример 5: Обработка ошибок парсинга с точным указанием строки

При разборе конфигурационного файла номера строк помогают быстро локализовать проблему.

def parse_config(filepath):
    with open(filepath) as f:
        for lineno, line in enumerate(f, 1):
            try:
                key, value = line.strip().split('=', 1)
                print(f'{key}: {value}')
            except ValueError as e:
                raise ValueError(f'Ошибка в строке {lineno}: {e}')

try:
    parse_config('bad_config.txt')
except ValueError as err:
    print(err)

Ошибка в строке 3: not enough values to unpack (expected 2, got 1)

Пример 6: Кастомный assert с номером строки

Функция, заменяющая стандартный assert и добавляющая в сообщение номер строки из вызывающего кода.

import sys

def assert_with_line(condition, msg=''):
    if not condition:
        frame = sys._getframe(1)
        raise AssertionError(f'Строка {frame.f_lineno}: {msg}')

x = 5
assert_with_line(x > 10, 'x должно быть больше 10')

AssertionError: Строка 12: x должно быть больше 10