（6） Decorator extension [email protected] Principle of use

（ One ）@pytest.mark.XXX How did it happen ？

Children's shoes that have played interface automation have been used pytest, Basically, it is realized by decorators , In the last few articles, we talked about the operating principle of the decorator , This time we expand a new knowledge point ：@pytest.mark.XXX What is the implementation principle ？

import pytest

@pytest.mark.skip
def test():
    pass

Let's take an example   ：

@pytest.mark.skip   It's actually using pytest In the module mark This object .skip Method 

@ modular . object . Method

（ Two ） Let's implement a decorator of this form by ourselves

（1） We build a new one lemon.py file , Create a Mark class , Write the inside skip function ,skip There is a nested function in the function wrapper.

 （2） Build another one demo1_ Decorator extension .py file , Create a test2 Method , then @lemon.mark.skip, Finally, call below test2() try .

  Print the results ：

----skip----1
------test2------
----skip----2

Let's try the source code for you

from py09_07day import lemon


class Mark:
    def skip(self, func):
        def wrapper(*args, **kwargs):
            print("----skip----1")
            res = func()
            print("----skip----2")
            return res

        return wrapper


mark = Mark()





# @lemon.mark.skip ===>  @lemon This class .mark object . Method 
# @skip   test2 = skip(test2)
@lemon.mark.skip
def test2():
    print('------test2------')


test2()



#  Print the results 
----skip----1
------test2------
----skip----2