feat: move createFaceByBundle.py to codegen.py and add mesh scraping
This commit is contained in:
@@ -0,0 +1,101 @@
|
||||
# /// script
|
||||
# requires-python = ">=3.13"
|
||||
# dependencies = ["requests>=2.33.0"]
|
||||
# ///
|
||||
|
||||
import time
|
||||
|
||||
import requests
|
||||
|
||||
faces = []
|
||||
|
||||
# get array of face asset ids
|
||||
with open("faces.csv", mode="r") as faces_file:
|
||||
for row in faces_file:
|
||||
faces.append(row.strip())
|
||||
|
||||
bundle_face_dict = {}
|
||||
|
||||
for face in faces:
|
||||
delay = 1.0
|
||||
for attempt in range(5):
|
||||
try:
|
||||
response = requests.get(url=f"https://roblox.com/catalog/{face}")
|
||||
break
|
||||
except requests.exceptions.ConnectionError:
|
||||
if attempt == 4:
|
||||
raise
|
||||
print(f"Connection error for face {face}, retrying in {delay}s...")
|
||||
time.sleep(delay)
|
||||
delay *= 2
|
||||
else:
|
||||
bundle_face_dict[face] = None
|
||||
continue
|
||||
|
||||
# some faces don't have bundles
|
||||
redirected_to_bundle = any(
|
||||
r.status_code == 302 and "/bundles/" in r.headers.get("Location", "")
|
||||
for r in response.history
|
||||
)
|
||||
if redirected_to_bundle:
|
||||
bundle_id = response.url.split("/")[4]
|
||||
else:
|
||||
bundle_id = None
|
||||
bundle_face_dict[face] = bundle_id
|
||||
time.sleep(1.0)
|
||||
|
||||
lines = ["local faceByBundle: {[number]: number} = {"]
|
||||
for face, bundle_id in bundle_face_dict.items():
|
||||
if bundle_id is not None:
|
||||
lines.append(f"\t[{bundle_id}] = {face},")
|
||||
lines.append("}")
|
||||
lines.append("")
|
||||
lines.append("return faceByBundle")
|
||||
|
||||
with open("../src/faceByBundle.luau", mode="w") as f:
|
||||
f.write("\n".join(lines) + "\n")
|
||||
|
||||
mesh_face_dict = {}
|
||||
|
||||
for face, bundle_id in bundle_face_dict.items():
|
||||
if bundle_id is None:
|
||||
continue
|
||||
delay = 1.0
|
||||
for attempt in range(5):
|
||||
try:
|
||||
response = requests.get(url=f"https://catalog.roblox.com/v1/bundles/{bundle_id}/details")
|
||||
if response.status_code == 429:
|
||||
print(f"Rate limited on bundle {bundle_id}, retrying in {delay}s...")
|
||||
time.sleep(delay)
|
||||
delay *= 2
|
||||
continue
|
||||
response.raise_for_status()
|
||||
break
|
||||
except requests.exceptions.ConnectionError:
|
||||
if attempt == 4:
|
||||
raise
|
||||
print(f"Connection error for bundle {bundle_id}, retrying in {delay}s...")
|
||||
time.sleep(delay)
|
||||
delay *= 2
|
||||
else:
|
||||
print(f"Skipping bundle {bundle_id} after 5 attempts")
|
||||
continue
|
||||
|
||||
details = response.json()
|
||||
head_item = next(
|
||||
(item for item in details["items"] if item["type"] == "Asset" and "- Head" in item["name"]),
|
||||
None,
|
||||
)
|
||||
if head_item is not None:
|
||||
mesh_face_dict[head_item["id"]] = face
|
||||
time.sleep(1.0)
|
||||
|
||||
lines = ["local faceByMesh: {[number]: number} = {"]
|
||||
for mesh_id, face in mesh_face_dict.items():
|
||||
lines.append(f"\t[{mesh_id}] = {face},")
|
||||
lines.append("}")
|
||||
lines.append("")
|
||||
lines.append("return faceByMesh")
|
||||
|
||||
with open("../src/faceByMesh.luau", mode="w") as f:
|
||||
f.write("\n".join(lines) + "\n")
|
||||
Reference in New Issue
Block a user