DR_Team/sr-download
DR_Team/sr-download
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

稍微改进一下（？

Browse Source
This commit is contained in:
shenjack 2024-08-30 18:17:31 +08:00
parent 6496799664
commit 38c8f5deff
Signed by: shenjack
GPG Key ID: 7B1134A979775551
2 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
sr_download/sql/.gitignore vendored Normal file
View File

@ -0,0 +1 @@
logs

12
sr_download/sql/xml_parse.py
View File

@ -27,13 +27,10 @@ def parse_part_types(xml_file):
tree = ET.parse(xml_file) tree = ET.parse(xml_file)
root = tree.getroot() root = tree.getroot()
# 定义命名空间
namespace = {'ns': 'http://shenjack.top:81/files/DR/xsd/partlist.xsd'}
part_dict = {} part_dict = {}
# 遍历所有 PartType 元素 # 遍历所有 PartType 元素
for part in root.findall('ns:PartType', namespace): for part in root.findall('PartType'):
part_id = part.get('id') part_id = part.get('id')
part_mass = float(part.get('mass')) # type: ignore part_mass = float(part.get('mass')) # type: ignore
part_dict[part_id] = part_mass * 500 # 缩放因子 part_dict[part_id] = part_mass * 500 # 缩放因子
@ -48,6 +45,7 @@ WITH data AS (
WHERE "save_type" = 'ship' WHERE "save_type" = 'ship'
AND full_data.xml_tested AND full_data.xml_tested
AND save_id >= 1200000 AND save_id >= 1200000
ORDER BY save_id ASC
LIMIT {limit} OFFSET {offset} LIMIT {limit} OFFSET {offset}
), ),
parts_data AS ( parts_data AS (
@ -79,11 +77,11 @@ def main():
db = get_db() db = get_db()
db_cur = db.cursor() db_cur = db.cursor()
offset = 51500 offset = 0
limit = 500 limit = 100
target_mass = 775150 target_mass = 775150
delta = 100 delta = 10000
start_time = time.time() start_time = time.time()