blob: 5af11af387f638988e17887d6b51645632e05e14 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
|
"""Test check for classes extending an Enum class."""
# pylint: disable=missing-class-docstring,invalid-name
from enum import Enum, IntFlag
# We don't flag the Enum class itself
class A(Enum):
x = 1
y = 2
# But we do flag any inheriting classes
# that try to extend the Enum class.
class B(A): # [invalid-enum-extension]
z = 3
# If no items have been added to the base
# Enum class then the lint is not raised.
class C(Enum):
pass
class D(C):
x = 3
class CustomFlags(IntFlag):
SUPPORT_OPEN = 1
SUPPORT_CLOSE = 2
|
