Max/pinapelz_more-tachi-import-scripts
1
0
Fork 0
mirror of https://gitea.tendokyu.moe/pinapelz/more-tachi-import-scripts.git synced 2026-10-11 17:08:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

sdvx_eamuse: add option to auto pull date data with cookie header

Browse Source
This commit is contained in:
Pinapelz
2025-05-25 01:45:14 -07:00
parent cbb6a4a913
commit bef905d2a9
2 changed files with 28 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sdvx/eamuse_csv/README.md
+1
View File
@@ -47,6 +47,7 @@ https://p.eagate.573.jp/game/sdvx/vi/playdata/profile/index.html
| `--time` | `-t` | UNIX time (in milliseconds) that should be added to the scores. Defaults to current UNIX time | `None` |
| `--date_file` | `-d` | SDVX e-amusement profile site saved HTML. See README in sdvx/eamuse_csv for instructions. Overrides with --time input if missing | — |
| `--timezone` | `-tz` | Only needed if -d is used, specifies what timezone to convert to | — |
| `--cookie` | `-c` | Automatically pull profile data from e-amusement using cookie | — |
## `eamuse_merge_to_csv`
sdvx/eamuse_csv/sdvx_csv_to_tachi.py
+27
View File
@@ -2,6 +2,7 @@ import csv
import json
import argparse
import time
import requests
DIFFICULTY_MAPPING = {
"NOVICE": "NOV",
@@ -23,6 +24,28 @@ LAMP_MAPPING = {
"EXCESSIVE COMPLETE": "EXCESSIVE CLEAR"
}
GAME_DATA = {
"EXCEED_GEAR": {
"PROFILE_URL": "https://p.eagate.573.jp/game/sdvx/vi/playdata/profile/index.html"
}
}
def get_site_data_with_cookie(url, cookie_header):
cookies = {}
for cookie in cookie_header.split(";"):
name, value = cookie.strip().split("=", 1)
cookies[name] = value
headers = {
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/117.0.0.0 Safari/537.36",
"Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,*/*;q=0.8",
"Accept-Language": "en-US,en;q=0.5",
"Connection": "keep-alive",
"Upgrade-Insecure-Requests": "1",
}
response = requests.get(url, cookies=cookies, headers=headers)
response.raise_for_status()
return response.text
def create_date_dict(game_ver:str, raw_site_data: str) -> dict:
from bs4 import BeautifulSoup
from datetime import datetime
@@ -123,6 +146,7 @@ if __name__ == "__main__":
parser.add_argument("-t", "--time", help="UNIX time (in milliseconds) that should be added to the scores. Defaults to current UNIX time", default=None)
parser.add_argument("-d", "--date_file", help="SDVX e-amusement profile site saved HTML. See README in sdvx/eamuse_csv for instructions. Overrides with --time input if missing", required=False)
parser.add_argument("-tz", "--timezone", help="Only needed if -d is used, specifies what timezone to convert to", required=False)
parser.add_argument("-c", "--cookie", help="Pass in a cookie header to automatically pull date data")
args = parser.parse_args()
assert args.game in ["EXCEED_GEAR"]
curr_time = int(time.time() * 1000) if (args.time == "now" or args.time is None) else args.time
@@ -136,6 +160,9 @@ if __name__ == "__main__":
with open(args.date_file, "r") as file:
site_data = file.read()
date_data = create_date_dict(args.game, site_data)
elif args.cookie:
site_data = get_site_data_with_cookie(GAME_DATA[args.game]["PROFILE_URL"])
date_data = create_date_dict(args.game, site_data)
try:
output_json = convert_sdvx_csv_to_tachi_json(args.csv_filename, "sdvx", "Single", args.service, curr_time, date_data)