Ketik petunjuk dengan kelas yang ditentukan pengguna

Sepertinya tidak dapat menemukan jawaban yang pasti. Saya ingin melakukan petunjuk tipe untuk suatu fungsi dan tipe tersebut menjadi beberapa kelas khusus yang telah saya tentukan, menyebutnya CustomClass().

Dan kemudian katakanlah di beberapa fungsi, sebut saja FuncA(arg), saya punya satu argumen bernama arg. Apakah cara yang benar untuk mengetik petunjuk FuncAadalah:

def FuncA(arg: CustomClass):

Atau apakah itu:

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

Yang pertama benar , jika argmenerima contoh dariCustomClass :

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

Jika Anda menginginkan kelas CustomClassitu sendiri (atau subtipe) , maka Anda harus menulis:

from typing import Type  # you have to import Type

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

Seperti yang tertulis dalam dokumentasi tentang Mengetik :

class typing.Type(Generic[CT_co])

Variabel yang dianotasi Cdapat menerima nilai tipe C. Sebaliknya, variabel yang dianotasi Type[C]dapat menerima nilai yang merupakan kelas itu sendiri - khususnya, ia akan menerima objek kelasC .

Dokumentasi menyertakan contoh dengan intkelas:

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