Типизация объекта класса в Python

Question

У меня есть класс User и я сделал функцию, которая на вход принимает объект этого класса, вопрос как типизировать этот аргумент

class User:
    def __init__(self, *args): ...
    def something(self, user: ?): ...

Если поставить туда User то Python скажет:

NameError: name 'User' is not defined

Подскажите пожалуйста что делать? Заранее спасибо!

Answer 1

Вот несколько вариантов:

from __future__ import annotations

class User:
    def __init__(self, *args): ...
    def something(self, user: User): ...

from typing import TypeVar

t = TypeVar('User', bound='User')

class User:
    def __init__(self, *args): ...
    def something(self, user: t): ...

class User:
    def __init__(self, *args): ...
    def something(self, user: 'User'): ...

Я бы рекомендовал использовать __future__, потому что он был запланирован как стандартное включение, но пока что релиз в стандарт отложили дважды.