Python中的枚舉類型是什么
這篇文章將為大家詳細講解有關(guān)Python中的枚舉類型是什么，小編覺得挺實用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。
創(chuàng)新互聯(lián)建站自2013年創(chuàng)立以來，是專業(yè)互聯(lián)網(wǎng)技術(shù)服務(wù)公司，擁有項目網(wǎng)站制作、做網(wǎng)站網(wǎng)站策劃，項目實施與項目整合能力。我們以讓每一個夢想脫穎而出為使命，1280元呼和浩特做網(wǎng)站,已為上家服務(wù),為呼和浩特各地企業(yè)和個人服務(wù),聯(lián)系電話:18982081108
在Python中，枚舉和我們在對象中定義的類變量時一樣的，每一個類變量就是一個枚舉項，訪問枚舉項的方式為：類名加上類變量。
雖然這樣是可以解決問題的，但是并不嚴謹，也不怎么安全，比如：
1、枚舉類中，不應(yīng)該存在key相同的枚舉項（類變量）
2、不允許在類外直接修改枚舉項的值
下面來看看如何實現(xiàn)枚舉類型：
import operator
class EnumValue(object):
def __init__(self, parent_name, name, value):
self._parent_name = parent_name
self._name = name
self._value = value
def _parents_equal(self, other):
return (
hasattr(other, '_parent_name')
and self._parent_name == other._parent_name)
def _check_parents_equal(self, other):
if not self._parents_equal(other):
raise TypeError(
'This operation is valid only for enum values of the same type')
def __eq__(self, other):
return self._parents_equal(other) and self._value == other._value
def __ne__(self, other):
return not self.__eq__(other)
def __lt__(self, other):
self._check_parents_equal(other)
return self._value < other._value
def __le__(self, other):
self._check_parents_equal(other)
return self._value <= other._value
def __gt__(self, other):
self._check_parents_equal(other)
return self._value > other._value
def __ge__(self, other):
self._check_parents_equal(other)
return self._value >= other._value
def __hash__(self):
return hash(self._parent_name + str(self._value))
def __repr__(self):
return '{}({!r}, {!r}, {!r})'.format(
self.__class__.__name__, self._parent_name, self._name, self._value)
def __int__(self):
return int(self._value)
def __str__(self):
return str(self._name)
class EnumMetaclass(type):
def __new__(cls, name, bases, dct):
uppercased = dict((k.upper(), v) for k, v in dct.items())
new_dct = dict(
name=name,
_enums_by_str=dict(
(k, EnumValue(name, k, v)) for k, v in uppercased.items()),
_enums_by_int=dict(
(v, EnumValue(name, k, v)) for k, v in uppercased.items()),
)
return super(EnumMetaclass, cls).__new__(cls, name, bases, new_dct)
def __getattr__(cls, name):
try:
return cls.__getitem__(name)
except KeyError:
raise AttributeError
def __getitem__(cls, name):
try:
name = name.upper()
except AttributeError:
pass
try:
return cls._enums_by_str[name]
except KeyError:
return cls._enums_by_int[name]
def __repr__(cls):
return '{}({!r}, {})'.format(
cls.__class__.__name__,
cls.name,
', '.join('{}={}'.format(v._name, v._value)
for v in sorted(cls._enums_by_str.values())))
def values(cls):
return sorted(cls._enums_by_str.values())
def _values_comparison(cls, item, comparison_operator):
"""
Return a list of values such that comparison_operator(value, item) is
True.
"""
return sorted(
[v for v in cls._enums_by_str.values()
if comparison_operator(v, item)])
def values_lt(cls, item):
return cls._values_comparison(item, operator.lt)
def values_le(cls, item):
return cls._values_comparison(item, operator.le)
def values_gt(cls, item):
return cls._values_comparison(item, operator.gt)
def values_ge(cls, item):
return cls._values_comparison(item, operator.ge)
def values_ne(cls, item):
return cls._values_comparison(item, operator.ne)
def enum_factory(name, **kwargs):
return EnumMetaclass(name, (), kwargs)Python中枚舉的測試:
import unittest
class EnumTestCase(unittest.TestCase):
def test_repr(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(
repr(ProfileAction),
"EnumMetaclass('ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)")
def test_value_repr(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(
repr(ProfileAction.VIEW), "EnumValue('ProfileAction', 'VIEW', 1)")
def test_attribute_error(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
with self.assertRaises(AttributeError):
ProfileAction.ASDFASDF
def test_cast_to_str(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(str(ProfileAction.VIEW), 'VIEW')
def test_cast_to_int(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(int(ProfileAction.VIEW), 1)
def test_access_by_str(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(ProfileAction['VIEW'], ProfileAction.VIEW)
def test_access_by_int(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(ProfileAction[1], ProfileAction.VIEW)
def test_equality(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(ProfileAction.VIEW, ProfileAction.VIEW)
self.assertEqual(ProfileAction['VIEW'], ProfileAction.VIEW)
self.assertEqual(ProfileAction[1], ProfileAction.VIEW)
def test_inequality(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertNotEqual(ProfileAction.VIEW, ProfileAction.EDIT_OWN)
self.assertNotEqual(ProfileAction['VIEW'], ProfileAction.EDIT_OWN)
self.assertNotEqual(ProfileAction[1], ProfileAction.EDIT_OWN)
DashboardAction = enum_factory(
'DashboardAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertNotEqual(ProfileAction.VIEW, DashboardAction.VIEW)
def test_invalid_comparison(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
DashboardAction = enum_factory(
'DashboardAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
with self.assertRaises(TypeError) as cm:
ProfileAction.VIEW < DashboardAction.EDIT_OWN
self.assertEqual(
str(cm.exception),
'This operation is valid only for enum values of the same type')
def test_values(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(
ProfileAction.values(), [
EnumValue('ProfileAction', 'VIEW', 1),
EnumValue('ProfileAction', 'EDIT_OWN', 2),
EnumValue('ProfileAction', 'EDIT_PUBLIC', 3),
EnumValue('ProfileAction', 'EDIT_FULL', 4),
])
def test_values_lt(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(
ProfileAction.values_lt(ProfileAction.EDIT_PUBLIC), [
EnumValue('ProfileAction', 'VIEW', 1),
EnumValue('ProfileAction', 'EDIT_OWN', 2),
])
def test_values_le(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(
ProfileAction.values_le(ProfileAction.EDIT_PUBLIC), [
EnumValue('ProfileAction', 'VIEW', 1),
EnumValue('ProfileAction', 'EDIT_OWN', 2),
EnumValue('ProfileAction', 'EDIT_PUBLIC', 3),
])
def test_values_gt(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(
ProfileAction.values_gt(ProfileAction.EDIT_PUBLIC), [
EnumValue('ProfileAction', 'EDIT_FULL', 4),
])
def test_values_ge(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(
ProfileAction.values_ge(ProfileAction.EDIT_PUBLIC), [
EnumValue('ProfileAction', 'EDIT_PUBLIC', 3),
EnumValue('ProfileAction', 'EDIT_FULL', 4),
])
def test_values_ne(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
self.assertEqual(
ProfileAction.values_ne(ProfileAction.EDIT_PUBLIC), [
EnumValue('ProfileAction', 'VIEW', 1),
EnumValue('ProfileAction', 'EDIT_OWN', 2),
EnumValue('ProfileAction', 'EDIT_FULL', 4),
])
def test_intersection_with_same_type(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
set_a = set([ProfileAction.VIEW, ProfileAction.EDIT_OWN])
set_b = set([ProfileAction.VIEW, ProfileAction.EDIT_PUBLIC])
self.assertEqual(set_a & set_b, set([ProfileAction.VIEW]))
def test_intersection_with_different_types(self):
ProfileAction = enum_factory(
'ProfileAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
DashboardAction = enum_factory(
'DashboardAction', VIEW=1, EDIT_OWN=2, EDIT_PUBLIC=3, EDIT_FULL=4)
set_a = set([ProfileAction.VIEW, ProfileAction.EDIT_OWN])
set_b = set([DashboardAction.VIEW, DashboardAction.EDIT_PUBLIC])
self.assertEqual(set_a & set_b, set([]))關(guān)于Python中的枚舉類型是什么就分享到這里了，希望以上內(nèi)容可以對大家有一定的幫助，可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到。
網(wǎng)站名稱：Python中的枚舉類型是什么
URL鏈接：http://jinyejixie.com/article46/ggedhg.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供網(wǎng)站制作、電子商務(wù)、關(guān)鍵詞優(yōu)化、網(wǎng)站營銷、網(wǎng)站內(nèi)鏈、營銷型網(wǎng)站建設(shè)
聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)
- 馬云回國，首談ChatGPT。又是新一個風口？ 2023-05-28
- ChatGPT是什么 2023-02-20
- 怎樣利用chatGPT快速賺錢？ 2023-05-05
- ChatGPT的發(fā)展歷程 2023-02-20
- 火爆的ChatGPT，來聊聊它的熱門話題 2023-02-20
- ChatGPT的應(yīng)用ChatGPT對社會的利弊影響 2023-02-20
- 爆紅的ChatGPT，誰會丟掉飯碗？ 2023-02-20
- ChatGPT是什么?ChatGPT是聊天機器人嗎? 2023-05-05