WaitAndTweetSchedule.py

import os
import tweepy
import requests
from datetime import datetime, timedelta
from PIL import Image, ImageDraw, ImageFont
import AtProWrapper
import io
import time
import pusher
#secretsで設定した値をとる
CONSUMER_KEY = os.environ.get('CONSUMER_KEY', "")
CONSUMER_SECRET = os.environ.get('CONSUMER_SECRET', "")
ACCESS_TOKEN = os.environ.get('ACCESS_TOKEN', "")
ACCESS_SECRET = os.environ.get('ACCESS_SECRET', "")

#認証
auth = tweepy.OAuthHandler(CONSUMER_KEY, CONSUMER_SECRET)
auth.set_access_token(ACCESS_TOKEN, ACCESS_SECRET)
#オブジェクト作成
api = tweepy.API(auth)
client = tweepy.Client(
	consumer_key = CONSUMER_KEY,
	consumer_secret = CONSUMER_SECRET,
	access_token = ACCESS_TOKEN,
	access_token_secret = ACCESS_SECRET
)
transjson = requests.get("https://splatoon3.ink/data/locale/ja-JP.json").json()
def GetTranslation(transtype, textid):
    return transjson.get(transtype,{}).get(textid, {}).get("name","翻訳できませんでした。")
def GetTranslationRegulation(transtype, textid):
    return transjson.get(transtype,{}).get(textid, {}).get("regulation","翻訳できませんでした。")
def GetSchedulesData(jsonobj, mode):
    objs = []
    for sch in jsonobj["data"][mode + "Schedules"]["nodes"]:
        obj = {}
        obj["start"] = datetime.strptime(sch.get("startTime"), '%Y-%m-%dT%H:%M:%SZ')
        obj["end"] = datetime.strptime(sch.get("endTime"), '%Y-%m-%dT%H:%M:%SZ')
        obj["settings"] = []

        if mode == "bankara":
            settings = sch.get(mode+"MatchSettings", [])
        else:
            settings = [sch.get(mode+"MatchSetting", {})]
        for setting in settings:
            settingobj = {}
            settingobj["rule"] = setting.get("vsRule", {}).get("id", "")
            stages = []
            for stage in setting.get("vsStages", []):
                stageobj = {}
                stageobj["name"] = stage.get("id")
                stageobj["image"] = stage.get("image", {}).get("url", "")
                stages.append(stageobj)
            settingobj["stages"] = stages
            obj["settings"].append(settingobj)
        objs.append(obj)
    return objs
schjson = requests.get("https://splatoon3.ink/data/schedules.json").json()

def crop_center(pil_img, crop_width, crop_height):
    img_width, img_height = pil_img.size
    return pil_img.crop((
        (img_width - crop_width) // 2,
        (img_height - crop_height) // 2,
        (img_width + crop_width) // 2,
        (img_height + crop_height) // 2
        ))

def crop_max_square(pil_img):
    return crop_center(pil_img, min(pil_img.size), min(pil_img.size))

def make(pil_img, r=100, fil=True):
    mask = DrawBack(pil_img, r)
    if fil:
        mask = mask.filter(ImageFilter.SMOOTH)
    result = pil_img.copy()
    result.putalpha(mask)

    return result


def DrawBack(img, r=100):
    #角丸四角を描画する[3]を参考。というかほぼコピペ。
    mask = Image.new("L", img.size, 0)
    draw = ImageDraw.Draw(mask)
    rx = r
    ry = r
    fillcolor = "#ffffff"
    draw.rectangle((0,ry)+(mask.size[0]-1,mask.size[1]-1-ry), fill=fillcolor)
    draw.rectangle((rx,0)+(mask.size[0]-1-rx,mask.size[1]-1), fill=fillcolor)
    draw.pieslice((0,0)+(rx*2,ry*2), 180, 270, fill=fillcolor)
    draw.pieslice((0,mask.size[1]-1-ry*2)+(rx*2,mask.size[1]-1), 90, 180, fill=fillcolor)
    draw.pieslice((mask.size[0]-1-rx*2,mask.size[1]-1-ry*2)+
    (mask.size[0]-1,mask.size[1]-1), 0, 180, fill=fillcolor)
    draw.pieslice((mask.size[0]-1-rx*2,0)+
    (mask.size[0]-1,ry*2), 270, 360, fill=fillcolor)
    return mask
