returnの位置について

Question

関数が呼び出された時に'start'、終了した時に'end'と表示するtestというデコレータを定義しよう（O'REILLYの入門 python3 P136復習課題4-10より）

関数new_funcの処理結果であるresultをstartとendの間に表示しようと考え、print(end)の前にresultが戻り値として表示されるよう組んだのですが、以下のようになりendが表示されなくなってしまいます。

>>> def test(func):
...     def new_func(*args, **kwargs):
...             print('start')
...             result = func(*args, **kwargs)
...             return result
...             print('end')
...     return new_func
... 
>>> @test
... def greeting():
...     print('Hello')
... 
>>> greeting()
start
Hello

試したこと

解答のコードは以下のようになっており、returnをprint('end')の後ろに持ってくる、というものでした。自分の感覚では、この位置にreturnを書くと、endの後にresultが表示されてしまう（start→end→resultの順）のでは？と考えたのでどうも納得いきません...returnの持つ意味と記述する位置について教えて頂けますと幸いです。

>>> def test(func):
...     def new_func(*args, **kwargs):
...             print('start')
...             result = func(*args, **kwargs)
...             print('end')
...             return result
...     return new_func
... 
>>> @test
... def greeting():
...     print('Hello')
... 
>>> greeting()
start
Hello
end
```


### 補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[Lhankor_Mhy]

markdownでのソースコードの書き方は、以下をご覧になるとわかりやすいです。

ソースコードを書きましょう
https://teratail.com/help/question-tips#questionTips3-5-1

3-7. markdownを利用しましょう
https://teratail.com/help/question-tips#questionTips3-7

[Lhankor_Mhy]

def new_func(*args, kwargs):

は、

def new_func(*args, **kwargs):

ではないですか？

[hamazon]

ご指摘ありがとうございます。
おっしゃる通り、kwargs→**kwargsの間違いでした。
ターミナルからそのままコピペして貼り付けたのですが、サイト上では仕様が変わってしまうのですね...気をつけます！

Answer 1

コードの挙動として、greeting() を呼び出したとき以下の順序で実行されます。

1. 引数 func に greeting() が渡されて、test() が呼び出される。(つまり、func = greeting())
2. print("start") で start が出力
3. func(*args, **kwargs)、つまり、greeting(*args, **kwargs) が呼び出される。
4. greeting() 内の print("Hello") で Hello が出力
5. print("end") で end が出力

詳細はデコレーターの仕組みについて理解する必要があります。

Pythonのデコレータについて - Qiita

def test(func):
    def new_func(*args, **kwargs):
        print("start")
        result = func(*args, **kwargs)
        print("end")

    return new_func


@test
def greeting():
    print("Hello")


greeting()

質問の1つ目のコードで return したら end が表示されなくなった理由は

1. 引数 func に greeting() が渡されて、test() が呼び出される。(つまり、func = greeting())
2. print("start") で start が出力
3. func(*args, **kwargs)、つまり、greeting(*args, **kwargs) が呼び出される。
4. greeting() 内の print("Hello") で Hello が出力
5. return result で test() 関数が終了する

となり、return より後の print("end") が実行されなくなったためです。

def test(func):
    def new_func(*args, **kwargs):
        print("start")
        result = func(*args, **kwargs)
        return result
        print("end")

    return new_func


@test
def greeting():
    print("Hello")


greeting()

Comments

[hamazon]

なるほど！
①greeting()の中のprint('Hello')が実行されて、その後にtest関数のprint('end')が実行される
②returnを実行するとそれ以降の関数内の処理が実行されない
という事ですね！大変わかりやすく助かりました。ありがとうございました！

Answer 2

endの後にresultが表示されてしまう（start→end→resultの順）のでは？と考えた

まさにそのとおりです。

def test(func):
    def new_func(*args, **kwargs): 
        print('start')
        result = func(*args, **kwargs)
        print('end')
        return 'result' # 変更
    return new_func
 
@test
def greeting():
    print('Hello')

print( greeting() ) # 変更

"""
start
Hello
end
result
"""

---

ちなみに、return文を実行すると、その後のコードは無視して関数を終了しますので、returnをprint('end')の前にすると実行されません。

Comments

[hamazon]

return文を実行するとその後のコードは実行されないという事ですね！
そこの理解が足りていませんでした。ありがとうございます！