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
|
import argparse
import csv
import os
import requests
DIFFICULTY_MAPPING = {
"NOVICE": "NOV",
"ADVANCED": "ADV",
"EXHAUST": "EXH",
"INFINITE": "INF",
"GRAVITY": "GRV",
"HEAVENLY": "HVN",
"VIVID": "VVD",
"EXCEED": "XCD",
"MAXIMUM": "MXM"
}
def merge_csv(input_file: str, tachi_url: str, username: str, output_file: str):
encoding = "utf-8"
header_written = False
with open(input_file, encoding=encoding) as old_csv, open(output_file, 'w', newline='', encoding=encoding) as new_csv:
reader = csv.reader(old_csv, delimiter=",")
writer = csv.writer(new_csv, delimiter=",")
header = next(reader)
# Count total rows for progress tracking
rows = list(reader)
total_rows = len(rows)
current_row = 0
for row in rows:
current_row += 1
title = requests.utils.quote(row[0])
diff = DIFFICULTY_MAPPING[row[1]]
score = int(row[5])
api_url = f"{tachi_url}/api/v1/users/{username}/games/sdvx/Single/pbs?search={title}"
print(f"[{current_row}/{total_rows}] Searching for score: {title} at difficulty {diff}")
try:
response = requests.get(api_url)
response.raise_for_status()
data = response.json()
charts = data["body"]["charts"]
is_unique = False
if len(charts) == 0:
print("Score is unique")
is_unique = True
if not header_written:
writer.writerow(header)
header_written = True
writer.writerow(row)
continue
chart_id = ""
for chart in charts:
if chart["difficulty"] == diff:
print("Found proper chart ID")
chart_id = chart["chartID"]
break
if chart_id == "":
print("Score is Unique, unable to find a match in the DB")
if not header_written:
writer.writerow(header)
header_written = True
writer.writerow(row)
continue
pbs = data["body"]["pbs"]
for pb in pbs:
if pb["chartID"] == chart_id and pb["scoreData"]["score"] == score:
print("Found match, score is not unique")
is_unique = False
break
if is_unique:
if not header_written:
writer.writerow(header)
header_written = True
writer.writerow(row)
except requests.RequestException as e:
print(f"Error fetching data for {title}: {e}")
continue
if __name__ == "__main__":
parser = argparse.ArgumentParser(
prog="pb_merge.py",
description="Takes in a SDVX e-amusement PB CSV, checks against scores on Tachi and generates a new file with only unique scores",
)
parser.add_argument(
"-f", "--file", help="SOUND VOLTEX score CSV (score.csv)", required=True
)
parser.add_argument(
"-o", "--output", help="Output filename", default="merged_sdvx_scores.csv"
)
parser.add_argument(
"-t",
"--tachi",
help="API URL for your Tachi instance",
default="https://kamai.tachi.ac",
)
parser.add_argument("-u", "--username", help="Your unique username on Tachi", default="")
args = parser.parse_args()
if args.file is None:
print("ERROR: Please specify the path to the score CSV")
exit(1)
if not os.path.exists(args.file):
print(f"ERROR: The file {args.file} does not exist.")
exit(1)
merge_csv(args.file, args.tachi, args.username, args.output)
|
