django的新功能,我正在尝试弄清楚如何向django模型中添加特定字段。我想上传一个csv文件并保存其标题。如您所见,我想使用与文档一起发送的标头,或者如果它们没有发送,则使用文档第一行中的标头。
from django.db import models
class CSV_API_DB(models.Model):
headers = models.CharField(max_length=250, blank=True)
delimiter = models.CharField(max_length=1, default=';')
filefield = models.FileField(upload_to='uploads/', max_length=100, default='empty')
actual_headers = ''
def __str__(self):
actual_headers = ''
if not self.headers:
file_path = str(self.filefield)
file = open(file_path)
for line in file:
string_line = str(line[:-1])
actual_headers = string_line.split(self.delimiter)
break
else:
actual_headers = self.headers.split(self.delimiter)
return str(actual_headers)
true_headers = models.CharField(max_length=250, default = str(actual_headers), editable=False)
问题似乎是,true_headers不会从'__ str __'函数中被覆盖,因为数据库中的true_headers值始终只是空字符串。
default
参数在迁移过程中被调用,但在创建新对象时不被调用。要提供动态默认值,您应该将可调用对象传递给default=
arg。
因此,要提供动态默认值,您需要传递一个返回它的函数,而不仅仅是字符串。像这样:
...
some_field = fields.CharField(max_length=10, default=get_some_field) # note that no () after function name
def get_some_field():
return some_dynamic_value
但是请注意,您无法从被调用的函数访问模型实例,因此,如果可能,必须使用YourModel.objects.get(...)
进行搜索,或者仅使用其他方式设置动态值。例如,您可以覆盖save()方法:
class CSV_API_DB(models.Model):
...
def save(self, *args, **kwargs):
if not self.true_headers and self._state.adding: # _state.adding indicates that object is being created (not just edited), so we have to set default value if none provided
self.true_headers = self.headers.split(self.delimiter)
super().save(*args, **kwargs)