Returning a function in Python

Question

I get the following error "TypeError: make_pretty() takes exactly 1 argument (2 given)" for the below code snippet. I'm learning python and any help would be helpful...

class test:
    def make_pretty(func):
        def inner():
            print("I got decorated")
            func()
        return inner

    def ordinary():
        print("I am ordinary")

pretty1 = test()
pretty = pretty1.make_pretty(pretty1.ordinary)
print(pretty())

I also tried it with decorators, but still face the same error..

class SmartDivide:
    def smart_divide(self,func):
        def inner(a,b):
            print("I am going to divide",a,"and",b)
            if b == 0:
                print("Whoops! cannot divide")
                return

            return func(a,b)
        return inner

@smart_divide
def divide(self,a,b):
    return a/b

dvd = SmartDivide()
print(dvd.divide(2,5))
dvd.divide(2,0)

Do this without a class. The problem is that methods defined in a class, the first argument will automatically refer to the instance itself. By convention, all methods are defined with `self` as the first argument. This argument is magically passed when using an instance. Before you start using classes, read the following: https://docs.python.org/3.5/tutorial/classes.html Python's class system is actually pretty simple, but a bit different from other common class-based OOP languages like Java. — juanpa.arrivillaga, Jul 05 '16 at 01:12

Etor Madiv · Accepted Answer · 2016-07-05T01:25:11.880

-1

You may need the self keyword

class test:
    def make_pretty(self, func):
        def inner():
            print("I got decorated")
            func()
        return inner

    def ordinary(self):
        print("I am ordinary")

pretty1 = test()
pretty = pretty1.make_pretty(pretty1.ordinary)
print(pretty())

edited Jul 05 '16 at 01:25

answered Jul 05 '16 at 01:12

Etor Madiv

69
1
7

thanks..That was very useful...I – jeff Jul 05 '16 at 01:45

Returning a function in Python

1 Answers1