Есть ли способ не выполнять выражения внутри тела with?

Question

[Альберт]

class Test:
    def __init__(self):
        pass

    def __enter__(self):
        pass

    def __exit__(self, exc_type, exc_val, exc_tb):
        pass


with Test():
    print (1) # any way not to print it?

нужно любым способом обойти выполнение тела with (в данном случае `print (1)`).
реально ли без особого шаманства?

PS: примечание наперед: это не проблема xy.

Comments

[Сергей Горностаев]

А зачем? Какая от этого может быть польза?

[Альберт]

Сергей Горностаев, конкретно моя цель — попробовать реализовать возможность использовать декораторы не только для отдельных функций, но также для отдельных строк и их последовательностей. Не знаю, насколько могу судить архитектуру декораторов python, но в некоторых случаях нахожу неудобным необходимость создания отдельных функций для декораторов. В результате если в коде много декораторов, то много и функций, многие из которых могут даже не быть логически обоснованными, а просто можно сказать обслуживают декоратор.
Такой код может быть запутанным.
моя изначальная идея вместо этого была использовать (писать) декораторы прямо в коде.
Но потом подумал, что более правильным решением было бы использовать для этого конструкцию with, так как это не настолько ломает синтаксис питона с одной стороны, и можно четко определить задать границы начала и конца строчек кода которые я хотел бы декорировать, с другой стороны.

Наивная реализация очень простая:
Допустим мы написали такую строку

with Test():
    print(1)

и хотим чтобы эта конструкция интерпретировалась как декорирование выражения print(1).

Тогда задача следующая:
1. В методе '__init__' класса Test (то есть при вызове инстанции класса) с помощью inspect находим начало (номер строки) а также модуль откуда был вызван объект. В результате мы получаем номер строки, где начинается конструкция with, которая вызывает объект Test().

2. Находим конец строки. В простом случае мы просто можем ориентироваться по количеству пробелов.

3. зная начало и конец строки мы получаем код (строку) соответствующего with блока.

4. Создаем из этого блока функцию, заменяя первую строку на что-то типа def my_temp_test():. также в начало добавляем декоратор, который будет декорировать эту функцию. Функция соответствующего декоратора, полагаю, должна уже быть объявлена в коде.

5. Выполняем эту функцию с помощью exec (например) — объявление и вызов.

Все. Мы получаем автопреобразование блока with в задекорированную функцию и вызов этой функции.
И не надо плодить лишних функций (по крайней мере вручную).

Это конечно должно немного замедлять производительность, но в проектах где критична не столько скорость выполнения, сколько удобство кодинга, это, думаю, может быть полезно (по крайней мере собираюсь это проверить).

Теперь надеюсь понятно, зачем я хотел бы игнорировать тело with — мне просто не нужно, чтобы оно выполнялось в данном случае, а сам with должен по большому счету просто являться индикатором, где начинать и где заканчивать декорировать.

Answer 1

[Альберт]

похоже, нашел один способ

import sys
import inspect

class My_Context(object):
    def __init__(self,mode=0):
        """
        if mode = 0, proceed as normal
        if mode = 1, do not execute block
        """
        self.mode=mode
    def __enter__(self):
        if self.mode==1:
            print 'Met block-skipping criterion ...'
            # Do some magic
            sys.settrace(lambda *args, **keys: None)
            frame = inspect.currentframe(1)
            frame.f_trace = self.trace
    def trace(self, frame, event, arg):
        raise
    def __exit__(self, type, value, traceback):
        print 'Exiting context ...'
        return True

with My_Context(mode=1):
    print 'Executing block of code ...'

написан на 2 питоне, но на 3 должно быть легко переписать.
mode=1 игнорирует выполнение тела with блока.
Кстати, советуют не использовать такое.

UPD:
вариант для python3

import sys

class SkipWithBlock(Exception):
    pass


class SkipContextManager:
    def __init__(self, skip = True):
        self.skip = skip

    def __enter__(self):
        if self.skip:
            sys.settrace(lambda *args, **keys: None)
            frame = sys._getframe(1)
            frame.f_trace = self.trace

    def trace(self, frame, event, arg):
        raise SkipWithBlock()

    def __exit__(self, type, value, traceback):
        if type is None:
            return  # No exception
        if issubclass(type, SkipWithBlock):
            return True  # Suppress special SkipWithBlock exception

   
with SkipContextManager():  
    print('In the with block')  # Won't be called

print('Out of the with block')

Comments

[longclaps]

Кстити, вы что курити?