mirror of
https://github.com/davidhalter/django-stubs.git
synced 2025-12-07 04:34:29 +08:00
add support for get_FIELD_display for choices fields
This commit is contained in:
20
test-data/typecheck/models/test_extra_methods.yml
Normal file
20
test-data/typecheck/models/test_extra_methods.yml
Normal file
@@ -0,0 +1,20 @@
|
||||
- case: if_field_has_choices_set_model_has_get_FIELDNAME_display_method
|
||||
main: |
|
||||
from myapp.models import MyUser
|
||||
user = MyUser(name='user', gender='M')
|
||||
user.get_name_display() # E: "MyUser" has no attribute "get_name_display"; maybe "get_gender_display"?
|
||||
reveal_type(user.get_gender_display()) # N: Revealed type is 'builtins.str'
|
||||
installed_apps:
|
||||
- myapp
|
||||
files:
|
||||
- path: myapp/__init__.py
|
||||
- path: myapp/models.py
|
||||
content: |
|
||||
from django.db import models
|
||||
GENDER_CHOICES = (
|
||||
('M', 'Male'),
|
||||
('F', 'Female'),
|
||||
)
|
||||
class MyUser(models.Model):
|
||||
name = models.CharField(max_length=100)
|
||||
gender = models.CharField(max_length=100, choices=GENDER_CHOICES)
|
||||
Reference in New Issue
Block a user