Digitare suggerimenti con classi definite dall'utente

Non riesco a trovare una risposta definitiva. Voglio fare un suggerimento sul tipo per una funzione e il tipo è una classe personalizzata che ho definito, chiamata CustomClass().

E poi diciamo in qualche funzione, chiamala FuncA(arg), ho un argomento chiamato arg. Il modo corretto di digitare hint FuncAsarebbe:

def FuncA(arg: CustomClass):

Oppure sarebbe:

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

Il primo è corretto , se argaccetta un'istanza diCustomClass :

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

Nel caso in cui desideri la classe CustomClassstessa (o un sottotipo) , dovresti scrivere:

from typing import Type  # you have to import Type

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

Come è scritto nella documentazione sulla digitazione :

class typing.Type(Generic[CT_co])

Una variabile annotata con Cpuò accettare un valore di tipo C. Al contrario, una variabile annotata con Type[C]può accettare valori che sono classi stesse - in particolare, accetterà l' oggetto classe diC .

La documentazione include un esempio con la intclasse:

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