Python中函数和构造函数的枚举

Question

我正在学习Python，并为PyGame库建立边界。它没有错，我只是想让方法和类更适合我的特定项目。简而言之，我想做一个颜色枚举，但枚举中的每个值都有RGB和Hex版本。由于我以前在Java中做过这件事，所以我将大致展示我在Java枚举中的意思。

public enum Color {
RED(new double[]{255, 0, 0}, "#ff0000"),
CYAN(new double[]{0, 255, 255}, "#00e1ff");
// Other colors...
double[] rgb;
String hex;
Color(double[] rgb, String hex) {
this.rgb = rgb;
this.hex = hex;
}
public double[] getRGB() {
return rgb;
}
public String getHex() {
return hex;
}
}

我在这里所做的就是为枚举创建一个构造函数(在Java中默认为私有(，并使枚举中的每个项都指定该构造函数中的参数。例如，RED通过构造函数指定rgb双数组等于｛255，0，0｝，十六进制字符串为"ff0000"。使用底部的方法，您可以在代码中的任何位置检索这些值。

最后，如果我将枚举导入到代码中，然后使用RED项，我可以说RED.getRGB((并获得一个双数组。我想在这个枚举中添加一堆颜色，但我也想知道它的语法，因为它适用于我程序的其他部分。

我想用Python来做这件事。我几乎不理解在Python中制作常规枚举，也不理解整个"class Clazz(Enum(："的东西，但我当然不知道如何在Python中完成上面的代码。我只是不知道语法。

Answer 1

不要想太多

class Color:
def __init__(self,r,g,b):
self.r = r
self.g = g
self.b = b
def __str__(self):
# return the string repr
return f"#{self.r:02X}{self.g:02X}{self.b:02X}"
def __getitem__(self,item):
# allow `color[0],color[1],color[2]` access
return [self.r,self.g,self.b][item]
def __iter__(self):
# cast to list
return iter([self.r,self.g,self.b])

class Colors:
RED = Color(255,0,0)
GREEN = Color(0,255,0)
BLUE = Color(0,0,255)
WHITE = Color(255,255,255)
BLACK = Color(0,0,0)

print(list(Colors.RED))
print(Colors.GREEN[1])
print("%s"%Colors.BLUE)

如果你真的想使用枚举类，你可以如下

from enum import Enum
class Color:
def __init__(self,r,g,b):
self.r = r
self.g = g
self.b = b
def toHex(self):
return f"#{self.r:02X}{self.g:02X}{self.b:02X}"

def toRGB(self):
return (self.r,self.g,self.b)
class Colors(Enum):
RED = Color(255,0,0)
GREEN = Color(0,255,0)
BLUE = Color(0,0,255)
WHITE = Color(255,255,255)
BLACK = Color(0,0,0)

print(Colors.RED)
print(Colors.GREEN.value.toRGB())
print("%s"%Colors.BLUE.value.toHex())