PyCharm中Django ORM属性提示缺失问题的解决方法

from django.db import models
from django.db.models.manager import Manager

class Customer(models.Model):
    name = models.CharField(max_length=100)
    email = models.EmailField(unique=True)
    
    # 添加类型提示
    objects: Manager['Customer'] = models.Manager()
    
    def __str__(self):
        return self.name

pip install django-stubs

# settings.py
import django_stubs_ext
django_stubs_ext.monkeypatch()

<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
  <component name="Django">
    <option name="myCustomEnabled" value="true" />
    <option name="myProjectRoot" value="$PROJECT_DIR$" />
    <option name="mySettingsModule" value="your_project/settings.py" />
    <option name="myManageScript" value="$PROJECT_DIR$/manage.py" />
  </component>
</project>

from django.db import models
from typing import TypeVar, Generic

_M = TypeVar('_M', bound='Model')

class CustomManager(models.Manager):
    def active(self) -> models.QuerySet[_M]:
        return self.filter(is_active=True)

class Customer(models.Model):
    name = models.CharField(max_length=100)
    email = models.EmailField(unique=True)
    is_active = models.BooleanField(default=True)
    
    # 使用自定义管理器
    objects = CustomManager()
    
    def __str__(self):
        return self.name

my_project/
├── my_project/
│   ├── __init__.py
│   ├── settings.py
│   ├── urls.py
│   └── wsgi.py
├── apps/
│   └── customers/
│       ├── __init__.py
│       ├── models.py
│       ├── admin.py
│       └── ...
├── manage.py
└── requirements.txt

python manage.py shell

from customers.models import Customer
Customer.objects.all()  # 应该能正常执行

Customer.objects.  # 输入点号后应显示
                   # all() | filter() | get() | create() ...

# customers/models.py
from django.db import models
from django.db.models.manager import Manager
from typing import TypeVar, Generic

_M = TypeVar('_M', bound='Model')

class CustomManager(models.Manager):
    """自定义管理器增强提示"""
    def active(self) -> models.QuerySet[_M]:
        """获取活跃客户"""
        return self.filter(is_active=True)
    
    def by_email(self, email: str) -> models.QuerySet[_M]:
        """按邮箱查询"""
        return self.filter(email__iexact=email)

class Customer(models.Model):
    name = models.CharField(max_length=100)
    email = models.EmailField(unique=True)
    is_active = models.BooleanField(default=True)
    created_at = models.DateTimeField(auto_now_add=True)
    
    # 显式声明管理器类型
    objects: CustomManager = CustomManager()
    
    class Meta:
        verbose_name = '客户'
        verbose_name_plural = '客户管理'
        ordering = ['-created_at']
    
    def __str__(self):
        return f"{self.name} <{self.email}>"

from pydantic import BaseModel
from django.db import models
from typing import Optional

# Pydantic 模型用于类型提示
class CustomerSchema(BaseModel):
    id: int
    name: str
    email: str
    is_active: bool
    
    class Config:
        orm_mode = True

class Customer(models.Model):
    # ... 模型字段定义 ...
    
    @classmethod
    def get_by_id(cls, customer_id: int) -> Optional[CustomerSchema]:
        """获取客户并返回 Pydantic 模型"""
        try:
            customer = cls.objects.get(id=customer_id)
            return CustomerSchema.from_orm(customer)
        except cls.DoesNotExist:
            return None

Customer.objects  # type: ignore