利用Python做自动化表报时遇到以下情况:
根据需求从数据库中输出数据写到Excel中,并对输出数据进行mapping,如果有缺失数据导致mapping不上的情况,在缺失的那一行自动空行,接着在空行后写入。
如A和B的数据均为mysql 输出数据,如果B缺失日期,则A对应的B为空,在下一行接着写入数据
A | B | |||
data_date | total_count | data_date | total_count | data_date_mapping |
2018-07-02 | 4468 | 2018-07-02 | 4458 | 相同 |
2018-07-09 | 4561 | 2018-07-09 | 4549 | 相同 |
2018-07-16 | 4530 | 2018-07-16 | 4517 | 相同 |
2018-07-23 | 4497 | 2018-07-23 | 4487 | 相同 |
2018-07-30 | 4416 | 2018-07-30 | 4407 | 相同 |
2018-08-06 | 4439 | 2018-08-06 | 4432 | 相同 |
2018-08-13 | 4494 | 2018-08-13 | 4487 | 相同 |
2018-08-27 | 4603 | 2018-08-27 | 4596 | 相同 |
2018-09-03 | 4465 | 2018-09-03 | 4457 | 相同 |
思路如下:
1.以A的数据为标准,B的数据为变量
2.因元组不能添加元素,因此先把元组转换为list
3.判断,如果B的时间在A中不存在,就插入一个元素,list.insert()
代码如下:
#! /usr/bin/env python
# -*- coding:utf-8 -*-
import MySQLdb
from openpyxl import load_workbook
import sys
import time
import os
reload(sys)
sys.setdefaultencoding('utf8')
db = MySQLdb.connect(host="localhost", user="zimu", passwd="zimu", db="database", port=3306, charset='utf8')
template_file = r"D:\tmepalte_path\template.xlsx"
output_path = r"D:\output_path\\"
if not os.path.exists(output_path):
os.mkdir(output_path)
book = load_workbook(template_file)
sheet = book['Sheet1']
B_data_list = []
A_sql = '''select '''
B_sql = '''select'''
# 处理A数据
cursor = db.cursor()
cursor.execute(A_sql)
A_data = cursor.fetchall()
t=0
while t<len(A_data):
for x in range (0,2):
sheet.cell(t + 3, x + 1).value = A_data[t][x]
t += 1
# 处理B 数据
cursor.execute(B_sql)
B_data = cursor.fetchall()
B_data_list = list(B_data)
t=0
while t<len(B_data_list):
if A_data[t][0] != B_data_list[t][0]:
B_data_list.insert(t,('',''))
for x in range(0, 2):
sheet.cell(t + 3, x + 3).value = B_data_list[t][x]
t += 1
#处理mapping
for i in range(3,sheet.max_row+1):
if sheet.cell(i,1).value == sheet.cell(i,3).value:
sheet.cell(i, 5).value = '相同'
else:
sheet.cell(i, 5).value = '不同'
book.save(output_path + "tmfs inv mapping_" + time.strftime("%Y%m%d", time.localtime())+ "_" + str(int(time.time())) +".xlsx")