我一直在尝试在Django管理页面中创建一个自定义上载按钮,但我不断收到一个指向CsvUploader.py文件的错误:object has no attribute 'model'
我有一个非常简单的模型:
class Link(models.Model):
name = models.CharField(db_column='Name', max_length=50)
url = models.URLField(db_column="URL", max_length=500)
def __str__(self):
return f"{self.name}: {self.url}"
我还修改了我的管理模板,如下所示:
#Extend Admin portal use
class CsvUploadAdmin(admin.ModelAdmin):
change_list_template = "custom_admin/csv_form.html"
def get_urls(self):
urls = super().get_urls()
additional_urls = [
path("upload-csv/", self.upload_csv),
]
return additional_urls + urls
urls = property(get_urls)
def changelist_view(self, request, extra_context=None):
extra = extra_context or {}
extra["csv_upload_form"] = CsvUploadForm()
return super(CsvUploadAdmin, self).changelist_view(request, extra_context=extra)
def upload_csv(self, request):
if request.method == "POST":
form = CsvUploadForm(request.POST, request.FILES)
if form.is_valid():
if request.FILES['csv_file'].name.endswith('csv'):
try:
decoded_file = request.FILES['csv_file'].read().decode('utf-8')
except UnicodeDecodeError as e:
self.message_user(
request,
"There was an error decoding the file:{}".format(e),
level=messages.ERROR
)
return redirect("..")
# Here we will call our class method
io_string = io.StringIO(decoded_file)
uploader = CsvUploader(io_string, self.model, portal)
result = uploader.create_records()
else:
self.message_user(
request,
"Incorrect file type: {}".format(
request.FILES['csv_file'].name.split(".")[1]
),
level=messages.ERROR
)
else:
self.message_user(
request,
"There was an error in the form {}".format(form.errors),
level=messages.ERROR
)
return redirect("..")
@admin.register(Link)
class LinkAdmin(CsvUploadAdmin):
pass
最后,这里是我的CsvUploader.py文件:
import csv
from portal.models import Link
from django.db.utils import IntegrityError
from django.core.exceptions import FieldDoesNotExist
from django.db import transaction
from django.conf.urls import *
class CsvUploader:
def __init__(self, csv_file, model_name, app_name):
self.reader = list(csv.DictReader(csv_file, delimiter=','))
self.keys = [k for k in self.reader[0]]
self.model_fields = [f.name for f in self.model._meta.get_fields()]
self.valid = self._validate_csv()
self.csv_pos = 0
def _validate_csv(self):
keys = []
for k in self.keys:
if k.endswith("_id"):
keys.append(k[:-3])
else:
keys.append(k)
return set(keys).issubset(self.model_fields)
def read_chunk(self):
chunk = []
for i in range(1000):
try:
chunk.append(self.model(**self.reader[self.csv_pos]))
except IndexError as e:
print(e)
break
self.csv_pos += 1
return chunk
def create_records(self):
if not self.valid:
return "Invalid csv file"
while True:
chunk = self.read_chunk()
if not chunk:
break
try:
with transaction.atomic():
self.model.objects.bulk_create(chunk)
except IntegrityError as e:
for i in chunk:
try:
i.save()
except IntegrityError:
continue
print("Exception: {}".format(e))
return "records successfully saved!"
我希望在这方面能有第二双眼睛,可以发现上面第13行的错误:
“CsvUploader”对象没有属性“model”
申请方式:邮寄
请求URL:http://127.0.0.1:8000/admin/portal/link/upload-csv/
Django版本:3.2.4
异常类型:AttributeError
异常值:
“CsvUploader”对象没有属性“model”
非常感谢您的帮助
问题是您希望self.model是
Link
模型对象,因为您在CsvUploadAdmin
中调用了如下内容:-这是不正确的,在
__init__
方法中定义的model_name变量中有self.model
值,例如:-所以,为了解决您的问题,您需要在
__init__
方法的下面添加一行所以最后的方法是
相关问题 更多 >
编程相关推荐