python学习笔记:利用getattr属性实现枚举Enum

最新推荐文章于 2025-02-18 21:38:42 发布

原创最新推荐文章于 2025-02-18 21:38:42 发布 · 670 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #enum

编程语言专栏收录该内容

30 篇文章

订阅专栏

本文介绍了两种自定义枚举类的方法：一种是通过 type 函数动态创建类，另一种是继承 object 类并重写 __getattr__ 方法。这两种方法分别适用于不同场景，并提供了具体的 Python 代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#自定义 enum
def enum(**enums):
	return type('Enum', (), enums)

m = enum(JAN=1, FEB=2, MAR=3, APR=4, MAY=5, JUN=6, JUL=7, AUG=8, SEP=9, OCT=10, NOV=11, DEC=12)
print('FEB = %d' % m.FEB)
d = {'JAN':1, 'FEB':2, 'MAR':3, 'APR':4, 'MAY':5, 'JUN':6, 'JUL':7, 'AUG':8, 'SEP':9, 'OCT':10, 'NOV':11, 'DEC':12}
m = enum(**d)  #返回的是一个类
print(m)
print(dir(m))
print('MAY = %d' % m.MAY)

#利用__getattr__属性实现枚举Enum
class Enum(object):
	def __init__(self, *enums_list, **enums_dict):
		if enums_list:
			self.__enums = list(enums_list)
			self.__type = 'list'
		elif enums_dict:
			self.__enums = dict(enums_dict)
			self.__type = 'dict'
		print(self.__enums)
	
	def __getattr__(self, attr):
		if attr in self.__enums:
			if self.__type == 'list':
				return self.__enums.index(attr) + 1
			elif self.__type == 'dict':
				return self.__enums[attr]

l = ['JAN', 'FEB', 'MAR', 'APR', 'MAY', 'JUN', 'JUL', 'AUG', 'SEP', 'OCT', 'NOV', 'DEC']
m = Enum(*l)
print('MAY = %d' % m.MAY)
d = {'JAN':1, 'FEB':2, 'MAR':3, 'APR':4, 'MAY':5, 'JUN':6, 'JUL':7, 'AUG':8, 'SEP':9, 'OCT':10, 'NOV':11, 'DEC':12}
m = Enum(**d)
print('MAY = %d' % m.MAY)