Bir dekoratör, imzasını değiştirmeden değişkenleri bir işleve nasıl iletebilir?

Question

Öncelikle yapmak istediğim şeyin aptalca kötülükten bir şey olarak kabul edilebileceğini kabul etmeme izin verin, ancak yine de Python'da yapıp yapamayacağımı öğrenmek istiyorum.

Diyelim ki, değişkenleri tanımlayan anahtar kelime argümanlarını alan bir işlev dekoratörüm var ve sarılmış işlevdeki bu değişkenlere erişmek istiyorum. Böyle bir şey yapabilir:

def more_vars(**extras): 
    def wrapper(f): 
     @wraps(f) 
     def wrapped(*args, **kwargs): 
      return f(extras, *args, **kwargs) 
     return wrapped 
    return wrapper 

Şimdi şöyle bir şey yapabilirsiniz:

@more_vars(a='hello', b='world') 
def test(deco_vars, x, y): 
    print(deco_vars['a'], deco_vars['b']) 
    print(x, y) 

test(1, 2) 
# Output: 
# hello world 
# 1 2 

Bu konuda sevmiyorum şey bu dekoratör kullandığınızda, zorunda olmasıdır dekoratöre tokat ek olarak ek değişken ekleyerek, işlevin çağrı imzasını değiştirmek. Eğer fonksiyon için yardım bakarsak Ayrıca, size işlevini çağırarak kullanılacak beklenen değiliz fazladan değişken bakın:

Bu kullanıcı işlevi çağırmak için beklenen gibi görünmesini sağlar

help(test) 
# Output: 
# Help on function test in module __main__: 
# 
# test(deco_vars, x, y) 

3 parametresi ile, ama açıkçası bu işe yaramaz. Bu nedenle, ilk parametrenin arayüzün bir parçası olmadığını belirten bir mesaj eklemeniz gerekecek, sadece bir uygulama detayı ve göz ardı edilmemelidir. Gerçi bu berbat bir şey. Bu değişkenleri küresel ölçekte bir şeye asmadan bunu yapmanın bir yolu var mı? İdeal olarak, ben aşağıdaki gibi bakmak istiyorum:

@more_vars(a='hello', b='world') 
def test(x, y): 
    print(a, b) 
    print(x, y) 

test(1, 2) 
# Output: 
# hello world 
# 1 2 
help(test) 
# Output: 
# Help on function test in module __main__: 
# 
# test(x, y) 

Bir Python 3 Varsa tek çözüm ile memnunum.

Answer 1

Yerlileri ad alanına enjekte etme fikrini sevmiyorum. Muhtemelen mümkün olsa da, enjekte edilen yeni yerliler ile zaten fonksiyonda bulunan isimler arasındaki çarpışmaların üstesinden gelmek kötü olacaktır.

İşlev nesnesinde attrs olarak saklamak kabul edilebilir mi? Bu şekilde isimleri verildi.

from functools import wraps 

def more_vars(**extras): 
    def wrapper(f): 
     @wraps(f) 
     def wrapped(*args, **kwargs): 
      return f(*args, **kwargs) 
     for k,v in extras.items(): 
      setattr(wrapped, k, v) 
     return wrapped 
    return wrapper 

@more_vars(a='hello', b='world') 
def test(x, y): 
    print(test.a, test.b) 
    print(x, y) 

Answer 2

Sen işlevin yerel değişkenlere dekoratör geçirilen değişkenleri ekler nasm'ın yapabilirsiniz:

import sys 
from functools import wraps 
from types import FunctionType 


def is_python3(): 
    return sys.version_info >= (3, 0) 


def more_vars(**extras): 
    def wrapper(f): 
     @wraps(f) 
     def wrapped(*args, **kwargs): 
      fn_globals = {} 
      fn_globals.update(globals()) 
      fn_globals.update(extras) 
      if is_python3(): 
       func_code = '__code__' 
      else: 
       func_code = 'func_code' 
      call_fn = FunctionType(getattr(f, func_code), fn_globals) 
      return call_fn(*args, **kwargs) 
     return wrapped 
    return wrapper 


@more_vars(a="hello", b="world") 
def test(x, y): 
    print("locals: {}".format(locals())) 
    print("x: {}".format(x)) 
    print("y: {}".format(y)) 
    print("a: {}".format(a)) 
    print("b: {}".format(b)) 


if __name__ == "__main__": 
    test(1, 2) 

Bunu Can

? Emin! yapmalısınız? Muhtemelen değil!

(Kod mevcut here.)

Answer 3

Bu tek sorun var gibi görünüyor help sarılmış fonksiyonun imza olarak ham test imzasını gösteren olmasıdır ve bunu istemiyoruz. oluyor

tek nedeni olduğunu wraps (daha doğrusu update_wrapper, hangi wraps aramaları) sargıya wrappee dan açıkça kopyalar bu.

Tam olarak ne yaptığınıza ve kopyalamak istemediğinize karar verebilirsiniz. Farklı yapmak istediğiniz şey basitse, yalnızca varsayılan WRAPPER_ASSIGNMENTS ve WRAPPER_UPDATES'dan gelen şeyleri filtrelemek meselesidir.Diğer şeyleri değiştirmek isterseniz, update_wrapper'u çatallamanız ve kendi sürümünüzü kullanmanız gerekebilir - ancak functools, okunabilir örnek olarak kullanılması gerektiğinden, dokümanların en üstünde the source bağlantısına sahip olan modüllerden biridir. kodu. Senin durumunda

, sadece wraps(f, updated=[]) meselesi olabilir veya f imzasını almak için kullanım inspect.signature gibi fantezi bir şey yapmak istiyorum ve ilk parametreyi kaldırmak için değiştirmek ve açıkça bir sarmalayıcı inşa edebilir etrafında inspect modülünü bile kandırmak için.