มัณฑนากรวิธีการเรียนที่มีข้อโต้แย้งตัวเอง?


154

ฉันจะส่งฟิลด์คลาสไปยัง decorator บนเมธอด class เป็นอาร์กิวเมนต์ได้อย่างไร สิ่งที่ฉันต้องการจะทำคือ:

class Client(object):
    def __init__(self, url):
        self.url = url

    @check_authorization("some_attr", self.url)
    def get(self):
        do_work()

มันบ่นว่าตัวเองไม่มีอยู่สำหรับการส่งผ่านself.urlไปยังมัณฑนากร มีวิธีแก้ไขไหม?


นั่นคือมัณฑนากรแบบกำหนดเองที่คุณสามารถควบคุมได้หรือว่าคุณไม่สามารถเปลี่ยนแปลงได้
Joel Cornett

มันเป็นมัณฑนากรของฉันดังนั้นฉันสามารถควบคุมมันได้อย่างสมบูรณ์
Mark

จะได้รับการเรียกว่าก่อน init ผมคิดว่าเป็นปัญหา ...
Joran บีสลีย์

7
ปัญหาคือว่าตัวเองไม่ได้อยู่ที่เวลานิยามฟังก์ชั่น คุณต้องทำให้มันเป็นฟังก์ชั่นบางส่วน
พลวง

คำตอบ:


208

ใช่. แทนที่จะส่งผ่านแอ็ตทริบิวต์อินสแตนซ์ ณ เวลากำหนดคลาสให้ตรวจสอบตอนรันไทม์:

def check_authorization(f):
    def wrapper(*args):
        print args[0].url
        return f(*args)
    return wrapper

class Client(object):
    def __init__(self, url):
        self.url = url

    @check_authorization
    def get(self):
        print 'get'

>>> Client('http://www.google.com').get()
http://www.google.com
get

มัณฑนากรดักจับอาร์กิวเมนต์ของเมธอด อาร์กิวเมนต์แรกเป็นอินสแตนซ์ดังนั้นจึงอ่านแอททริบิวต์นั้น คุณสามารถส่งชื่อแอตทริบิวต์เป็นสตริงไปยังมัณฑนากรและใช้getattrหากคุณไม่ต้องการ hardcode ชื่อแอตทริบิวต์:

def check_authorization(attribute):
    def _check_authorization(f):
        def wrapper(self, *args):
            print getattr(self, attribute)
            return f(self, *args)
        return wrapper
    return _check_authorization

38
from re import search
from functools import wraps

def is_match(_lambda, pattern):
    def wrapper(f):
        @wraps(f)
        def wrapped(self, *f_args, **f_kwargs):
            if callable(_lambda) and search(pattern, (_lambda(self) or '')): 
                f(self, *f_args, **f_kwargs)
        return wrapped
    return wrapper

class MyTest(object):

    def __init__(self):
        self.name = 'foo'
        self.surname = 'bar'

    @is_match(lambda x: x.name, 'foo')
    @is_match(lambda x: x.surname, 'foo')
    def my_rule(self):
        print 'my_rule : ok'

    @is_match(lambda x: x.name, 'foo')
    @is_match(lambda x: x.surname, 'bar')
    def my_rule2(self):
        print 'my_rule2 : ok'



test = MyTest()
test.my_rule()
test.my_rule2()

ouput: my_rule2: ตกลง


@raphael ในการตั้งค่านี้ฉันไม่สามารถเข้าถึง _lambda หรือรูปแบบได้ ฉันจะแก้ไขได้อย่างไร
Jonathan

1
@ ราฟาเอล: ฉันจะทำเช่นเดียวกันสำหรับวิธีการเรียนเนื่องจากวิธีการทั้งหมดที่นี่เป็นวิธีการอินสแตนซ์
Apurva Kunkulol

38

ตัวอย่างที่กระชับยิ่งขึ้นอาจเป็นดังนี้:

#/usr/bin/env python3
from functools import wraps

def wrapper(method):
    @wraps(method)
    def _impl(self, *method_args, **method_kwargs):
        method_output = method(self, *method_args, **method_kwargs)
        return method_output + "!"
    return _impl

class Foo:
    @wrapper
    def bar(self, word):
        return word

f = Foo()
result = f.bar("kitty")
print(result)

สิ่งที่จะพิมพ์:

kitty!

6

อีกทางเลือกหนึ่งคือการละทิ้งน้ำตาล syntactic และตกแต่งใน__init__ชั้นเรียน

def countdown(number):
    def countdown_decorator(func):
        def func_wrapper():
            for index in reversed(range(1, number+1)):
                print("{}".format(index))
            func()
        return func_wrapper
    return countdown_decorator

class MySuperClass():
    def __init__(self, number):
        self.number = number
        self.do_thing = countdown(number)(self.do_thing)

    def do_thing(self):
        print('im doing stuff!')


myclass = MySuperClass(3)

myclass.do_thing()

ซึ่งจะพิมพ์

3
2
1
im doing stuff!

4

คุณทำไม่ได้ ไม่มีตัวตนselfในคลาสเนื่องจากไม่มีอินสแตนซ์อยู่ คุณจะต้องผ่านมันพูดชื่อstrที่มีชื่อแอตทริบิวต์เพื่อค้นหาในอินสแตนซ์ซึ่งฟังก์ชั่นที่ส่งคืนสามารถทำได้หรือใช้วิธีการอื่นทั้งหมด

โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.