def GetSalmonData(jsonobj, mode):
    objs = []
    for sch in jsonobj["data"]["coopGroupingSchedule"][mode + "Schedules"]["nodes"]:
        obj = {}
        obj["start"] = datetime.strptime(sch.get("startTime"), '%Y-%m-%dT%H:%M:%SZ')
        obj["end"] = datetime.strptime(sch.get("endTime"), '%Y-%m-%dT%H:%M:%SZ')
        obj["stagename"] = sch.get("setting",{}).get("coopStage",{}).get("id","翻訳がありません")
        obj["stageimage"] = sch.get("setting",{}).get("coopStage",{}).get("image",{}).get("url","")
        obj["kingsalmonid"] = sch.get("__splatoon3ink_king_salmonid_guess")
        obj["weapons"] = []
        for weapon in sch.get("setting",{}).get("weapons",[]):
            obj["weapons"].append({
                "name":weapon.get("__splatoon3ink_id","翻訳がありませんでした"),
                "image":weapon.get("image",{}).get("url")
                })
        objs.append(obj)
    return objs
def CreateSalmonImage(jsons, text, text_x, IsBackPaste = True):
    back = Image.open("SplatoonBack_White.png").crop((0,0,960,625))
    img = Image.open("SplatoonBack_Black.png").crop((0,0,900,600))
    font = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 60)
    font_big = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 65)
    font_chu = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 45)
    font_mini = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 30)
    font_supermini = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 20)
    
    # 角丸にするためのマスクを作成
    mask = Image.new("L", img.size, 0)
    radius = 50
    draw = ImageDraw.Draw(mask)
    draw.rounded_rectangle((0, 0, img.width, img.height), radius, fill=255)

    draw = ImageDraw.Draw(img)
    
    draw.text((text_x, 0), text, font = font, fill = "#FFFFFF")

    draw.text((120, 350), GetTranslation("stages", jsons["stagename"]), font = font, fill = "#FFFFFF")
    stage = Image.open(io.BytesIO(requests.get(jsons["stageimage"]).content))
    stage = stage.resize((int(stage.size[0] / 1.7),
                          int(stage.size[1] / 1.7)))
    img.paste(stage, (50, 80))
    
    if datetime.utcnow() > jsons["start"]:
        starttext = ""
    else:
        starttext = (jsons["start"] + timedelta(hours=9)).strftime('%Y年%m月%d日%H時')+"開始"
    draw.text((550, 80), starttext, font = font_mini, fill = "#FFFFFF")

    #オカシラシャケ
    draw.text((545, 130), "オカシラシャケ", font = font_big, fill = "#FFFFFF")
    draw.text((615, 200), GetSalmonoidName(jsons["kingsalmonid"]), font = font_chu, fill = "#FFFFFF")
    
    #武器
    index = 0
    for weapon in jsons["weapons"]:
        weaponimg = Image.open(io.BytesIO(requests.get(weapon["image"]).content))
        weaponimg = weaponimg.resize((int(weaponimg.size[0] / 2),
                              int(weaponimg.size[1] / 2)))
        additional = 50
        img.paste(weaponimg, (40 + additional + (index * 190), 425), weaponimg)
        weapontext = weapon["name"]
        if weapontext is not "6e17fbe20efecca9":
            weapontext = GetTranslation("weapons", weapontext)
        else:
            weapontext = ""
        draw.text((50 + additional + (index * 190), 550), weapontext, font = font_supermini, fill = "#FFFFFF")
        index += 1
    back.paste(img, (30,10), mask=mask)
    return back
def GetSalmonoidName(salmonid):
    if salmonid == "Cohozuna":
        return "ヨコズナ"
    elif salmonid == "Horrorboros":
        return "タツ"
    elif salmonid == None:
        return "翻訳がありませんでした"
    else:
        return salmonid
def CreateSchImage(jsons, text, text_x, IsBackPaste = True):
    back = Image.open("SplatoonBack_White.png").crop((0,0,800,920))
    img = Image.open("SplatoonBack_Black.png").crop((0,0,600,900))
    #img = make(img, 10, False)
    #Image.new("RGBA", (600, 900), (107, 107, 107))
    
    rx = 10
    ry = 20
    fillcolor = "#FFFFFF"

    font = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 55)

    # 角丸にするためのマスクを作成
    mask = Image.new("L", img.size, 0)
    radius = 50
    draw = ImageDraw.Draw(mask)
    draw.rounded_rectangle((0, 0, img.width, img.height), radius, fill=255)

    draw = ImageDraw.Draw(img)
    draw.text((text_x, 0), text, font = font , fill = "#FFFFFF")
    
    page = Image.open(io.BytesIO(requests.get(jsons["stages"][0]["image"]).content))
    page = page.resize((int(page.width * 1.25), int(page.height * 1.25)))
    draw.text((290, 100), GetTranslation("rules", jsons["rule"]), font = font , fill = "#FFFFFF",anchor='mm')
    img.paste(page, (50, 150))
    draw.text((300, 440), GetTranslation("stages", jsons["stages"][0]["name"]), font = font , fill = "#FFFFFF",anchor='mm')

    page = Image.open(io.BytesIO(requests.get(jsons["stages"][1]["image"]).content))
    page = page.resize((int(page.width * 1.25), int(page.height * 1.25)))    
    img.paste(page, (50, 490))
    draw.text((300, 780), GetTranslation("stages", jsons["stages"][1]["name"]), font = font , fill = "#FFFFFF",anchor='mm')
    
    del draw
    if IsBackPaste:
        back.paste(img, (105,10),mask=mask)
        img = back
    return img
