Ignore incompatible type error when variable is checked in if statement

I'm using mypy for type checking in my Python projects and I've come across a situation where I believe mypy could be improved.

Consider the following code:

def foo1(a: str): ...

def foo2(a: str | None, b: str | None):
    if not a and not b:
        return None
    if not a:
        return foo1(b)
    return None

In this case, mypy raises an error:

Argument 1 to "foo1" has incompatible type "str | None"; expected "str" [arg-type].

However, the function foo1 is only called when b
is not None, so in practice, there is no type error.

The logic of the function ensures that b cannot be None when foo1(b) is called.
This is because in the first if statement, we check if both a and b are None.
If they are not, we proceed to the next if statement where we check if a is None.
If a is None, then b must not be None because of the previous check.
Therefore, foo1(b) is only called when b is not None.

Versions

Python: 3.10.10
MyPy: 1.7.1

Bắt đầu bằng cách chạy mypy trên trình tái hiện Python được cung cấp và theo dõi cách hai câu lệnh if ảnh hưởng đến việc thu hẹp kiểu. Xác định các kiểm thử kiểm tra kiểu liên quan hoặc entry point, sau đó thêm một trường hợp hồi quy cho luồng điều khiển này. Hoàn tất khi mypy chấp nhận foo1(b) mà không có lỗi arg-type, đồng thời giữ nguyên các kiểm tra hiện có.