Prevent useless list creations

This commit is contained in:
Pierre LALET 2017-10-24 11:19:06 +02:00
parent b75dac5763
commit 47a8252622
1 changed files with 2 additions and 1 deletions

View File

@ -18,6 +18,7 @@ from scapy.utils import *
from scapy.base_classes import BasePacket, Gen, Net, Field_metaclass from scapy.base_classes import BasePacket, Gen, Net, Field_metaclass
from scapy.error import warning from scapy.error import warning
import scapy.modules.six as six import scapy.modules.six as six
from scapy.modules.six.moves import range
############ ############
@ -919,7 +920,7 @@ class _EnumField(Field):
if isinstance(enum, list): if isinstance(enum, list):
keys = range(len(enum)) keys = range(len(enum))
elif isinstance(enum, DADict): elif isinstance(enum, DADict):
keys = enum.keys() keys = enum.iterkeys()
else: else:
keys = list(enum) keys = list(enum)
if any(isinstance(x, str) for x in keys): if any(isinstance(x, str) for x in keys):