import pandas as pd
import xlwings as xw

def insertRow(zgfj,deFile,b3df,b3ws):
  '''
  '''
  print('正在匯總:' + zgfj)
  dedf = pd.read_excel(deFile,header=3)
  dedf[['項目名稱','主管部門']] = dedf[['項目名稱','主管部門']].fillna(axis=0,method='ffill')
  dewb = app.books.open(deFile)
  dews = dewb.sheets[0]
  #對比兩個表的不同
  df1 = pd.pivot_table(dedf[dedf['主管部門']==zgfj],values='責(zé)任單位',index='項目名稱',aggfunc=[len])
  df2 = pd.pivot_table(b3df[b3df['主管部門']==zgfj],values='責(zé)任單位',index='項目名稱',aggfunc=[len])
  df12 = df1 - df2
  if df12.shape[0]:
    diff = df12[df12[('len','責(zé)任單位')] != 0]
  #兩個表不同則修改excel表，使相同項目的行數(shù)相同
  #if diff.shape[0]:
    for xmmc in list(diff.index):
      for r in range(1,b3ws.used_range.shape[0]+1):                  
        if b3ws.range(r,2).value == xmmc:
          if diff.loc[xmmc][0] > 0:
            print(str(r+1) + ':' + str(int(r+diff.loc[xmmc][0])))
            b3ws.api.rows(str(r+1) + ':' + str(int(r+diff.loc[xmmc][0]))).insert #插入部門多出的行
            break
          else:
            b3ws.api.rows(str(r+1) + ':' + str(int(r-diff.loc[xmmc][0]))).delete #刪除多余的行
            break
  #復(fù)制部門內(nèi)容到匯總表
  for xmmc in list(df2.index):
    for r in range(1,b3ws.used_range.shape[0]+1):                  
        if b3ws.range(r,2).value == xmmc:
          #項目名稱在部門excel表的行號
          rfj = dedf[dedf['項目名稱'] == xmmc].index[0] + 5
          #需要插入的行數(shù)
          rows = df1.loc[xmmc][0]
          #復(fù)制部門excel表格項目名稱所在行到匯總表
          dews.api.rows(str(rfj) + ':' + str(rfj+rows-1)).Copy(b3ws.api.rows(str(r) + ':' + str(r+rows-1)))
          break #因為項目名稱唯一，復(fù)制后可跳出進行下一項目
if __name__ == '__main__':
  #匯總表格文件
  b3File = '匯總文件.xls'
  #各部門表格文件所在位置
  fjFile = {'B部門':'B部門.xls',\
       'A部門':'A部門.xls',\
       'C部門':'C部門.xls',\
       'D部門':'D部門.xls'}
  app = xw.App(visible=False,add_book=False)
  app.display_alerts = False
  app.screen_updating = False
  b3wb = app.books.open(b3File)
  b3ws = b3wb.sheets[0]
  b3df = pd.read_excel(b3File,header=3)
  for zgfj,file in fjFile.items():
    b3df[['項目名稱','主管部門']] = b3df[['項目名稱','主管部門']].fillna(axis=0,method='ffill') #填充合并單元格內(nèi)容
    #print(b3df.shape[0])
    insertRow(zgfj,file,b3df,b3ws)
    
  b3wb.save('匯總后文件.xls')
  app.quit()
  app.kill()