fgoscdata/extract_event.py at master · fgosc/fgoscdata · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
#!/usr/bin/env python3
import json
from pathlib import Path
import csv
import argparse
import datetime
import logging

import requests

logger = logging.getLogger(__name__)

mstQuest_url = "https://raw.githubusercontent.com/FZFalzar/FGOData/master/JP_tables/quest/mstQuest.json"
mstQuestInfo_url = "https://raw.githubusercontent.com/FZFalzar/FGOData/master/JP_tables/quest/viewQuestInfo.json"
#mstQuestInfo_url = "https://raw.githubusercontent.com/FZFalzar/FGOData/81542cc7fa715efbc55fca95b4fffea5fcf1450a/JP_tables/quest/viewQuestInfo.json"
mstQuestPhase_url = "https://raw.githubusercontent.com/FZFalzar/FGOData/master/JP_tables/quest/mstQuestPhase.json"


def list2class(enemy):
    enemy_dic = {
                 1:"剣", 2:"弓", 3:"槍", 4:"騎", 5:"術", 6:"殺", 7:"狂",
                 8:"盾", 9:"裁", 10:"分", 11:"讐", 23:"月", 25:"降"
                 }
    out = ""
    for e in enemy:
        out += enemy_dic[e]
    return out

def main(args):
    if args.openedat:
        opened = int(datetime.datetime.strptime(args.openedat, "%Y/%m/%d").timestamp())
        logger.debug("opendat: %s", opened)
    if args.closedat:
        closed = int(datetime.datetime.strptime(args.closedat + ' 23:59:59', "%Y/%m/%d %H:%M:%S").timestamp())
        logger.debug("closedat: %s", closed)

    r_get = requests.get(mstQuest_url)
    mstQuest_list = r_get.json()
    r_get2 = requests.get(mstQuestInfo_url)
    mstQuestInfo_list = r_get2.json()
    r_get3 = requests.get(mstQuestPhase_url)
    mstQuestPhase_list = r_get3.json()
    questId2classIds = {q["questId"]: q["classIds"] for q in mstQuestPhase_list}

    q_list = []
    for quest in mstQuest_list:
        # if opened <= quest["openedAt"] and quest["closedAt"] <= closed:
        if not (94000000 < quest["id"] < 100000000):
            continue
        if "種火集め" in quest["name"] or "宝物庫" in quest["name"] or  "修練場" in quest["name"]:
            continue
        if quest["closedAt"] == 1901199599:
            continue
        logger.info(quest["id"])
        for q in mstQuestInfo_list:
            if q["questId"] == quest["id"] and "高難易度" not in quest["name"]:
                enemy = questId2classIds[quest["id"]]
                q_list.append([quest["id"], quest["name"], quest["recommendLv"], list2class(enemy)])
                break

    with open("event_list.csv", "w", encoding='UTF-8') as f:
        writer = csv.writer(f, lineterminator="\n")
        writer.writerows(q_list)


if __name__ == '__main__':
    parser = argparse.ArgumentParser(description='FGOクエストをCSV出力')
    parser.add_argument('-o', '--openedat',
                        help='quest open date: e.g. 2015/7/30')
    parser.add_argument('-c', '--closedat',
                        help='quest close date: e.g.  2020/11/03')
    parser.add_argument('-l', '--loglevel',
                        choices=('debug', 'info'), default='info')
    args = parser.parse_args()    # 引数を解析
    lformat = '[%(levelname)s] %(message)s'
    logging.basicConfig(
        level=logging.INFO,
        format=lformat,
    )
    logger.setLevel(args.loglevel.upper())
    main(args)