最近项目重构,海外使用api接口的一些数据需要在本地库也进行存储,所以写了个脚本获取api接口的数据并导出为excel表格供信息部同事进行录入工作.
项目使用django框架,需要用到项目中的一些表格来获取特定几组使用海外api接口的对象,所以模块首先导入了django配置并初始化
import os
import django
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "your_project.settings")
django.setup()
然后就是愉快的获取数据过程了,
class APIServiceList(object):
"""
获取海外接口信息
"""
def __init__(self, begin_date, end_date):
self.begin_date = begin_date
self.end_date = end_date
def get_msg_list(self):
"""
此处省略代码段
"""
最后获取到的数据是列表格式的,需要对应输出到excel表格中,且存在一个子项中对应多个子项的情况,所以可能存在合并单元格或单元格空置的需求,接下来就是将列表中数据整合到excel工作流之中了.
# 设置HTTPResponse的类型
response = HttpResponse(content_type='application/vnd.ms-excel')
response['Content-Disposition'] = 'attachment;filename=order.xls'
# 创建一个文件对象
wb = xlwt.Workbook(encoding='utf8')
# 创建一个sheet对象,如果需要可以创建多个工作簿
sheet = wb.add_sheet('apollo-sheet', cell_overwrite_ok=True)