如何将webdriver类型代码作为整数

from enum import Enum

class WebDriverCode(Enum):
    CHROME = 1
    FIREFOX = 2
    SAFARI = 3
    EDGE = 4

def get_int_value(webdriver_code):
    return webdriver_code.value

def get_webdriver_code(int_value):
    for code in WebDriverCode:
        if code.value == int_value:
            return code

# 将webdriver类型代码转换为整数
webdriver_type = WebDriverCode.CHROME
int_value = get_int_value(webdriver_type)
print(int_value)  # 输出：1

# 将整数转换为webdriver类型代码
int_value = 3
webdriver_type = get_webdriver_code(int_value)
print(webdriver_type)  # 输出：WebDriverCode.SAFARI