Python typing.Literal转list

waketzheng

已于 2024-11-01 10:56:09 修改

阅读量175

点赞数 1

文章标签： python linux 开发语言

于 2024-11-01 10:51:45 首次发布

本文链接：https://blog.youkuaiyun.com/jaket5219999/article/details/143426356

版权

直接用typing.get_args即可

代码如下：

from typing import Literal, get_args

system = Literal['linux', 'win32', 'darwin']

system_tuple = get_args(system)
print(system_tuple)
# ('linux', 'win32', 'darwin')
assert system_tuple == ('linux', 'win32', 'darwin')

system_list = list(get_args(system))
print(system_list)
# ('linux', 'win32', 'darwin')
assert system_list == ['linux', 'win32', 'darwin']

Usage

import sys
import typing

SystemName = typeing.Literal['linux', 'win32', 'darwin']


def get_os_system() -> SystemName:
    return sys.platform


def is_system_name(value: str) -> bool:
    return value in typing.get_args(SystemName)