-
-
Notifications
You must be signed in to change notification settings - Fork 22
/
Copy pathOTA.py
141 lines (133 loc) · 8.46 KB
/
OTA.py
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
import requests
import brotli
import logging
import sqlite3
import time
import json
import os
import PIL.Image as Image
# https://redive.estertion.win/last_version_jp.json
# https://redive.estertion.win/last_version_cn.json
# https://redive.estertion.win/db/redive_jp.db.br
# https://redive.estertion.win/db/redive_cn.db.br
global_logger = logging.getLogger()
def updateCharacterIndexListByURL(url):
global_logger.warning("开始尝试更新CharacterIndexList。来源：%s" % url)
headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/89.0.4389.82 Safari/537.36',
'Referer': 'https://redive.estertion.win/api.htm',
'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.9',
'Accept-Encoding': 'br',
'Accept-Language': 'en-US,en;q=0.9,zh-CN;q=0.8,zh-TW;q=0.7,zh;q=0.6,ja;q=0.5',
'Accept-Charset': "UTF-8"
}
try:
# 获取网页内容，返回html数据
response = requests.get(url, headers=headers)
# 通过状态码判断是否获取成功
if response.status_code == 200:
# response.encoding = 'utf-8'
key = 'Content-Encoding'
# print(response.headers[key])
data = brotli.decompress(response.content)
savedBinFile = open("./Database.db", "wb")
savedBinFile.write(data)
savedBinFile.close()
connection = sqlite3.connect("./Database.db")
unitDataCursor = connection.cursor().execute("SELECT a.* FROM unit_skill_data b, unit_data a WHERE a.unit_id = b.unit_id AND a.unit_id < 400000")
characterIndexList = []
for row in unitDataCursor:
characterIndexList.append({"unit_id": row[0], "unit_name": row[1]})
characterIndexListJsonFile = open(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "characterIndexList.json"),'w',encoding='utf-8')
json.dump(characterIndexList, characterIndexListJsonFile, ensure_ascii=False)
characterIndexListJsonFile.close()
global_logger.warning("成功更新CharacterIndexList。来源：%s" % url)
return None
except Exception as e:
print(e)
global_logger.exception("updateCharacterIndexListByURL失败")
global_logger.exception("Exception %s" % e)
return None
def updateAssetsByCharacterIndexList(characterIndexList):
SixCharIDList = []
try:
connection = sqlite3.connect("./Database.db")
unitDataCursor = connection.cursor().execute("SELECT * from unlock_rarity_6 where unlock_flag = 1")
for row in unitDataCursor:
SixCharIDList.append(str(row[0])[:4])
except:
global_logger.warning("没有获取到六星角色信息。")
def checkIfMissingSix(charId):
flag = False
if charId in SixCharIDList:
if not os.path.exists(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s61.webp" % charId)):
flag = True
return flag
global_logger.warning("开始尝试更新角色图像。")
for entry in characterIndexList:
charId = str(entry["unit_id"])[:4]
__missingOne = not os.path.exists(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s11.webp" % charId))
__missingThree = not os.path.exists(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s31.webp" % charId))
__missingSix = checkIfMissingSix(charId)
if __missingOne or __missingThree or __missingSix:
global_logger.warning("发现%s(id:%s)的头像文件缺失，尝试重新获取" % (entry["unit_name"], entry["unit_id"]) )
webpURLOne = "https://redive.estertion.win/icon/unit/%s11.webp" % charId
webpURLThree = "https://redive.estertion.win/icon/unit/%s31.webp" % charId
webpURLSix = "https://redive.estertion.win/icon/unit/%s61.webp" % charId
try:
if __missingOne:
global_logger.warning("尝试获取%s(id:%s)的一星头像文件（url: %s)" % (entry["unit_name"], entry["unit_id"], webpURLOne))
webpContentOne = requests.get(webpURLOne)
if webpContentOne.status_code == 200:
with open(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s11%s" % (charId, ".webp")), 'wb') as file:
file.write(webpContentOne.content)
global_logger.warning("成功下载角色头像（角色名：%s, 角色id：%s，角色头像星级：1)" % (entry["unit_name"], entry["unit_id"]))
if __missingThree:
global_logger.warning("尝试获取%s(id:%s)的三星头像文件（url: %s)" % (entry["unit_name"], entry["unit_id"], webpURLThree))
webpContentThree = requests.get(webpURLThree)
if webpContentThree.status_code == 200:
with open(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s31%s" % (charId, ".webp")), 'wb') as file:
file.write(webpContentThree.content)
global_logger.warning("成功下载角色头像（角色名：%s, 角色id：%s，角色头像星级：3)" % (entry["unit_name"], entry["unit_id"]))
if __missingSix:
global_logger.warning("尝试获取%s(id:%s)的六星头像文件（url: %s)" % (entry["unit_name"], entry["unit_id"], webpURLSix))
webpContentSix = requests.get(webpURLSix)
if webpContentSix.status_code == 200:
with open(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s61%s" % (charId, ".webp")), 'wb') as file:
file.write(webpContentSix.content)
global_logger.warning("成功下载角色头像（角色名：%s, 角色id：%s, 角色头像星级：6)" % (entry["unit_name"], entry["unit_id"]))
except Exception as e:
global_logger.error("尝试获取%s（id:%s)头像文件失败(FlagOne:%s, FlagThree:%s, FlagSix:%s)" % (entry["unit_name"], entry["unit_id"],__missingOne, __missingThree, __missingSix ))
global_logger.warning("更新角色图像完成。")
def generateRefImageByCharacterIndexList(characterIndexList):
global_logger.warning("开始尝试拼接头像生成refImage。")
refImagePath = os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "refImage.png")
refImage = Image.new('RGBA', (len(characterIndexList) * (60+2) * 3, 62)) #创建一个新图
for index in range(len(characterIndexList)):
charId = str(characterIndexList[index]['unit_id'])[:4]
try:
icon_imageOne = Image.open(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s11.webp" % charId)).resize((60, 60),Image.ANTIALIAS)
coordinateImageOne = ((2+3*62*index, 2))
refImage.paste(icon_imageOne, coordinateImageOne)
icon_imageThree = Image.open(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s31.webp" % charId)).resize((60, 60),Image.ANTIALIAS)
coordinateImageThree = ((64+3*62*index, 2))
refImage.paste(icon_imageThree, coordinateImageThree)
icon_imageSix = Image.open(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "%s61.webp" % charId)).resize((60, 60),Image.ANTIALIAS)
coordinateImageSix = ((126+3*62*index, 2))
refImage.paste(icon_imageSix, coordinateImageSix)
except Exception as e:
global_logger.debug("图片粘贴失败, 可能是没有成功下载到文件。Exception: %s" % e)
refImage.save(refImagePath)
global_logger.warning("成功更新refImage")
def devMain():
logging.basicConfig(format='%(asctime)s - %(pathname)s[line:%(lineno)d] - %(levelname)s: %(message)s',
level=logging.WARNING)
# url = "https://redive.estertion.win/db/redive_jp.db.br"
# updateCharacterIndexListByURL(url)
characterIndexListJsonFile = open(os.path.join(os.path.expanduser('~'), "PCRJJCAnalyzer", "CharData", "characterIndexList.json"),'r',encoding='utf-8')
characterIndexList = json.load(characterIndexListJsonFile)
updateAssetsByCharacterIndexList(characterIndexList)
generateRefImageByCharacterIndexList(characterIndexList)
if __name__ == "__main__":
# 直接执行则为开发模式入口
devMain()