Làm thế nào để xác định "nullable" kiểu trả về với kiểu gợi ý

Question

Giả sử tôi có một chức năng:

def get_some_date(some_argument: int=None) -> %datetime_or_None%: 
    if some_argument is not None and some_argument == 1: 
     return datetime.utcnow() 
    else: 
     return None 

Làm thế nào để xác định kiểu trả về cho một cái gì đó mà có thể None?

Answer 1

Kể từ kiểu trả về của bạn, hoặc có thể datetime (như trở về từ datetime.utcnow()) hoặc None bạn nên sử dụng Optional[datetime]:

from typing import Optional 

def get_some_date(some_argument: int=None) -> Optional[datetime]: 
    # as defined 

Từ các tài liệu, Optional là viết tắt cho:

Optional[X] tương đương đến Union[X, None].

nơi Union[X, Y] là giá trị loại X hoặc Y.

---

Nếu bạn muốn trở thành rõ ràng do lo ngại rằng những người khác có thể vấp ngã trên Optional và không nhận ra đó là ý nghĩa, bạn luôn có thể sử dụng Union:

from typing import Union 

def get_some_date(some_argument: int=None) -> Union[datetime, None]: 

Nhưng tôi nghi ngờ đây là một ý tưởng tốt, Optional là một tên chỉ dẫn và nó lưu một vài tổ hợp phím.

_{Như được nêu trong các nhận xét của @ Michael0x2a Union[T, None] được chuyển thành Union[T, type(None)] vì vậy không cần sử dụng type tại đây.}

Hình ảnh này có thể khác nhau nhưng theo chương trình, trong cả hai trường hợp, kết quả là giống hệt nhau; Union[datetime.datetime, NoneType] sẽ là loại lưu trữ trong get_some_date.__annotations__^*:

>>> from typing import get_type_hints 
>>> print(get_type_hints(get_some_date)) 
{'return': typing.Union[datetime.datetime, NoneType], 
'some_argument': typing.Union[int, NoneType]} 

^* Sử dụng typing.get_type_hints để lấy các đối tượng __annotations__ thuộc tính thay vì truy cập trực tiếp.