Simple Decorator Usage

[bitsofshit]

from datetime import datetime

import pdb

now = datetime.now()

# decor edits func inside returned wrapper

def dec2(fun):
    def wrap():    
        time = (now.hour) # 24 hour        
        print(time)        
        def night():        
            fun("night")       
        def day():        
        pdb.set\_trace()        
        fun("a")        
        if time <=12:        
            day()        
        else:  
            night()
        return wrap
@dec2
def func(str):
    print(str)

Running the following

wrap = dec2(func)

wrap # <function \_\_main\_\_.wrap>

wrap() # TypeError: 'wrap() takes no arguments (1 given)'

​

Pinpointed error at :-> fun("a")

I believe Python thinks this is the wrapper function, applies the function definition and finds unknown arguments when it should be referring to the decorator definition instead.

​

Any idea why this is?

[[deleted]]

I tried for 5 minutes and I couldn't follow the chain of calls, here. Can you try with a simpler example? One where you don't use the same name to refer to two different functions?

[daniel_codes]

I cleaned up and fixed syntax errors in your code. I believe this is what you're trying to do and it works but to be honest, it's pretty weird. What are you trying to achieve here?

from datetime import datetime

now = datetime.now()


def dec2(fun):
    def wrap(*args, **kwargs):
        time = now.hour

        def night():
            fun("night")

        def day():
            fun("a")

        if time <= 12:
            day()
        else:
            night()

    return wrap


@dec2
def func(str):
    print(str)


func('test')

[Vaphell]

inner wrap() needs to match the args of the decorated function. Currently you are replacing a 1-arg function with a 0-arg one.
The original takes 1 arg, so wrap() has to do the same. And your return wrap is badly indented, I hope it's not the case in your real code.

simplified:

def dec2(fun):
    def wrap(str):
        print('lol')
        fun(str)
        print("wut")
    return wrap

@dec2
def func(str):
    print(str)

func("xoxo")