def CreateEventImage(jsons):
    back = Image.open("SplatoonBack_White.png").crop((0,0,960,625))
    img = Image.open("SplatoonBack_Black.png").crop((0,0,900,600))
    font = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 60)
    font_big = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 65)
    font_chu = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 45)
    font_mini = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 30)
    font_supermini = ImageFont.truetype("Corporate-Logo-Rounded-Bold-ver3.otf", 20)
    
    # 角丸にするためのマスクを作成
    mask = Image.new("L", img.size, 0)
    radius = 50
    draw = ImageDraw.Draw(mask)
    draw.rounded_rectangle((0, 0, img.width, img.height), radius, fill=255)

    draw = ImageDraw.Draw(img)
    
    draw.text((460, 50), "イベントマッチ", font = font, fill = "#FFFFFF", anchor='mm')

    draw.text((460, 130), GetTranslation("events", jsons["leagueMatchSetting"]["leagueMatchEvent"]["id"]), font = font, fill = "#FFFFFF", anchor="mm")
    stage = Image.open(io.BytesIO(requests.get(jsons["leagueMatchSetting"]["vsStages"][0]["image"]["url"]).content))
    stage = stage.resize((int(stage.size[0] / 1.25),
                          int(stage.size[1] / 1.25)))
    img.paste(stage, (140, 400))

    stage = Image.open(io.BytesIO(requests.get(jsons["leagueMatchSetting"]["vsStages"][-1]["image"]["url"]).content))
    stage = stage.resize((int(stage.size[0] / 1.25),
                          int(stage.size[1] / 1.25)))
    img.paste(stage, (480, 400))

    draw.text((440, 190), (datetime.strptime(jsons["timePeriods"][0]["startTime"], '%Y-%m-%dT%H:%M:%SZ') + timedelta(hours=9)).strftime('%Y年%m月%d日%H時')+"開始", font = font_chu, fill = "#FFFFFF", anchor="mm")
    draw.text((440, 240), (datetime.strptime(jsons["timePeriods"][1]["startTime"], '%Y-%m-%dT%H:%M:%SZ') + timedelta(hours=9)).strftime('%Y年%m月%d日%H時')+"開始", font = font_chu, fill = "#FFFFFF", anchor="mm")
    draw.text((440, 290), (datetime.strptime(jsons["timePeriods"][2]["startTime"], '%Y-%m-%dT%H:%M:%SZ') + timedelta(hours=9)).strftime('%Y年%m月%d日%H時')+"開始", font = font_chu, fill = "#FFFFFF", anchor="mm")
    draw.text((480, 350), GetTranslation("rules", jsons["leagueMatchSetting"]["vsRule"]["id"]), font = font, fill = "#FFFFFF", anchor="mm")
    
    """if datetime.utcnow() > jsons["start"]:
        starttext = ""
    else:
        starttext = (jsons["start"] + timedelta(hours=9)).strftime('%Y年%m月%d日%H時')+"開始"
    draw.text((550, 80), starttext, font = font_mini, fill = "#FFFFFF")

    #オカシラシャケ
    draw.text((545, 130), "オカシラシャケ", font = font_big, fill = "#FFFFFF")
    draw.text((615, 200), GetSalmonoidName(jsons["kingsalmonid"]), font = font_chu, fill = "#FFFFFF")
    
    #武器
    index = 0
    for weapon in jsons["weapons"]:
        weaponimg = Image.open(io.BytesIO(requests.get(weapon["image"]).content))
        weaponimg = weaponimg.resize((int(weaponimg.size[0] / 2),
                              int(weaponimg.size[1] / 2)))
        additional = 50
        img.paste(weaponimg, (40 + additional + (index * 190), 425), weaponimg)
        weapontext = weapon["name"]
        if weapontext is not "6e17fbe20efecca9":
            weapontext = GetTranslation("weapons", weapontext)
        else:
            weapontext = ""
        draw.text((50 + additional + (index * 190), 550), weapontext, font = font_supermini, fill = "#FFFFFF")
        index += 1
        """
    back.paste(img, (30,10), mask=mask)
    return back

