CODEWITHSUNDEEP
pythondecoratorpython-decorators
Mujeeb Ur Rehman
Mujeeb Ur Rehman

Reputation: 87

Different decorators for the same class

I have a decorator that takes arguments and I want to apply it to a class individually passing different arguments each time to create a separate different version of the same class. For example:

@decorator(arg1)
class SomeClass:
   pass

@decorator(arg2)
class SomeOtherClassWithSameContent:
    pass  # repetition of code here

How can I avoid creating two classes with the same functionality and avoid code repetition? Thanks!

Upvotes: 0

Views: 122

Answers (1)

deceze
deceze

Reputation: 521994

Decorators don't have to be applied literally using @:

class Foo:
    pass


SomeClass = decorator(arg1)(Foo)
SomeOtherClassWithSameContent = decorator(arg2)(Foo)

The @ syntax is basically just shorthand for the above anyway.

Upvotes: 1

Related Questions