Mypy: Type of the same class inside the class

[mrkaspa]

I have a method that returns itself but it doesn't know the type of the class inside

class TcpFlow(object):

    def __init__(self) -> None:
        self.matches = []  # type: List[str]

    def __enter__(self) -> TcpFlow:
        return self

I got this error

Traceback (most recent call last):
  File "example/sample.py", line 2, in <module>
    from tcpflow import TcpFlow
  File "/Users/mrkaspa/code/py/tcpflow.py/tcpflow/__init__.py", line 1, in <module>
    from .tcpflow import TcpFlow
  File "/Users/mrkaspa/code/py/tcpflow.py/tcpflow/tcpflow.py", line 8, in <module>
    class TcpFlow(object):
  File "/Users/mrkaspa/code/py/tcpflow.py/tcpflow/tcpflow.py", line 16, in TcpFlow
    def __enter__(self) -> TcpFlow:
NameError: name 'TcpFlow' is not defined

[ilevkivskyi]

In such cases you should put the name of class in quotes (this is called forward reference), like this:

class TcpFlow(object):
    def __enter__(self) -> 'TcpFlow':
        return self

[Garrett-R]

As of Python 3.7, you can now do:

from __future__ import annotations

class TcpFlow(object):
    def __enter__(self) -> TcpFlow:
        return self

(notice the quotes around the return type have been removed)