如何在Python中实现以下C代码？ （typedef，enums和switch）

Question

我在C中有以下部分代码：

typedef enum {
    CAN_NORMAL_MODE = 0x00,
    CAN_SLEEP_MODE = 0x01,
    CAN_INTERNAL_LOOPBACK_MODE = 0x02,
    CAN_LISTEN_ONLY_MODE = 0x03,
    CAN_CONFIGURATION_MODE = 0x04,
    CAN_EXTERNAL_LOOPBACK_MODE = 0x05,
    CAN_CLASSIC_MODE = 0x06,
    CAN_RESTRICTED_MODE = 0x07,
    CAN_INVALID_MODE = 0xFF
} CAN_OPERATION_MODE;

//more code here in the middle...

d = something 
switch (d) {
    case CAN_NORMAL_MODE:
        mode = CAN_NORMAL_MODE;
        break;
    case CAN_SLEEP_MODE:
        mode = CAN_SLEEP_MODE;
        break;
   // more cases here, not shown for readability
    default:
        mode = CAN_INVALID_MODE;
        break;
}

这里d是我从其他来源读取的字节。 switch语句在另一个函数内部，并且有更多代码，但我认为没有必要显示，因为它与此部分无关。

我的问题是：我正在尝试将此代码转换为Python，原因在于它与原始C代码（即，我不希望Python代码完全不同，因此，我不希望Python代码完全不同）已经使用原始C代码可以轻松理解和使用Python代码）。我不确定如何在Python语言中实现此代码片段中的功能，而无需进行无休止的if-elif ... else语句流。我认为应该有一种更容易（pythonic）的方法，但我不确定如何在Python中实现typedef，enum和switch语句。我已经阅读了一些实现enum子句的方法来创建一个以别名和值作为属性的类，如下所示：

class myClass():
    def __init__(self):
        self.CAN_NORMAL_MODE = 0x00
        self.CAN_SLEEP_MODE = 0x01
        self.CAN_INTERNAL_LOOPBACK_MODE = 0x02
    #etc

我还遇到了一种在Python中实现switch语句的聪明方法，它具有这样的函数（与我的实际问题没有任何关系，只是为了显示实现的结构）：

def switch(argument):
    switcher = {
        1: "January",
        2: "February",
        3: "March",
        4: "April",
        5: "May",
        6: "June",
        7: "July",
        8: "August",
        9: "September",
        10: "October",
        11: "November",
        12: "December"
    }
    print switcher.get(argument, "Invalid month")

但我无法以一种简单易懂的方式将这两件事结合起来。谢谢你，祝你有美好的一天！

Answer 1

对于枚举本身，你似乎正在变得比它需要的更难。枚举可以非常干净地映射到只包含一堆变量定义的python模块，例如：

惨_operation_摸得.朋友:

CAN_NORMAL_MODE = 0x00
CAN_SLEEP_MODE = 0x01
CAN_INTERNAL_LOOPBACK_MODE = 0x02
CAN_LISTEN_ONLY_MODE = 0x03
CAN_CONFIGURATION_MODE = 0x04
CAN_EXTERNAL_LOOPBACK_MODE = 0x05
CAN_CLASSIC_MODE = 0x06
CAN_RESTRICTED_MODE = 0x07
CAN_INVALID_MODE = 0xFF

这提供了有用的命名空间（C枚举没有），但也能够将这些名称导入另一个模块，以便使用它们C风格，没有模块名称。当然，权衡是它们是可修改的。

至于switch，不，Python与C的switch没有直接的类比，至少如果你取消了你在问题中描述的if / else if / else构造的类别：

other_module.朋友:

import can_operation_mode.*

d = something 
if   d == CAN_NORMAL_MODE:
    mode = CAN_NORMAL_MODE
elif d == CAN_SLEEP_MODE:
    mode = CAN_SLEEP_MODE
// more cases here ...
else:
    mode = CAN_INVALID_MODE

坦率地说，从保持与原始C的句法相似性的角度来看，我没有看到它有什么问题。它看起来非常类似于我，直到使用冒号。

Answer 2

from enum import IntEnum

class CAN_OPERATION_MODE(IntEnum):
    NORMAL_MODE = 0x00
    SLEEP_MODE = 0x01
    INTERNAL_LOOPBACK_MODE = 0x02
    LISTEN_ONLY_MODE = 0x03
    CONFIGURATION_MODE = 0x04
    EXTERNAL_LOOPBACK_MODE = 0x05
    CLASSIC_MODE = 0x06
    RESTRICTED_MODE = 0x07
    INVALID_MODE = 0xFF

    @classmethod
    def _missing_(cls, value):
        return cls.INVALID_MODE

d = something
mode = CAN_OPERATION_MODE(d)

我也会删除所有枚举名称中的_MODE。

Answer 3

字典怎么样？您可以将每个值分配给相应的模式，然后调用它：

modes = {
    0x00 : 'CAN_NORMAL_MODE',
    0x01 : 'CAN_SLEEP_MODE',
    0x02 : 'CAN_INTERNAL_LOOPBACK_MODE',
    0x03 : 'CAN_LISTEN_ONLY_MODE',
    0x04 : 'CAN_CONFIGURATION_MODE',
    0x05 : 'CAN_EXTERNAL_LOOPBACK_MODE',
    0x06 : 'CAN_CLASSIC_MODE',
    0x07 : 'CAN_RESTRICTED_MODE',
    0xFF : 'CAN_INVALID_MODE'
}

d = something
mode = modes[d]