Geben Sie Hinweise mit benutzerdefinierten Klassen ein

Konnte keine endgültige Antwort finden. Ich möchte einen Typhinweis für eine Funktion erstellen, und der Typ ist eine benutzerdefinierte Klasse, die ich definiert habe und die ich sie nenne CustomClass().

Und dann sagen wir in einer Funktion, nennen wir es FuncA(arg), ich habe ein Argument namens arg. Wäre die richtige Art, einen Hinweis einzugeben FuncA, Folgendes :

def FuncA(arg: CustomClass):

Oder wäre es:

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

Die erstere ist richtig , wenn argakzeptiert eine InstanzCustomClass :

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

Wenn Sie die Klasse CustomClassselbst (oder einen Subtyp) möchten , sollten Sie schreiben:

from typing import Type  # you have to import Type

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

Wie es in der Dokumentation über Typing geschrieben steht :

class typing.Type(Generic[CT_co])

Eine mit annotierte Variable Ckann einen Wert vom Typ annehmen C. Im Gegensatz dazu kann eine mit Annotationen verseheneType[C] Variable Werte akzeptieren, die selbst Klassen sind. Insbesondere akzeptiert sie das Klassenobjekt vonC .

Die Dokumentation enthält ein Beispiel mit der intKlasse:

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