Skip to content

Commit 9033c72

Browse files
committed
276 - Converted is_class local function using closure to be callable class
1 parent a36b9e7 commit 9033c72

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

pylint_django/augmentations/__init__.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -761,9 +761,17 @@ def allow_meta_protected_access(node):
761761
return False
762762

763763

764-
def is_class(class_name):
764+
class IsClass:
765+
def __init__(self, class_name):
766+
self.class_name = class_name
767+
768+
def __call__(self, node):
769+
return node_is_subclass(node, self.class_name)
770+
771+
772+
def is_class(class_name) -> IsClass:
765773
"""Shortcut for node_is_subclass."""
766-
return lambda node: node_is_subclass(node, class_name)
774+
return IsClass(class_name)
767775

768776

769777
def wrap(orig_method, with_method):

0 commit comments

Comments
 (0)