import os
import sys
from datetime import date
import glob
import shutil
import csv
import pandas as pd

rootdir = "/Users/normrasmussen/Documents/Resources/Walmart/"
downloadir = "/Users/normrasmussen/Google Drive/My Drive/Shared with Clients/Walmart/"
basefile = "WalmartTemplate.xlsx"

def copytemplate(rootdir, basefile):
    today = date.today()
    today = today.strftime("%m.%d.%Y")
    template = rootdir+basefile
    todayFile = f"Walmart-{today}.xlsx"
    if os.path.exists(rootdir):
        shutil.copy2(template, rootdir+todayFile)
        #dirfiles = os.listdir(rootdir)
        #print(dirfiles)
        currentDash = rootdir+f"Walmart-{today}.xlsx"
        print(currentDash)
        findlatestExport(currentDash)

def findlatestExport(currentDash):
    listfiles = glob.glob(downloadir + '*.csv')
    latestdownload = max(listfiles, key=os.path.getctime)
    print(latestdownload)
    copytoDash(latestdownload, currentDash)

def copytoDash(latestdownload, currentDash):
    readExport = pd.read_csv(
            latestdownload, 
            index_col=False,
            header=0,
            low_memory=False,
            dtype={
                'Progress':object
        }
            )
    print(readExport)
    readExport.drop(
            readExport.filter(
                regex="Unname"
                ),axis=1, 
                inplace=True)
    copiedData = readExport.copy()
    bringtoExcel(latestdownload, currentDash, copiedData)

def bringtoExcel(latestdownload, currentDash, copiedData):
    with pd.ExcelWriter(
            currentDash,
            mode="a",
            engine="openpyxl",
            if_sheet_exists="overlay",
    ) as writer:
        copiedData.to_excel(
                writer,
                engine="xlsxwriter",
                sheet_name="Data",
                index=False,
    )

def cleanitUp(currentDash):
    cleanExcel = pd.read_excel(currentDash, sheet_name="Data", index_col=None)
    cleanExcel.columns.values[0] = "tmp"
    cleanExcel.drop(columns="tmp", axis=1, inplace=True)
    print(cleanExcel)

if __name__ == "__main__":
    copytemplate(rootdir, basefile)