自定义通用视图
web后端开发的工作就是对数据的增删改查, 回顾前面的各种功能的代码,会发现有很多的代码冗余。我们一直在做重复的事情。
django框架的一个强大的功能就是提供了一个即插即用的管理后台,django-admin。我不会教大家去开发一个django-admin,这对于现阶段的大家来说太过复杂和庞大。
先学会一点点的优化,不要尝试一步到位。我们的目的是复用代码, 将相同的视图写成即插即用的通用视图
I. MyListView(列表视图)
后台管理功能中,总是需要以数据表的形式展示数据, 复杂一点会有过滤,查询,分页等操作。
但它们的业务逻辑一致,都是接收请求,然后去数据库中获取数据,再进行过滤,分页,然后展示。
那么我们就可以抽象出一个MyListView
来完成这些通用功能,只需要做少量参数配置就可以实现对不同的模型的数据展示。
将相同的代码抽取出来, 把它做成通用的
1>通用的类属性
找出通用属性,定义在类属性中,继承它的类通过覆盖这些属性来完成自定义。
# 创建myadmin/generic.py并写入以下代码
class MyListView(View):
model = None # 模型
template_name = None # 模板名称
is_paginate = False # 是否分页
per_page = None # 每页条数
page_header = None # 页头大标题
page_option = None # 页头小标题
table_title = None # 内容标题
fields = None # 需要展示的字段
2>方法
2.1>get
get方法是用来响应get请求的,根据请求,调用相应模型,查询数据然后渲染模板。
def get(self, request):
context = self.get_context_data()
return render(request, self.get_template_name(), context=context)
2.2>get_template_name
获取模板名称,如果不提供类属性template_name,会根据模型名称拼接出默认的模板myadmin/model_name/model_name_list.html
def get_template_name(self):
"""获取模板名"""
if self.template_name is None:
self.template_name = 'myadmin/{0}/{0}_list.html'.format(self.model._meta.model_name)
return self.template_name
2.3>get_context_data
获取视图的上下文变量,如果需要额外添加,可以复写次方法。
def get_context_data(self, **kwargs):
"""获取视图的上下文变量,如要添加额外变量,请复写此方法"""
queryset = self.get_queryset()
if self.is_paginate:
page_size = self.per_page
if page_size:
page = self.paginate_queryset(queryset, page_size)
else:
page = self.paginate_queryset(queryset, DEFAULT_PAGE_SIZE)
else:
page = queryset
context = {
'page_obj': page,
'page_header': self.page_header,
'page_option': self.page_option,
'table_title': self.table_title
}
context.update(kwargs)
return context
# 在myadmin/constants.py中添加
DEFAULT_PAGE_SIZE = 10
2.4>get_queryset
获取查询集,根据是否提供fields属性,来确定返回的数据。如果有搜索,过滤,可以复写该方法。
def get_queryset(self):
"""获取查询集,如需过滤,请复写此方法"""
if self.model is not None:
queryset = self.model._default_manager.all()
# 私有方法_default_manager可以获取到模型的管理器,
# 这里同样也可以直接写成objects,但可扩展性相对较低
else:
raise ImproperlyConfigured(
"%(cls)s 是不是傻,你没有指定model,我怎么知道呢?" % {
'cls': self.__class__.__name__
}
)
if self.fields:
queryset = queryset.only(*self.fields)
return queryset
2.5>paginate_queryset
分页
def paginate_queryset(self, queryset, page_size):
"""如果需要进行分页"""
paginator = Paginator(queryset, page_size)
try:
# 尝试获取页码,获取不到默认为1
page_number = int(self.request.GET.get('page', 1))
except Exception as e:
# 若出现其他错误,也使其变为1
page_number = 1
page = paginator.get_page(page_number)
return page
2.6>应用在标签列表
上面我们已经定义好了模板, 接下来我们就来使用它
# 在myadmin/views.py中添加以下代码
class TagsListView(MyPermissionRequiredMixin, MyListView):
permission_required = ('myadmin.news_tags_list', )
model = Tags
page_header = '系统设置'
page_option = '新闻标签管理'
table_title = '新闻标签列表'
fields = ['name', 'is_delete']
# 在myadmin/urls.py中添加以下代码
path('tags/', views.TagsListView.as_view(), name='tags_list'),
这样一个列表页面的后端部分就搞定了
3>模板
3.1>obj_list
接下来我们看模板, 在templates/myadmin/base
中创建一个obj_list.html
, 然后在模板中将需要自定义的地方挖坑obj_list.html
<!-- templates/myadmin/base/obj_list.html -->
{% load static %}
{% load news_template_filters %}
<section class="content-header">
<h1>
{{ page_header }}
<small>{{ page_option }}</small>
</h1>
</section>
<!-- Main content -->
<section class="content container-fluid">
<div class="box">
<div class="box-header with-border">
<h3 class="box-title">{{ table_title }}</h3>
<div class="box-tools">
{% block add_button %}
<!-- 添加按钮 -->
{% endblock %}
</div>
</div>
<!-- /.box-header -->
<div class="box-body">
{% block search_form %}
<!-- 搜索功能表单 -->
{% endblock %}
<table class="table table-bordered">
<tbody>
{% block table_content %}
<!-- 表头和表格内容 -->
{% endblock %}
</tbody>
</table>
</div>
<!-- 分页 -->
{% if page_obj.paginator %}
<div class="box-footer clearfix">
<div class="row">
<div class="col-sm-6">
<div class="dataTables_info" id="example2_info" role="status" aria-live="polite">
总共:{{ page_obj.paginator.count }}条 第{{ page_obj.start_index }}到{{ page_obj.end_index }}条
</div>
</div>
<div class="col-sm-6">
<ul class="pagination pagination-sm no-margin pull-right">
<li {% if not page_obj.has_previous %}class="disabled"{% endif %}
data-page="{{ page_obj.number|add:-1 }}"><a href="#">«</a></li>
{% for n in page_obj|page_bar %}
<li {% if n == page_obj.number %}class="active" {% endif %} data-page="{{ n }}"><a
href="#">{{ n }}</a></li>
{% endfor %}
<li {% if not page_obj.has_next %}class="disabled"{% endif %}
data-page="{{ page_obj.number|add:1 }}"><a href="#">»</a></li>
</ul>
</div>
</div>
</div>
{% endif %}
</div>
</section>
<!-- /.content -->
{% block script %}
{% endblock %}
4>自定义过滤器page_bar
上面obj_list.html
中引用了一个自定义过滤器news_template_filters
, 这个是在之前的代码中定义过的, 作用是根据分页页面对象生成分页页码, 代码:
from django import template
register = template.Library()
@register.filter
def page_bar(page):
page_list = []
if page.number != 1:
page_list.append(1)
if page.number - 3 > 1:
page_list.append('...')
if page.number - 2 > 1:
page_list.append(page.number - 2)
if page.number - 1 > 1:
page_list.append(page.number - 1)
page_list.append(page.number)
if page.paginator.num_pages > page.number + 1:
page_list.append(page.number + 1)
if page.paginator.num_pages > page.number + 2:
page_list.append(page.number + 2)
if page.paginator.num_pages > page.number + 3:
page_list.append('...')
if page.paginator.num_pages != page.number:
page_list.append(page.paginator.num_pages)
return page_list
5>应用在标签模板
在templates/myadmin
中创建一个tags
文件夹, 然后创建tags_list.html
<!-- templates/myadmin/tags/tags_list.html -->
{% extends 'myadmin/base/obj_list.html' %}
{% block table_content %}
<tr>
<th>#</th>
<th>标签名</th>
<th>是否可用</th>
</tr>
{% for obj in page_obj %}
<tr>
<td style="width: 40px" data-url=""><a href="#" data-id="{{ obj.id }}">{{ forloop.counter }}</a>
</td>
<td>{{ obj.name }}</td>
<td>{% if obj.is_active %}是{% else %}否{% endif %}</td>
</tr>
{% endfor %}
{% endblock %}
运行服务, 打开后台管理, 我们需要先在菜单管理中创建标签管理
, 才可以看到它
创建完成后的结果:
II. UpdateView
在列表视图中, 需要设计出每项的详情页面, 用于管理每项的信息.
这样我们就需要一个详情页的通用视图, 将其命名为UpdateView
1>通用的类属性
找出通用属性,定义在类属性中,子视图类通过覆盖这些属性来完成自定义。
class UpdateView(View):
model = None # 模型
form_class = None # 模型表单类
template_name = None # 模板名称
page_header = None # 页头大标题
page_option = None # 页头小标题
table_title = None # 内容标题
fields = None # 需要修改的字段
pk = None # url路径参数名,默认pk对象的pk字段
2>方法
2.1>get
get方法是用来响应get请求的,根据请求,获取模型对象,生成表单,渲染页面。
def get(self, request, **kwargs):
# 1. 获取对象
self.obj = self.get_obj(**kwargs)
# 2. 获取模板变量
context = self.get_context_data()
# 3. 返回渲染的模板
return render(request, self.get_template_name(), context=context)
2.2>get_obj
根据传入的对象主键获取需要修改的对象
def get_obj(self, **kwargs):
"""获取需要修改的对象"""
# 1. 拿到主键
self.get_obj_id(**kwargs)
# 2. 根据模型获取对象
if self.model is None:
raise ImproperlyConfigured('没有设置模型')
obj = self.model.objects.filter(pk=self.obj_id).first()
if not obj:
raise ObjectDoesNotExist('找不到pk=%s的对象')
# 3. 返回对象
return obj
2.3>get_obj_id
获取传入的对象主键
def get_obj_id(self, **kwargs):
"""获取传递的主键"""
if self.pk is None:
# 如果没有设置pk的名字, 那么就获取 变量名pk的值
self.obj_id = kwargs.get('pk')
else:
# 如果设置了pk的名字, 那么就获取 对象pk的值
self.obj_id = kwargs.get(self.pk)
2.4>get_context_data
获取视图的上下文变量,如果需要额外添加,可以复写次方法。
def get_context_data(self, **kwargs):
"""获取上下文变量"""
# 1. 获取表单类
self.form_class = self.get_form_class()
# 2. 创建表单对象
form = self.form_class(instance=self.obj)
# 3. 构造变量对象
context = {
'form': form,
'page_header': self.page_header,
'page_option': self.page_option,
'table_title': self.table_title
}
context.update(kwargs)
return context
2.5>get_form_class
获取表单类,如果没有提供,通过给定字段属性自动生成
def get_form_class(self):
"""获取表单类"""
if self.form_class is None:
# 如果没有提供,通过给定字段属性自动生成
if self.fields is None:
raise ImproperlyConfigured('未设置form和字段,无法生成表单')
# modelform_factory接收参数帮你创建一个表单类
return modelform_factory(self.model, fields=self.fields)
else:
return self.form_class
2.6>get_template_name
获取模板名称,如果不提供类属性template_name,会根据模型名称拼接出默认的模板myadmin/model_name/model_name_detail.html
def get_template_name(self):
"""获取模板名"""
if self.template_name is None:
# 如果未指定模板名, 则生成一个模板名
self.template_name = 'myadmin/{0}/{0}_detail.html'.format(self.model._meta.model_name)
return self.template_name
2.7>put
put方法是用来响应put请求的,根据请求,获取模型对象,获取参数,填充表单,校验,修改对象。
def put(self, request, **kwargs):
# 1. 获取对象
self.obj = self.get_obj(**kwargs)
# 2. 获取表单类
self.form_class = self.get_form_class()
# 获取模型对象
form = self.form_class(QueryDict(request.body), instance=self.obj)
if form.is_valid():
# 校验成功,保存表单数据
self.save(form)
return json_response(errmsg='修改数据成功!')
else:
# 校验失败,返回渲染错误的表单对象
context = self.get_context_data(form=form)
return render(request, self.get_template_name(), context=context)
2.8>save
保存对象,如果有额外操作,复写此方法。
def save(self, form):
if form.has_changed():
instance = form.save(commit=False)
instance.save(update_fields=form.changed_data)
3>前端代码
3.1>obj_detail
在templates/myadmin/base
中创建obj_detail.html
, 然后在模板中将需要自定义的地方挖坑
<!-- templates\myadmin\obj_detail.html -->
{% load admin_customer_tags %}
{% load admin_customer_filters %}
<section class="content-header">
<h1>
{{ page_header }}
<small>{{ page_option }}</small>
</h1>
</section>
<!-- Main content -->
<section class="content container-fluid">
<div class="box">
<div class="box-header with-border">
<h3 class="box-title">{{ table_title }}</h3>
</div>
<!-- /.box-header -->
<div class="box-body">
<form class="form-horizontal">
{% csrf_token %}
{% block form_content %}
{% for field in form %}
{% if field|is_checkbox %}
<div class="form-group">
<div class="col-sm-offset-1 col-sm-11">
<div class="checkbox">
<label for="{{ field.id_for_label }}">{{ field }}{{ field.label }}</label>
</div>
</div>
</div>
{% elif field|is_url_field %}
<div class="form-group {% if field.errors %}has-error{% endif %}">
<label for="{{ field.id_for_label }}"
class="col-sm-1 control-label">{{ field.label }}</label>
<div class="col-sm-11">
{% for error in field.errors %}
<label class="control-label"
for="{{ field.id_for_label }}">{{ error }}</label>
{% endfor %}
<div class="input-group">
{% add_class field 'form-control' %}
<span class="input-group-btn"><input class="hidden" type="file">
<button type="button" class="btn btn-info btn-flat">上传文件</button>
</span>
</div>
</div>
</div>
{% else %}
<div class="form-group {% if field.errors %}has-error{% endif %}">
<label for="{{ field.id_for_label }}"
class="col-sm-1 control-label">{{ field.label }}</label>
<div class="col-sm-11">
{% for error in field.errors %}
<label class="control-label"
for="{{ field.id_for_label }}">{{ error }}</label>
{% endfor %}
{% add_class field 'form-control' %}
</div>
</div>
{% endif %}
{% endfor %}
{% endblock %}
</form>
</div>
<div class="box-footer">
<button type="button" class="btn btn-default back">返回</button>
<button type="button" {% if form.instance.id %}
data-url="{% block update_url %}{% endblock %}"
data-type="PUT"
{% else %}
data-url="{% block add_url %}{% endblock %}"
data-type="POST"
{% endif %}
class="btn btn-primary pull-right save">保存
</button>
</div>
</div>
</section>
{% block script %}
<script>
$(() => {
{% block back_button %}
// 返回按钮
$('.box-footer button.back').click(() => {
let url = $('.sidebar-menu li.active a').data('url');
if (!url) {
return
}
$('#content').load(
url,
(response, status, xhr) => {
if (status !== 'success') {
message.showError('服务器超时,请重试!')
}
}
);
});
{% endblock %}
{% block save_button %}
$('.box-footer button.save').click(function () {
let url = $(this).data('url');
if (!url) {
return
}
$
.ajax({
url: url,
data: $('form').serialize(),
type: $(this).data('type')
})
.done((res) => {
if (res.errno === '0') {
message.showSuccess(res.errmsg);
$('#content').load(
$('.sidebar-menu li.active a').data('url'),
(response, status, xhr) => {
if (status !== 'success') {
message.showError('服务器超时,请重试!')
}
}
);
} else {
$('#content').html(res)
}
})
.fail((res) => {
message.showError('服务器超时,请重试!')
})
});
{% endblock %}
{% block upload %}
// 上传文件input
let $fileInput = $('.input-group-btn input');
let $uploadBtn = $('.input-group-btn button');
$uploadBtn.click(function () {
$(this).prev('input[type="file"]').click()
}
);
// 自动上传文件
$fileInput.change(function () {
$this = $(this);
if ($this.val() !== '') {
let formData = new FormData();
formData.append('upload', $this[0].files[0]);
formData.append('csrfmiddlewaretoken', $('input[name="csrfmiddlewaretoken"]').val());
$
.ajax({
url: '/admin/upload/',
// 使用ckeditor_uploader 就使用下面的url
// url: '/ckeditor/upload/&responseType=json',
type: 'POST',
data: formData,
processData: false,
contentType: false
})
.done((res) => {
if (res.uploaded === '1') {
message.showSuccess('封面图片上传成功!');
$this.parent().prev('input').val(res.url);
// 清空一下
$this.val('')
} else {
message.showError('封面图片上传失败!')
}
})
.fail(() => {
message.showError('服务器超时, 请重新尝试!')
})
}
});
{% endblock %}
});
</script>
{% endblock %}
<!-- /.content -->
4>自定义过滤器和自定义标签
模板中用到了以下几个自定义过滤器和自定义标签
4.1>is_checkbox
判断字段是否checkbox 类型,决定渲染方式
# apps/myadmin/templatetags/admin_customer_filters.py
from django.template import Library
from django.forms.widgets import CheckboxInput
register = Library()
@register.filter()
def is_checkbox(field):
return isinstance(field.field.widget, CheckboxInput)
4.2>is_url_field
判断字段是否url类型,决定渲染方式
# apps/myadmin/templatetags/admin_customer_filters.py
@register.filter()
def is_url_field(field):
return True if 'url' in field.label else False
4.3>add_class
给form字段添加class样式
# apps/myadmin/templatetags/admin_customer_tags.py
from django.template import Library
from django.shortcuts import reverse
register = Library()
@register.simple_tag()
def add_class(field, class_str):
return field.as_widget(attrs={'class': class_str})
5>应用到标签详情
创建tags/tags_detail.html
, 并写入以下代码:
{% extends 'myadmin/base/obj_detail.html' %}
{% block update_url %}
<!-- 详情页上, 保存按钮的链接 -->
{% url 'myadmin:tags_update' form.instance.id %}
{% endblock %}
页面上的效果:
III. AddView
在列表视图上做一个添加按钮, 可以点击它创建一个新的标签
接下来我们将其做成一个通用的视图模型
1>通用类属性
class AddView(View):
model = None
form_class = None
template_name = None
fields = None # 新建对象需要的字段
page_header = None # 页头大标题
page_option = None # 页头小标题
table_title = None # 表格标题
2>方法
2.1>get
渲染一个对于模型的表单页面返回。
def get(self, request):
context = self.get_context_data()
return render(request, self.get_template_name(), context=context)
2.2>post
根据post参数生成表单,校验,创建新对象。
def post(self, request):
self.form_class = self.get_form_class()
form = self.form_class(request.POST)
if form.is_valid():
self.save(form)
return json_response(errmsg='添加数据成功!')
else:
context = self.get_context_data(form=form)
return render(request, self.get_template_name(), context=context)
2.3>get_context_data
获取视图的上下文变量,如果需要额外添加,可以复写次方法。
def get_context_data(self, **kwargs):
self.form_class = self.get_form_class()
form = self.form_class()
context = {
'form': form,
'page_header': self.page_header,
'page_option': self.page_option,
'table_title': self.table_title
}
context.update(kwargs)
return context
2.4>get_template_name
获取模板名称,如果不提供类属性template_name,会根据模型名称拼接出默认的模板`myadmin/model_name/model_name_detail.html
def get_template_name(self):
"""获取模板名"""
if self.template_name is None:
self.template_name = 'myadmin/{0}/{0}_detail.html'.format(self.model._meta.model_name)
return self.template_name
2.5>get_form_class
获取表单类,如果没有提供,通过给定字段属性自动生成
def get_form_class(self):
if self.form_class is None:
if self.fields is None:
raise ImproperlyConfigured('你有不设置form,又不设置fields字段,那怎么生成表单呢?')
return modelform_factory(self.model, fields=self.fields)
else:
return self.form_class
2.6>save
保存对象,如果有额外操作,复写此方法。
def save(self, form):
form.save()
3>模板
AddView的页面模板与UpdateView一致, 除此之外, 还需要在tags_list.html
中填坑添加按钮
{% block add_button %}
<button type="button" class="btn btn-primary btn-sm"
data-url="{% url 'myadmin:tags_add' %}">添加
</button>
{% endblock %}
4>应用到添加标签
添加视图类:
class TagsAddView(MyPermissionRequiredMixin, AddView):
permission_required = ('myadmin.news_tags_add',)
model = Tags
page_header = '系统设置'
page_option = '新闻标签管理'
table_title = '添加新闻标签'
fields = ['name', 'is_delete']
添加url路径:
path('add_tag/', views.TagsAddView.as_view(), name='tags_add'),
现在已经可以访问添加页面了, 但是还不能添加标签, 需要以下配置
//在tags/tags_detail.html中写入以下代码
{% block add_url %}
{% url 'myadmin:news_tag_add' %}
{% endblock %}
IIII.删除功能
1>添加删除按钮
如果想在某个模型中添加删除功能, 需要在它的list模板中添加删除按钮, 例如此处在tags_list.html
中添加
<!-- 添加到指定的tr标签中 -->
<td style="width: 100px">
<button type="button" class="btn btn-danger btn-xs delete"
data-url="{% url 'myadmin:tags_update' obj.id %}">删除</button>
</td>
2>编写通用视图
class UpdateView(View):
...
def delete(self, request, **kwargs):
# 1. 获取对象
self.obj = self.get_obj(**kwargs)
# 2. 判断对象是否存在
if self.obj:
# 2.1 存在则删除并提示成功
self.obj.delete()
return json_response(errmsg='删除%s成功' % self.obj.name)
else:
# 2.2 否则报错
return json_response(errno=Code.NODATA, errmsg='%s不存在!' % self.obj.name)
3>编写通用js
在obj_list.html
中添加以下代码即可实现删除功能
{% block script %}
<script>
$(() => {
...
{% block delete %}
$('button.delete').click(function () {
let $this = $(this);
let url = $this.data('url');
if (!url) {
return
}
$
.ajax({
url: url,
type: 'DELETE',
dataType: 'json'
})
.done((res) => {
if (res.errno === '0') {
// 删除表格中的一行元素
$this.parent().parent().remove();
message.showSuccess(res.errmsg)
} else if (res.errno === '4105') {
message.showError(res.errmsg)
} else if (res.errno === '4101') {
message.showError(res.errmsg);
setTimeout(() => {
window.location.href = res.data.url
}, 1500)
} else {
message.showError(res.errmsg)
}
})
.fail(() => {
message.showError('服务器超时,请重试!')
})
});
{% endblock %}
});
</script>
{% endblock %}
V. 继续抽象
仔细观察上面的三个视图,发现有很多共同的方法,还需要将这些代码进一步的抽离,然后通过继承来达到减少冗余和解耦的作用。
1>TemplateView
抽取公用的代码;
class TemplateView(View):
"""
此类视图仅用于继承, 不单独使用
"""
# 这些属性是那三个视图中公用的属性
model = None # 模型
template_name = None
page_header = None # 页头大标题
page_option = None # 页头小标题
table_title = None # 内容标题
fields = None # 需要展示的字段
def get(self, request):
context = self.get_context_data()
return render(request, self.get_template_name(), context=context)
def get_context_data(self, **kwargs):
context = {
'page_header': self.page_header,
'page_option': self.page_option,
'table_title': self.table_title
}
context.update(kwargs)
return context
def get_template_name(self):
"""获取模板名"""
if self.template_name is None:
if isinstance(self, MyListView):
# 如果self代表MyListView,就返回list模板
self.template_name = 'myadmin/{0}/{0}_list.html'.format(self.model._meta.model_name)
else:
# 否则返回detail模板
self.template_name = 'myadmin/{0}/{0}_detail.html'.format(self.model._meta.model_name)
return self.template_name
2>DetailView
class DetailView(View):
"""
此类视图仅用于继承, 不单独使用
"""
form_class = None
def get_form_class(self):
if self.form_class is None:
if self.fields is None:
raise ImproperlyConfigured('你有不设置form,又不设置fields字段,那怎么生成表单呢?')
return modelform_factory(self.model, fields=self.fields)
else:
return self.form_class
def save(self, form):
form.save()
3>MyListView
class MyListView(TemplateView):
is_paginate = False # 是否分页
per_page = None # 每页条数
def get_queryset(self):
"""获取查询集,如需过滤,请复写此方法"""
if self.model is not None:
queryset = self.model._default_manager.all()
else:
raise ImproperlyConfigured(
"%(cls)s 是不是傻,你没有指定model,我怎么知道呢?" % {
'cls': self.__class__.__name__
}
)
if self.fields:
queryset = queryset.only(*self.fields)
return queryset
def get_context_data(self, **kwargs):
"""获取视图的上下文变量,如要添加额外变量,请继承此方法"""
queryset = self.get_queryset()
if self.is_paginate:
page_size = self.per_page
if page_size:
page = self.paginate_queryset(queryset, page_size)
else:
page = self.paginate_queryset(queryset, 10)
else:
page = queryset
return super().get_context_data(page_obj=page)
def paginate_queryset(self, queryset, page_size):
"""如果需要进行分页"""
paginator = Paginator(queryset, page_size)
try:
page_number = int(self.request.GET.get('page', 1))
except Exception as e:
page_number = 1
page = paginator.get_page(page_number)
return page
4>UpdateView
class UpdateView(TemplateView, DetailView):
pk = None # url路径参数名,默认pk 对象的pk字段
def get(self, request, **kwargs):
self.obj = self.get_obj(**kwargs)
return super().get(request)
def put(self, request, **kwargs):
self.obj = self.get_obj(**kwargs)
self.form_class = self.get_form_class()
form = self.form_class(QueryDict(request.body), instance=self.obj)
if form.is_valid():
self.save(form)
return json_response(errmsg='修改数据成功!')
else:
context = self.get_context_data(form=form)
return render(request, self.get_template_name(), context=context)
def get_obj_pk(self, **kwargs):
if self.pk is None:
self.obj_id = kwargs.get('pk')
else:
self.obj_id = kwargs.get(self.pk)
def get_obj(self, **kwargs):
self.get_obj_pk(**kwargs)
if self.model is None:
raise ImproperlyConfigured('没有设置模型')
obj = self.model.objects.filter(pk=self.obj_id).first()
if not obj:
raise ObjectDoesNotExist('找不到pk=%s的对象' % self.obj_id)
return self.model.objects.filter(pk=self.obj_id).first()
def get_context_data(self, **kwargs):
self.form_class = self.get_form_class()
form = self.form_class(instance=self.obj)
context = super().get_context_data(form=form)
context.update(kwargs)
return context
def save(self, form):
if form.has_changed():
instance = form.save(commit=False)
instance.save(update_fields=form.changed_data)
5>AddView
class AddView(TemplateView, DetailView):
def post(self, request):
self.form_class = self.get_form_class()
form = self.form_class(request.POST)
if form.is_valid():
self.save(form)
return json_response(errmsg='添加数据成功!')
else:
context = self.get_context_data(form=form)
return render(request, self.get_template_name(), context=context)
def get_context_data(self, **kwargs):
self.form_class = self.get_form_class()
form = self.form_class()
context = super().get_context_data(form=form)
context.update(kwargs)
return context