在平時的開發過程中,避免不了需要獲取 Model 中的字段列表。
那需要把所有字段都再複製一份嗎?這樣的話就太麻煩了,而且後期也不好維護。
其實,Django 內置了一個方法,可以很輕鬆解決這個問題。
下面以 User 表舉一個例子。
>>> from django.contrib.auth.models import User
>>> User._meta.get_fields()
(<ManyToOneRel: admin.logentry>, <django.db.models.fields.AutoField: id>, <django.db.models.fields.CharField: password>, <django.db.models.fields.DateTimeField: last_login>, <django.db.models.fields.BooleanField: is_superuser>, <django.db.models.fields.CharField: username>, <django.db.models.fields.CharField: first_name>, <django.db.models.fields.CharField: last_name>, <django.db.models.fields.EmailField: email>, <django.db.models.fields.BooleanField: is_staff>, <django.db.models.fields.BooleanField: is_active>, <django.db.models.fields.DateTimeField: date_joined>, <django.db.models.fields.related.ManyToManyField: groups>, <django.db.models.fields.related.ManyToManyField: user_permissions>)
通過 get_fields()
方法會返回一個 tuple,每個元素都是 model field 類型。
這個結果並不是我們想要的,我們需要的是字段名稱列表。只需要對每個 field 字段取 name 屬性就可以了。
>>> [field.name for field in User._meta.get_fields()]
['logentry', 'id', 'password', 'last_login', 'is_superuser', 'username', 'first_name', 'last_name', 'email', 'is_staff', 'is_active', 'date_joined', 'groups', 'user_permissions']
這樣就可以了。
接下來,我們將其抽象出一個函數,這樣如果有需要直接調用就可以了。
def get_model_fields(model):
fields = [field.name for field in model._meta.get_fields()]
return fields
以上就是本文的全部內容,如果覺得還不錯的話,歡迎點贊,轉發和關注,感謝支持。
推薦閱讀: