Làm thế nào để sử dụng gợi ý loại Python với Django QuerySet?

Question

Có thể chỉ định loại bản ghi trong Django QuerySet với gợi ý kiểu Python không? Một cái gì đó như QuerySet[SomeModel]?

Ví dụ, chúng ta có mô hình:

class SomeModel(models.Model): 
    smth = models.IntegerField() 

Và chúng tôi muốn vượt qua QuerySet của mô hình đó như param trong func:

def somefunc(rows: QuerySet): 
    pass 

Nhưng làm thế nào để xác định loại bản ghi trong QuerySet, như với List[SomeModel]:

def somefunc(rows: List[SomeModel]): 
    pass 

nhưng với QuerySet?

Answer 1

Nếu bạn muốn vượt qua một param chỉ làm điều này:

def somefunc(smth): 
    pass 

nếu bạn muốn trả về một hàm sử dụng loại danh sách list().

Answer 2

Một giải pháp có thể đang sử dụng lớp nhập Union.

from typing import Union, List 
from django.db.models import QuerySet 
from my_app.models import MyModel 

def somefunc(row: Union[QuerySet, List[MyModel]]): 
    pass 

Bây giờ khi bạn cắt lập luận row nó sẽ biết rằng các loại trở lại là một trong hai một danh sách của MyModel hoặc một thể hiện của MyModel, trong khi cũng gợi ý rằng các phương pháp của lớp QuerySet có sẵn trên lập luận row quá .

Answer 3

Tôi cũng đang tìm giải pháp cho vấn đề này. Danh sách Nhà phát triển Django có một số thread, trong đó họ đang thảo luận cách triển khai tính năng như vậy.

Họ hiện đang phát triển một Django extension to mypy, nhưng có vẻ như chúng tôi có thể không may mắn cho yêu cầu cụ thể của chúng tôi. Trong lộ trình của họ dưới tiêu đề của "Có lẽ Never":

queryset thể có một số hỗ trợ một phần, nhưng lập luận phức tạp (như những người cho bộ lọc và nhận được truy vấn) hoặc Q và các đối tượng F nằm ngoài biểu cảm khả năng của mypy như bây giờ.

Nói rằng, chúng tôi sẽ chỉ phải sử dụng đồng bằng ole QuerySet cho đến khi điều kiện được cải thiện.