Skip to content

Commit eb35ac9

Browse files
committed
Make linter happy
1 parent 2b50220 commit eb35ac9

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

qiling/const.py

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
#
55

66
from enum import EnumMeta, IntEnum
7-
from typing import Mapping
7+
from typing import Mapping, TypeVar
88

99
class QL_ENDIAN(IntEnum):
1010
EL = 1
@@ -64,16 +64,18 @@ class QL_INTERCEPT(IntEnum):
6464
QL_HOOK_BLOCK = 0b0001
6565
QL_CALL_BLOCK = 0b0010
6666

67-
def __reverse_enum(e: EnumMeta) -> Mapping[str, int]:
67+
__QL_CE = TypeVar('__QL_CE', QL_DEBUGGER, QL_ARCH, QL_OS, QL_VERBOSE)
68+
69+
def __reverse_enum(e: EnumMeta) -> Mapping[str, __QL_CE]:
6870
'''Create a reverse mapping for an enum.
6971
'''
7072

71-
return dict((k.lower(), v.value) for k, v in e.__members__.items())
73+
return dict((v.name.lower(), v.value) for v in e.__members__.values())
7274

73-
debugger_map = __reverse_enum(QL_DEBUGGER)
74-
arch_map = __reverse_enum(QL_ARCH)
75-
os_map = __reverse_enum(QL_OS)
76-
verbose_map = __reverse_enum(QL_VERBOSE)
75+
debugger_map: Mapping[str, QL_DEBUGGER] = __reverse_enum(QL_DEBUGGER)
76+
arch_map : Mapping[str, QL_ARCH] = __reverse_enum(QL_ARCH)
77+
os_map : Mapping[str, QL_OS] = __reverse_enum(QL_OS)
78+
verbose_map : Mapping[str, QL_VERBOSE] = __reverse_enum(QL_VERBOSE)
7779

7880
loader_map = {
7981
QL_OS.LINUX : "ELF",

0 commit comments

Comments
 (0)