kw_tools/joblist/jobcheck.py

import requests
import datetime
from dateutil import parser
import gspread
from oauth2client.service_account import ServiceAccountCredentials
import time
import smtplib
from userdata import *
from email.message import EmailMessage

# drop this at PUBLIC2

# TASK LIST HERE
# https://docs.google.com/spreadsheets/d/153w046qHhpnZTYU7FJ-S2xyGIH_BXbycLYfLI2fFsZc/edit#gid=0

scope = ['https://spreadsheets.google.com/feeds','https://www.googleapis.com/auth/drive']
creds = ServiceAccountCredentials.from_json_keyfile_name('choozmointernal-2e314f3d4e12.json', scope)
client = gspread.authorize(creds)
message = "提醒近期預定執行的作業如下：\n\n"
message2 = "提醒近期需完成的作業如下：\n\n"

def notify_group(msg):
    gid='WekCRfnAirSiSxALiD6gcm0B56EejsoK89zFbIaiZQD'
    headers = {"Authorization": "Bearer " + gid,"Content-Type": "application/x-www-form-urlencoded"}
    r = requests.post("https://notify-api.line.me/api/notify",headers=headers, params={"message": msg})

def send_line(messages):
    for message in messages:
        msg = [message[i:i+1000] for i in range(0, len(message), 1000)]
        for m in msg:
            notify_group(m)
            print(m)

def send_email(emails, content, has_overdue):
    server = smtplib.SMTP_SSL('smtp.gmail.com', 465)
    server.ehlo()
    server.login(gmail_username, gmail_password)

    for name in content:
        msg = EmailMessage()
        msg['Subject'] = '執行表單過期作業通知' + str(datetime.date.today())
        msg['From'] = "service@choozmo.com"
        msg['To'] = emails[name]

        if has_overdue[name] == 1:
            msg.set_content(content[name])
        else:
            msg.set_content(name + "您好，\n\n您目前沒有過期的作業。\n更新執行表單請至 https://cmm.ai/pm ，謝謝您。\n\n")
        
        print ('Sending email to ' + msg['To'])
        server.send_message(msg)
        print ('Email sent to ' + msg['To'])
        fname = name+'.txt'
        with open(fname, 'w', encoding="UTF+8") as f:
            if has_overdue[name] == 1:
                f.write(content[name])
            else:
                f.write(name + "您好，\n\n您目前沒有過期的作業。\n更新執行表單請至 https://cmm.ai/pm ，謝謝您。\n\n")

    server.close()

sheet = client.open('ChoozMo 專案執行表單_20220720')
sheet_instance = sheet.get_worksheet(0)

records_data = sheet_instance.get_all_records()
#print(records_data[0])

emails = {"Jared": "jared@choozmo.com", "Morrison": "morrison@choozmo.com", "Ginia": "ginia@choozmo.com", "Doris": "doris@choozmo.com", "Nina": "nina.huang@choozmo.com", "Yukyo": "yukyo@choozmo.com", "Andy": "andy@choozmo.com", "Oak": "oak@choozmo.com", "Zooey": "zooey@choozmo.com", "Syuanyuri": "syuanyuri@choozmo.com", "Wizer": "wizer@choozmo.com", "Jason": "jason@choozmo.com", "Tomoya": "tomoya@choozmo.com", "詠心": "noodlesloves@gmail.com", "献竤": "fxp87257@gmail.com"}
overdue_email_content = {}
has_overdue = {}

for e in emails.keys():
    overdue_email_content[e] = e + "您好，\n\n提醒您有已過期的作業，請儘快完成以下作業。\n更新執行表單請至 https://cmm.ai/pm ，謝謝您。\n\n過期作業如下：\n\n"
    has_overdue[e] = 0

for data in records_data:
    try:
        print(parser.parse(data["起始日"]), parser.parse(data["完成日"]))
        deltastart = (parser.parse(data["起始日"]).date()-datetime.date.today()).days
        deltaend = (parser.parse(data["完成日"]).date()-datetime.date.today()).days
        print(deltastart, deltaend)
        info = ""

        if(deltastart <= 7 and data["完成度"] == "預定執行"):
            info += "專案：" + data["專案"] + " / " + data["專案項目"] + "\n"
            info += "作業：" + data["執行項目"] + "\n"
            info += "執行人員：" + data["指派對象"] + "\n"
            info += "PM：" + data["PM"] + "\n"
            info += "預定執行期間：" + data["起始日"] + "-" + data["完成日"]
            if(deltastart>0):
                info += " (尚有"+ str(deltastart) + "天)\n\n"
            elif(deltastart==0):
                info += " (當天開始執行)\n\n"
            else:
                info += " (過期"+ str(deltastart).replace('-','') + "天)\n\n"
                overdue_email_content[data["PM"]] += info
                has_overdue[data["PM"]] = 1
                if [data["PM"]]!=[data["指派對象"]]: #sometimes you have to do the task and you are the PM
                    overdue_email_content[data["指派對象"]] += info
                    has_overdue[data["指派對象"]] = 1
                if data["PM"] != "Jared": #sometimes Jared is the PM
                    overdue_email_content["Jared"] += info
                    has_overdue["Jared"] = 1
            message += info


        elif(deltaend <= 7 and data["完成度"] == "執行中"):
            info += "專案：" + data["專案"] + " / " + data["專案項目"] + "\n"
            info += "作業：" + data["執行項目"] + "\n"
            info += "執行人員：" + data["指派對象"] + "\n"
            info += "PM：" + data["PM"] + "\n"
            info += "完成日：" + data["完成日"]
            if(deltaend>0):
                info += " (尚有"+ str(deltaend) + "天)\n\n"
            elif(deltaend==0):
                info += " (當天完成)\n\n"
            else:
                info += " (過期"+ str(deltaend).replace('-','') + "天)\n\n"
                overdue_email_content[data["PM"]] += info
                has_overdue[data["PM"]] = 1
                if [data["PM"]]!=[data["指派對象"]]: #sometimes you have to do the task and you are the PM
                    overdue_email_content[data["指派對象"]] += info
                    has_overdue[data["指派對象"]] = 1
                if data["PM"] != "Jared": #sometimes Jared is the PM
                    overdue_email_content["Jared"] += info
                    has_overdue["Jared"] = 1
            message2 += info

    except:
        print("PASS")

message += "更新執行表單請至 https://cmm.ai/pm"
message2 += "更新執行表單請至 https://cmm.ai/pm"

messages = []
messages.append(message)
messages.append(message2)

print(has_overdue)

send_line(messages)
send_email(emails, overdue_email_content, has_overdue)

'''notify_group(message)
time.sleep(5)
notify_group(message2)'''