autotask.py

import datetime
import hashlib
import json
import os
import time

import yaml
import requests
from apscheduler.schedulers.blocking import BlockingScheduler
from zhconv import convert
from modules.config import proxies, music_meta_url, cheerful_predict_url
from modules.sk import currentevent


def time_printer(str):
    timeArray = time.localtime(time.time())
    Time = time.strftime("%Y-%m-%d %H:%M:%S", timeArray)
    print(Time, str)

def detectplaydata():
    time_printer('检查playdata更新')
    try:
        jsondata = requests.get('https://gitlab.com/pjsekai/database/musics/-/raw/main/musicDifficulties.json',
                                proxies=proxies)
        json.loads(jsondata.text)
    except:
        print('playdata下载失败')
        return

    with open('masterdata/realtime/musicDifficulties.json', 'rb') as f:
        data = f.read()
    if hashlib.md5(data).hexdigest() != hashlib.md5(jsondata.content).hexdigest():
        print('更新musicDifficulties.json')
        with open("masterdata/realtime/musicDifficulties.json", "wb") as f:
            f.write(jsondata.content)

    try:
        jsondata = requests.get('https://gitlab.com/pjsekai/database/musics/-/raw/main/musics.json',
                                proxies=proxies)
        json.loads(jsondata.text)
    except:
        print('playdata下载失败')
        return

    with open('masterdata/realtime/musics.json', 'rb') as f:
        data = f.read()
    if hashlib.md5(data).hexdigest() != hashlib.md5(jsondata.content).hexdigest():
        print('更新musics.json')
        with open("masterdata/realtime/musics.json", "wb") as f:
            f.write(jsondata.content)

def get_filectime(file):
    return datetime.datetime.fromtimestamp(os.path.getmtime(file))


def cleancache(path='piccache/'):
    nowtime = datetime.datetime.now()
    deltime = datetime.timedelta(seconds=300)
    nd = nowtime - deltime
    for file in os.listdir('piccache/'):
        if file[-4:] == '.png' or file[-4:] == '.mp3' or file[-4:] == '.jpg':
            filectime = get_filectime(path + file)
            if filectime < nd:
                os.remove(path + file)
                # print(f"删除{file} (缓存{nowtime - filectime})")
            else:
                pass
                # print(f"跳过{file} (缓存{nowtime - filectime})")

def updatetranslate(raw, value):
    with open('yamls/translate.yaml', encoding='utf-8') as f:
        translation = yaml.load(f, Loader=yaml.FullLoader)
    try:
        if translation[value] is None:
            translation[value] = {}
    except KeyError:
        translation[value] = {}
    try:
        request = requests.get(f'https://raw.githubusercontent.com/Sekai-World/sekai-i18n/main/zh-TW/{raw}.json',
                               proxies=proxies)
        data = request.json()
    except:
        print(raw + '翻译下载失败')
        return

    for i in data:
        try:
            translation[value][int(i)]
        except KeyError:
            zhhan = convert(data[i], 'zh-cn')
            translation[value][int(i)] = zhhan
            print('更新翻译', value, i, zhhan)
    with open('yamls/translate.yaml', 'w', encoding='utf-8') as f:
        yaml.dump(translation, f, allow_unicode=True)


def updatealltrans():
    time_printer('检查新增翻译')
    updatetranslate('music_titles', 'musics')
    updatetranslate('event_name', 'events')
    updatetranslate('card_prefix', 'card_prefix')
    updatetranslate('cheerful_carnival_teams', 'cheerfulCarnivalTeams')
    updatetranslate('card_gacha_phrase', 'card_phrase')
    updatetranslate('card_skill_name', 'skill_name')
    updatetranslate('skill_desc', 'skill_desc')


def update_music_meta():
    time_printer('尝试更新music_metas.json')
    try:
        jsondata = requests.get(music_meta_url, proxies=proxies)
        jsondata.json()
    except:
        print('\n' + 'music_meta下载失败')
        return
    try:
        with open('masterdata/realtime/music_metas.json', 'rb') as f:
            data = f.read()
    except FileNotFoundError:
        data = b''
    if hashlib.md5(data).hexdigest() != hashlib.md5(jsondata.content).hexdigest():
        print('\n' + '更新music_metas.json')
        with open("masterdata/realtime/music_metas.json", "wb") as f:
            f.write(jsondata.content)


def update_cheerful_predict():
    event = currentevent('jp')
    if event['status'] == 'going' and event['eventType'] == 'cheerful_carnival':
        time_printer('尝试更新cheerful_predict.json')
        try:
            jsondata = requests.get(cheerful_predict_url, proxies=proxies)
            jsondata.json()
        except:
            print('\n' + 'cheerful_predict下载失败')
            return
        try:
            with open('masterdata/realtime/cheerful_predict.json', 'rb') as f:
                data = f.read()
        except FileNotFoundError:
            data = b''
        if hashlib.md5(data).hexdigest() != hashlib.md5(jsondata.content).hexdigest():
            time_printer('更新cheerful_predict.json')
            with open("masterdata/realtime/cheerful_predict.json", "wb") as f:
                f.write(jsondata.content)


if __name__ == '__main__':
    cleancache()
    detectplaydata()
    updatealltrans()
    update_music_meta()
    scheduler = BlockingScheduler()
    scheduler.add_job(detectplaydata, 'interval', seconds=300, id='playinfocheck')
    scheduler.add_job(cleancache, 'interval', seconds=300, id='cleancache')
    scheduler.add_job(updatealltrans, 'interval', hours=2, id='updatealltrans')
    scheduler.add_job(update_cheerful_predict, 'cron', minute=52)
    scheduler.add_job(update_music_meta, 'cron', hour=13, id='update_music_meta')
    scheduler.start()