在Dataclass Reader中使用Typing.Literal

Question

我有一个CSV文件，我想将其处理到数据级别。我想检查数据集中的成绩仅来自预定列表，如果不是这样，我想记录错误/警告。我的课程看起来如下，

from dataclasses import dataclass
from typing import Literal

grade_options = Literal['1A', '1B', '1C']

class Student:
    name: str
    age: int
    grade: grade_options

我读了我的CSV文件（使用dataclass-csv 库），但是实例化此类型存在问题

from dataclass_csv import DataclassReader

with open('students.csv', encoding="utf-8-sig") as read_csv:
    reader = DataclassReader(read_csv, Student, delimiter=";")
    students = [student for student in reader]

，这将导致typeError：无法实例化键入。文字

除了创建手动检查器以查看我的值是否在CSV文件是指定的选项之一？

Answer 1

我自己遇到了这个。您必须使用扩展enum的具体类。

基于“创建枚举类”， a>：

from enum import Enum


class Grade(str, Enum):
    _1a: '1A'
    _1b: '1B'
    _1c: '1C'

class Student:
    name: str
    age: int
    grade: Grade