Início / coding / Perguntas / 79000754
Accepted
Pro Q
Asked: 2024-09-19 11:25:17 +0800 CST

Como digitar uma dica de uma classe de dados criada dinamicamente

  • 772

Eu odeio escrever as coisas duas vezes, então eu inventei uma maneira decente de não ter que escrever as coisas duas vezes. No entanto, isso parece quebrar meu type-hinting:

from enum import Enum
from dataclasses import make_dataclass, field, dataclass

class DatasetNames(Enum):
    test1 = "test1_string"
    test2 = "test2_string"
    test3 = "test3_string"

def get_path(s: str) -> str:
    return s + "_path"

# the normal way to do this, but I have to type every new dataset name twice
# and there's a lot of duplicate code
@dataclass(frozen=True)
class StaticDatasetPaths:
    test1 = get_path("test1_string")
    test2 = get_path("test2_string")
    test3 = get_path("test3_string")

# mypy recognizes that `StaticDatasetPaths` is a class
# mypy recognizes that `StaticDatasetPaths.test2` is a string
print(StaticDatasetPaths.test2) # 'test2_string_path'

# this is my way of doing it, without having to type every new dataset name twice and no duplicate code
DynamicDatasetPaths = make_dataclass(
    'DynamicDatasetPaths', 
    [
        (
            name.name,
            str,
            field(default=get_path(name.value))
        )
        for name in DatasetNames
    ],
    frozen=True
)

# mypy thinks `DynamicDatasetPaths` is a `variable` of type `type`
# mypy thinks that `DynamicDatasetPaths.test2` is an `function` of type `Unknown`
print(DynamicDatasetPaths.test2) # 'test2_string_path'

Como posso informar ao mypy que DynamicDatasetPaths é uma classe de dados congelada cujos atributos são strings?

Normalmente, quando me deparo com casos como esse, consigo simplesmente usar a caste dizer ao mypy qual é o tipo correto, mas não sei o tipo correto para "classe de dados congelada cujos atributos são strings".

(Além disso, se houver uma maneira melhor de não ter o código duplicado, ficaria feliz em saber sobre isso também.)

python

1 respostas

  1. Best Answer

    Uma classe de dados serve para criar instâncias. Como você não está instanciando a classe de dados, mas acessando test1, test2, etc. como atributos de classe, você realmente não precisa de uma classe de dados, mas pode simplesmente criar pathuma propriedade da Enumclasse. E como todos os membros da sua Enumclasse têm valores de string, você pode criar uma StrEnumclasse para operações de string mais fáceis:

    from enum import StrEnum

class DatasetNames(StrEnum):
    test1 = "test1_string"
    test2 = "test2_string"
    test3 = "test3_string"

    @property
    def path(self) -> str:
        return self + '_path'

print(DatasetNames.test2.path) # outputs test2_string_path

    Se a get_pathfunção for cara no seu caso de uso real, considere fazer pathuma cached_property.

    • 1

relate perguntas