Tapez des indices avec des classes définies par l'utilisateur

Impossible de trouver une réponse définitive. Je veux faire un indice de type pour une fonction et le type étant une classe personnalisée que j'ai définie, appelée CustomClass().

Et puis disons dans une fonction, appelez-la FuncA(arg), j'ai un argument nommé arg. La manière correcte de taper un indice FuncAserait-elle:

def FuncA(arg: CustomClass):

Ou serait-ce:

def FuncA(Arg:Type[CustomClass]):?

Le premier est correct , s'il argaccepte une instance deCustomClass :

def FuncA(arg: CustomClass):
    #     ^ instance of CustomClass

Au cas où vous voudriez la classe CustomClasselle - même (ou un sous-type) , vous devez écrire:

from typing import Type  # you have to import Type

def FuncA(arg: Type[CustomClass]):
    #     ^ CustomClass (class object) itself

Comme il est écrit dans la documentation sur la saisie :

class typing.Type(Generic[CT_co])

Une variable annotée avec Cpeut accepter une valeur de type C. En revanche, une variable annotée avec Type[C]peut accepter des valeurs qui sont elles - mêmes des classes - en particulier, elle acceptera l' objet de classe deC .

La documentation comprend un exemple avec la intclasse:

a = 3         # Has type 'int'
b = int       # Has type 'Type[int]'
c = type(a)   # Also has type 'Type[int]'