import win32com.client
from pathlib import Path
from datetime import datetime, timedelta
import openpyxl

PricingVolumeFile = 'C:\\Users\\PricingVolumes2022.xlsx'

outlook = win32com.client.Dispatch('outlook.application')
mapi = outlook.GetNamespace("MAPI")
    
inbox = mapi.GetDefaultFolder(6)

inbox = mapi.GetDefaultFolder(6).Folders["Notifications"].Folders["Pricing"].Folders["Peak Future Volume"]

# THIS IS USED TO GRAB THE MESSAGE DATE SO IT CAN PROCESS ONLY THE LATEST EMAILS (EXAMPLE BELOW IS LAST 1 DAY)
received_dt = datetime.now() - timedelta(days=1)
received_dt = received_dt.strftime('%m/%d/%Y %H:%M %p')

edateFileSaveDate = datetime.now()
edateFileSaveDate = edateFileSaveDate.strftime('%Y%m%d')

messages = inbox.Items
messages = messages.Restrict("[ReceivedTime] >= '" + received_dt + "'")

for message in messages: 
    #     # THIS IS USED TO READ THE SUBJECT AND PARSE OUT THE DATE FOR THE APPROPRIATE MESSAGES
    sub = message.subject
    # TRY USING THIS ACTUAL MESSAGE DATE FOR THE IF CONDITION BELOW
    messD = message.ReceivedTime
    
    edDate = sub.split()[5]
    edreceived_dt = datetime.now()
    edreceived_dt = edreceived_dt.strftime('%m/%d/%Y %H:%M %p')
    tdDate = edreceived_dt.split()[0]      
    
    # CHECK IF MESSAGE IS TODAY SO THAT THE MOST CURRENT SCREENSHOT IS UPDATED IN THE FILE.
    if edDate == tdDate:
        attachments = message.Attachments
        # THIS ENSURES THAT THE EMAIL HAS AT LEAST 2 ATTACHMENTS BEFORE IT TRIES TO DOWNLOAD FILES THAT DONT EXIST    
        if attachments.count >= 1:
            
            #LOOP THRU ALL ATTACHMENTS AND ONLY DOWNLOAD CSV FILES
            for att in attachments:
               p = Path(str(att).lower())
               
               #newName = "{0}_{1}".format(p.stem,p.suffix)
               newName = edateFileSaveDate + "{0}".format(p.suffix)
               
               # CHECK IF ITS A PNG FILE BEFORE ATTEMPTING TO DOWNLOAD
               if p.suffix == ".png":
                  imgFileLoc = "C:\PricingVolume"+ '\\' + newName
                  # DOWNLOAD CSV FILE ONLY WITH THE NEW FILE NAME
                  figSaved  = att.SaveASFile(imgFileLoc)
                  #print(att)
                  # THIS IS THE MASTER EXCEL FILE THAT IS UPDATED AT THE END WITH THE FILTERED DATA
                  wrkb = openpyxl.Workbook()
                  ws = wrkb.worksheets[0]
                  img = openpyxl.drawing.image.Image(imgFileLoc)
                  img.height = 757
                  img.width= 1533
                  img.anchor = 'A1'
                  ws.add_image(img)
                  wrkb.save(PeakPricingVolumeFile)
    
    message.Unread = False
    message.Save()