IsTweeted = False

IsEvent = False
IsSalmon = False
lasteventrule = None

while True:
    utcnow = datetime.utcnow()
    if utcnow.hour % 2 == 0 and not IsTweeted:
        break
    if utcnow.minute >= 50 and not IsTweeted:        #ツイートする
        medias = []
        bankaradata = GetSchedulesData(schjson, "bankara")
        regulardata = GetSchedulesData(schjson, "regular")
        CreateSchImage(regulardata[1]["settings"][0], "レギュラーマッチ", 120).save("regular.png")

        bankaraopen = CreateSchImage(bankaradata[1]["settings"][0], "バンカラマッチ(チャレンジ)", 33, False)
        bankarachallenge = CreateSchImage(bankaradata[1]["settings"][1], "バンカラマッチ(オープン)", 40, False)

        # 角丸にするためのマスクを作成
        mask = Image.new("L", bankaraopen.size, 0)
        radius = 50
        draw = ImageDraw.Draw(mask)
        draw.rounded_rectangle((0, 0, bankaraopen.width, bankaraopen.height), radius, fill=255)

        back = Image.open("SplatoonBack_White.png").crop((0,0,1450,920))
        back.paste(bankaraopen, (105,10), mask=mask)
        back.paste(bankarachallenge, (755,10), mask=mask)
        back.save("bankara.png")

        xmatchdata = GetSchedulesData(schjson, "x")
        CreateSchImage(xmatchdata[1]["settings"][0], "Xマッチ", 200).save("xmatch.png")
        salmondata = GetSalmonData(schjson, "regular")
        
        filemedias = []
        filemedias.append("regular.png")
        filemedias.append("bankara.png")
        filemedias.append("xmatch.png")        

        isbreak = True
        if ((salmondata[1]["start"] - datetime.utcnow()).total_seconds() <= 3600):
            CreateSalmonImage(salmondata[1],"サーモンラン", 300).save("salmon.png")
            filemedias.append("salmon.png")
            isbreak = False
            IsSalmon = True
        if datetime.strptime(schjson["data"]["eventSchedules"]["nodes"][0]["timePeriods"][-1]["startTime"], '%Y-%m-%dT%H:%M:%SZ') < datetime.utcnow() and datetime.strptime(schjson["data"]["eventSchedules"]["nodes"][0]["timePeriods"][-1]["endTime"], '%Y-%m-%dT%H:%M:%SZ') > datetime.utcnow():
            isbreak = False
            IsEvent = True
            lasteventrule = schjson["data"]["eventSchedules"]["nodes"][-1]["timePeriods"][-1]["startTime"]
        tweettext = "もうすぐでスケジュール更新！\n"
        hourtime = utcnow.hour + 1 + 9
        if hourtime >= 24:
            hourtime -= 24
        tweettext += str(hourtime) + "時からのスケジュールです！\n"
        
        tweettext += "・レギュラーマッチ\n=>ナワバリバトル\n"
        #tweettext += GetTranslation("stages", regulardata[1]["settings"][0]["stages"][0]["name"])+"、"+GetTranslation("stages", regulardata[1]["settings"][0]["stages"][1]["name"])+"\n"
        
        tweettext += "・バンカラマッチ(オープン)\n=>"+GetTranslation("rules", bankaradata[1]["settings"][0]["rule"])+"\n"
        #tweettext += GetTranslation("stages", bankaradata[1]["settings"][0]["stages"][0]["name"])+"、"+GetTranslation("stages", bankaradata[1]["settings"][0]["stages"][1]["name"])+"\n"
        
        tweettext += "・バンカラマッチ(チャレンジ)\n=>"+GetTranslation("rules", bankaradata[1]["settings"][1]["rule"])+"\n"
        #tweettext += GetTranslation("stages", bankaradata[1]["settings"][1]["stages"][0]["name"])+"、"+GetTranslation("stages", bankaradata[1]["settings"][1]["stages"][1]["name"])+"\n"

        tweettext += "・Xマッチ\n=>"+GetTranslation("rules", xmatchdata[1]["settings"][0]["rule"])+"\n"
        #tweettext += GetTranslation("stages", xmatchdata[1]["settings"][0]["stages"][0]["name"])+"、"+GetTranslation("stages", xmatchdata[1]["settings"][0]["stages"][1]["name"])+"\n"

        for md in filemedias:
            medias.append(api.media_upload(filename=md).media_id)
        tweetid = client.create_tweet(text=tweettext, media_ids = medias).data["id"]
        
        medias = []
        for md in filemedias:
            medias.append((md,""))
        AtProWrapper.PostMessage(tweettext, imagedata=medias)
        try:
            payload = {"title":"スケジュールもうすぐ更新",
                       "url":"https://twitter.com/SplatoonSokuhou/status/"+str(tweetid),
                       "body":"スケジュールがもうすぐで更新されます！"
                       }
            pusher.PushMsg("ChangeSchedule",payload)
        except Exception as e:
            print(str(e))
        IsTweeted = True
        if isbreak:
            break
    elif IsTweeted and utcnow.hour % 2 == 0:
       time.sleep(30)
       schjson = requests.get("https://splatoon3.ink/data/schedules.json").json()
       IsBreak = False
       if IsSalmon:
        salmondata = GetSalmonData(schjson, "regular")
        if salmondata[0]["end"] > datetime.utcnow():
            CreateSalmonImage(salmondata[-1],"サーモンラン", 300).save("salmonnew.png")
            medias = []
            tweettext =  "新しいサーモンランのスケジュールが公開されました！\n"
            tweettext += "▼開始日時\n"
            tweettext += (salmondata[-1]["start"] + timedelta(hours=9)).strftime('%Y年%m月%d日%H時')+"\n"
            tweettext += "▼ステージ\n"
            tweettext += GetTranslation("stages", salmondata[-1]["stagename"])
            tweettext += "▼ブキ\n"
            for weapon in salmondata[-1]["weapons"]:
                weapontext = weapon["name"]
                weapontext = GetTranslation("weapons", weapontext)
                tweettext += "・" + weapontext + "\n"
            medias.append(api.media_upload(filename="salmonnew.png").media_id)
            tweetid = client.create_tweet(text=tweettext, media_ids = medias).data["id"]
            AtProWrapper.PostMessage(tweettext, imagedata=[("salmonnew.png","")])
            try:
                payload = {"title":"サーモンラン新シフト公開",
                           "url":"https://twitter.com/SplatoonSokuhou/status/"+str(tweetid),
                           "body":"サーモンランの新シフトが公開されました！\\nステージ:"+GetTranslation("stages", salmondata[-1]["stagename"])
                           }
                pusher.PushMsg("NewSalmonShift",payload)
            except Exception as e:
                print(str(e))
            IsBreak = True
       if IsEvent:
           if len(schjson["data"]["eventSchedules"]["nodes"])>1 and  datetime.strptime(schjson["data"]["eventSchedules"]["nodes"][0]["timePeriods"][-1]["endTime"], '%Y-%m-%dT%H:%M:%SZ') > datetime.utcnow() and schjson["data"]["eventSchedules"]["nodes"][-1]["timePeriods"][-1]["startTime"] is not lasteventrule:
            transjson = requests.get("https://splatoon3.ink/data/locale/ja-JP.json").json()
            CreateEventImage(schjson["data"]["eventSchedules"]["nodes"][-1]).save("event.png")
            tweettext =  "新しいイベントマッチスケジュールが公開されました！\n"
            tweettext += "▼ルール\n"
            tweettext += GetTranslation("events", schjson["data"]["eventSchedules"]["nodes"][-1]["leagueMatchSetting"]["leagueMatchEvent"]["id"])
            medias = []
            medias.append(api.media_upload(filename="event.png").media_id)
            tweetid = client.create_tweet(text=tweettext, media_ids = medias).data["id"]
            atmsgdata = AtProWrapper.PostMessage(tweettext, imagedata=[("event.png","")])
            try:
                payload = {"title":"イベントマッチ新スケジュール",
                           "url":"https://twitter.com/SplatoonSokuhou/status/"+str(tweetid),
                           "body":"イベントマッチの新スケジュールが公開されました！"
                           }
                pusher.PushMsg("NewEventSch",payload)
            except Exception as e:
                print(str(e))
            tweettext = "▼ルール説明\n"
            tweettext += GetTranslationRegulation("events", schjson["data"]["eventSchedules"]["nodes"][-1]["leagueMatchSetting"]["leagueMatchEvent"]["id"]).replace("<br","\n").replace("\n\n","\n").replace(" ","").replace(">","").replace("/","")
            client.create_tweet(text=tweettext, in_reply_to_tweet_id = tweetid)
            AtProWrapper.PostMessage(tweettext, reply_to=atmsgdata)
            IsBreak = True
       if IsBreak:
           break           
       time.sleep(30)