diff options
Diffstat (limited to 'src/vpp-api/python')
-rw-r--r-- | src/vpp-api/python/setup.py | 2 | ||||
-rw-r--r-- | src/vpp-api/python/vpp_papi/vpp_serializer.py | 10 |
2 files changed, 8 insertions, 4 deletions
diff --git a/src/vpp-api/python/setup.py b/src/vpp-api/python/setup.py index 952944578c8..b5fc11a590a 100644 --- a/src/vpp-api/python/setup.py +++ b/src/vpp-api/python/setup.py @@ -30,7 +30,7 @@ setup( license='Apache-2.0', test_suite='vpp_papi.tests', install_requires=['cffi >= 1.6'] if stdlib_enum else - ['cffi >= 1.6', 'enum34'], + ['cffi >= 1.6', 'aenum'], packages=find_packages(), long_description='''VPP Python language binding.''', zip_safe=True) diff --git a/src/vpp-api/python/vpp_papi/vpp_serializer.py b/src/vpp-api/python/vpp_papi/vpp_serializer.py index d62e3a4a57a..e6012018f08 100644 --- a/src/vpp-api/python/vpp_papi/vpp_serializer.py +++ b/src/vpp-api/python/vpp_papi/vpp_serializer.py @@ -15,11 +15,15 @@ import struct import collections -from enum import IntEnum +import sys +if sys.version[0] == '2': + from aenum import IntEnum, IntFlag +else: + from enum import IntEnum, IntFlag import logging from . import vpp_format import ipaddress -import sys + import socket # @@ -276,7 +280,7 @@ class VPPEnumType(object): continue ename, evalue = f e_hash[ename] = evalue - self.enum = IntEnum(name, e_hash) + self.enum = IntFlag(name, e_hash) types[name] = self def __getattr__(self, name): |