import pandas as pd 
import xlwings as xw
import pymssql


# 各品類月同期 
def get_last_year_sale(start_date, end_date):
  """各品類同期銷量, 對比19年"""
  sql_01 = f"""
  SELECT 
   品類
   , SUM(數(shù)量) AS QTY
  FROM XXX
  WHERE 是否電商 = 1 
   AND 銷售時間 BETWEEN DATEADD(YEAR, -2, '{start_date}') AND DATEADD(YEAR, -2, '{end_date}')   
  GROUP BY 品類
  """
  df = pd.read_sql(sql_01, con=con)
  df_xtc = df[df['品類'] == 'A品類'][['品類', 'QTY']]
  df_bbk = df[df['品類'] == 'B品類'][['品類', 'QTY']]
  return df_xtc, df_bbk 
  
def get_anget_sale(start_date, end_date):
    """返回各品類, 各區(qū)域的時間段銷量"""
    sql = f"""
    SELECT 
     品類
     , AGENT
     , SUM(數(shù)量) AS QTY
     , ROW_NUMBER()OVER(PARTITION BY 品類 ORDER BY SUM(數(shù)量) DESC) MY_RANK
    FROM XXX
    WHERE 是否電商 = 1 
     AND 銷售時間 BETWEEN '{start_date}' AND '{end_date}'
    GROUP BY AGENT, 品類
    """
    df = pd.read_sql(sql, con=con)
    df_xtc = df[df['品類'] == 'A品類'][['AGENT', 'QTY']]
    df_bbk = df[df['品類'] == 'B品類'][['AGENT', 'QTY']]
    df_pad = df[df['品類'] == 'C品類'][['AGENT', 'QTY']]

    return df_xtc, df_bbk, df_pad 
  
def get_machine_sale(start_date, end_date):
  """返回各品類, 各區(qū)域的時間段銷量"""
  sql = f"""
  SELECT 
   品類
   , 機型
   , SUM(數(shù)量) AS QTY
   , ROW_NUMBER()OVER(PARTITION BY 品類 ORDER BY SUM(數(shù)量) DESC) MY_RANK
  FROM V_REALSALE
  WHERE 是否電商 = 1 
   AND 銷售時間 BETWEEN '{start_date}' AND '{end_date}'
  GROUP BY 機型, 品類
  """
  df = pd.read_sql(sql, con=con)
  df_xtc = df[df['品類'] == 'A品類'][['機型', 'QTY']]
  df_bbk = df[df['品類'] == 'B品類'][['機型', 'QTY']]

  return df_xtc, df_bbk 


# main 
con = pymssql.connect('xxxxx', 'sxxx', 'xxxxxx', 'xxxxx')

# 基礎(chǔ)配置: 根據(jù)用戶輸入當(dāng)前日期, 輸出當(dāng)月, 當(dāng)季度第一天 
print("歡迎哦, 此小程序?qū)ｉT為XX看板做數(shù)據(jù)自動更新呢~")
print()

today = input("請輸入截止日期(昨天), 形如: 2021/5/20 按回車結(jié)束:  ")

if len(today.split('/')) != 3:
  raise "日期格式輸入錯誤!!, 請按照形如 '2021/5/20'的格式重新輸入"
else:
  m_cur = today.split('/')[1]
  m_first_day = '2021/' + m_cur + '/1'

# 季度第一天 
if m_cur in ('1', '01', '2', '02', '3', '03'):
  q_time_start = '2021/1/1'
  
elif m_cur in ('4', '04', '5', '05', '6', '06'):
  q_time_start = '2021/4/1'
  
elif m_cur in ('7', '07', '8', '08', '9', '09'):
  q_time_start = '2021/7/1'
else:
  q_time_start = '2021/10/1'

print()
print("正在開始更新....")
print("提示, 接下看到閃退, 是正常現(xiàn)象, 就程序模擬人去打開文件, 填充數(shù)據(jù), 不要緊張哦~~~")

# 去年月, 季度同期 
df_mm_xtc, df_mm_bbk = get_last_year_sale(m_first_day, today)
df_qq_xtc, df_qq_bbk = get_last_year_sale(q_time_start, today)

# 當(dāng)月各地區(qū)累積銷量
df_m_xtc, df_m_bbk, df_m_pad = get_anget_sale(m_first_day, today)

# 各地區(qū)當(dāng)季度銷量 
df_q_xtc, df_q_bbk, df_q_pad = get_anget_sale(q_time_start, today)

# 各機型當(dāng)季度銷量 
df_q_type_xtc, df_q_type_bbk = get_machine_sale(q_time_start, today) 
# 過濾掉 銷量為0的型號 
df_q_type_xtc = df_q_type_xtc[df_q_type_xtc.QTY > 0]
df_q_type_xtc.replace('Z6áÛ·å°æ', 'Z6巔峰版', inplace=True)

df_q_type_bbk = df_q_type_bbk[df_q_type_bbk.QTY > 0]

# 打開excel 模板 等待數(shù)據(jù)填充 
app = xw.App(visible=True, add_book=False)

app.display_alerts = False  # 關(guān)閉一些提示信息，可以加快運行速度。 默認(rèn)為 True。
app.screen_updating = True

wb = app.books.open("XXX_全品類_看板.xlsx")
data_sht = wb.sheets['數(shù)據(jù)']

# 19年當(dāng)月同期銷量
data_sht.range('B9').value = df_mm_xtc.values
data_sht.range('G9').value = df_mm_bbk.values

# 當(dāng)季度同比
data_sht.range('B10').value = df_qq_xtc.values
data_sht.range('G10').value = df_qq_bbk.values

# 填充各品類當(dāng)月銷量, 注意單元格是寫死的哦
data_sht.range('I72').value = df_m_xtc.values
data_sht.range('T72').value = df_m_bbk.values
data_sht.range('AE72').value = df_m_pad.values

# 填充當(dāng)季度銷量, 同理是寫死的
data_sht.range('A54').value = df_q_xtc.values
data_sht.range('F54').value = df_q_bbk.values
data_sht.range('K54').value = df_q_pad.values

# 填充當(dāng)季度各型號, 同理是寫死的
data_sht.range('A21').value = df_q_type_xtc.values
data_sht.range('F21').value = df_q_type_bbk.values

wb.save()
app.quit()

print()
print("~~更新結(jié)束了哦~~")
print()
input("請按任意鍵退出~~")
print()
print('BYE~~ 人生若只如初見呢~~')