Django-xadmin+django-import-export导入导出的实现

醉酒当歌 提交于 2021-02-10 06:03:31

转载自:https://blog.csdn.net/zcyuefan/article/details/77748850

1. 需求vs现状
1.1 需求
要求做一个ERP后台辅助管理的程序,有以下几项基本要求:
1. 基本的增删改查功能
2. 基于对象的权限控制(如:系统用户分为平台运营人员和商家用户,商家用户小A只能查看编辑所属商家记录,而管理员可以纵览全局)
3. 数据库记录导入导出(xsl, json等),并且拥有对象级的权限控制(如:小A不能导出小B公司的信息,更不能导入小B公司信息进行更新和新增)

1.2 现状
实现需求1:Django-admin让我们能够很方便的实现一个管理后台程序。django-xadmin则在拥有admin基本功能的基础上增加了更为丰富的功能、界面也更加漂亮。类似还有django-suit等,本文使用xadmin(功能更丰富);
实现需求2:django-admin,以及xadmin都只有基于model级的权限控制机制,需要自己扩展或者使用开源解决方案,如django-guardian,django-rules,本文结合django-rules实现了该功能;
实现需求3:xadmin虽然自带导出功能,但是导入功能没有实现,django自带后台结合django-import-export可以很容易实现,但是xadmin并不直接兼容,只有通过xadmin的插件机制实现。
2. 功能实现
本节主要展示需求3导入导出功能实现。

2.1 安装并配置django-import-export
django-import-export主要原理是为待导入导出的model类设置一个Resource类,通过Resource类的export()方法进行导出,import_data()方法进行导入,灵活设置Resource便能实现多种文件格式数据导入导出。
pip安装:

$ pip install django-import-export
1
配置settings.py

# settings.py
INSTALLED_APPS = (
...
'import_export',
)

2.2 建立model(同上篇)
新增CompanyUser模型表示商家账户(即对django自带user模块进行扩展,使每个账号绑定自己的公司码),新增Customer模型表示商家的客户信息并包含公司码字段,商家账号只能查看、编辑、导入、导出公司码一致的商家客户信息

# model.py
class CompanyUser(models.Model):
user = models.OneToOneField(User, verbose_name='用户名')
is_taixiang_admin = models.BooleanField('是否运营人员', default=False)
company_code = models.CharField('公司码', max_length=20, blank=True, default='')

def __unicode__(self):
return '%s' % self.user

class Meta:
verbose_name = '导入账号'
verbose_name_plural = verbose_name

class Customer(models.Model):
name = models.CharField('客户姓名', max_length=50)
phone = models.CharField('客户电话', max_length=12)
type_choice = ((1, '普通'), (2, '批发'), (3, 'VIP'))
creator = models.ForeignKey(settings.AUTH_USER_MODEL, verbose_name='创建人', blank=True, null=True)
company_code = models.CharField('公司码', max_length=20, blank=True, null=True)

def __unicode__(self):
return '%s-%s-%s' % (self.company_code, self.name, self.phone1)

class Meta:
permissions = (
("simulate_import_customer", "允许模拟导入客户"),
("import_customer", "允许导入客户至商家系统"),
)
verbose_name = "客户"
verbose_name_plural = verbose_name

2.3 创建Resource
在model统计目录新增resources.py,配置Resource类

# resources.py
from import_export import resources
from .models import Customer

class CustomerResource(resources.ModelResource):

class Meta:
model = Customer
fields = ('id', 'name', 'phone1', 'c_type', 'company_code')

2.4 原生django-admin导入导出实现
使用原生的django-admin,ModelAdminClass可以通过继承ImportExportModelAdmin类实现导入导出,继承ImportExportActionModelAdmin类实现导出action:

# admin.py
from django.contrib import admin
from .models import Customer
from import_export.admin import ImportExportModelAdmin, ImportExportActionModelAdmin
from .resources import CustomerResource


class CustomerAdmin(ImportExportActionModelAdmin, ImportExportModelAdmin):
resource_class = CustomerResource

admin.site.register(Customer, CustomerAdmin)

2.5 django-xadmin导入导出实现(代码修改挺多的,此处展示基本思路,希望能push到xadmin项目中)
基于xadmin强大的plugin功能(参考Xadmin插件制作),这次我们开发一个importexport插件。具体实现转下章。

3. xadmin-importexport-plugin开发
开发思路:

类似admin使用importexport,需要在adminx.py ModelAdmin类中设置import_export_args属性用于关联Resource类,在list页面中会显示导入和导出按钮(ImportMenuPlugin,ExportMenuPlugin生效)。
# adminx.py
class CustomerAdmin(object):

import_export_args = {'import_resource_class': CustomerResource, 'export_resource_class': CustomerResource}
...

2. 点击导出按钮,会弹出若干选项,点击确定发送get请求后,触发ExportPlugin插件调用resource的export()方法进行导出。

3. 点击导入按钮,则会跳转至导入页面(ImportView),上传文件后点击导入会进行基本的检测

4. 点击确认导入(ImportProcessView)调用django-import-export导入至数据库。


3.1.1 插件部分代码
# xadmin/plugins/importexport.py

# 继承BaseAdminPlugin类
class ImportMenuPlugin(BaseAdminPlugin):
# 使用插件时需要在ModelAdmin类中设置import_export_args属性,插件初始化时使用ModelAdmin的import_export_args进行赋值
import_export_args = {}

# 返回True则加载插件,在list列表中显示导入按钮
def init_request(self, *args, **kwargs):
return bool(self.import_export_args.get('import_resource_class'))

def block_top_toolbar(self, context, nodes):
has_change_perm = self.has_model_perm(self.model, 'change')
has_add_perm = self.has_model_perm(self.model, 'add')
if has_change_perm and has_add_perm:
model_info = (self.opts.app_label, self.opts.model_name)
import_url = reverse('xadmin:%s_%s_import' % model_info, current_app=self.admin_site.name)
context = get_context_dict(context or {}) # no error!
context.update({
'import_url': import_url,
})
nodes.append(loader.render_to_string('xadmin/blocks/model_list.top_toolbar.importexport.import.html',
context=context))


class ExportMenuPlugin(BaseAdminPlugin):
import_export_args = {}

# Media
def get_media(self, media):
return media + self.vendor('xadmin.plugin.importexport.css', 'xadmin.plugin.importexport.js')

def init_request(self, *args, **kwargs):
return bool(self.import_export_args.get('export_resource_class'))

def block_top_toolbar(self, context, nodes):
formats = self.get_export_formats()
form = ExportForm(formats)

context = get_context_dict(context or {}) # no error!
context.update({
'form': form,
'opts': self.opts,
'form_params': self.admin_view.get_form_params({'_action_': 'export'}),
})
nodes.append(loader.render_to_string('xadmin/blocks/model_list.top_toolbar.importexport.export.html',
context=context))


site.register_plugin(ImportMenuPlugin, ListAdminView)
site.register_plugin(ExportMenuPlugin, ListAdminView)
---------------------
作者:zcyuefan
来源:CSDN
原文:https://blog.csdn.net/zcyuefan/article/details/77748850
版权声明:本文为博主原创文章,转载请附上博文链接!

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!