mirror of
https://github.com/Kvan7/Exiled-Exchange-2.git
synced 2026-05-03 16:01:14 +00:00
kvan7
129 lines
4.7 KiB
Python
129 lines
4.7 KiB
Python
import argparse
|
|
import json
|
|
import logging
|
|
import shutil
|
|
|
|
from tqdm import tqdm
|
|
|
|
from constants.client_string_data import (
|
|
CLIENT_STRING_ARRAYS,
|
|
CLIENT_STRING_KEY_VALUES,
|
|
CLIENT_STRING_REGEX,
|
|
)
|
|
from constants.lang import ENGLISH, LANG, LANGUAGES
|
|
from providers.game_api import GameDataProvider
|
|
from providers.trade_api import TradeApiProvider
|
|
from services.client_strings_builder import ClientStringsBuilder
|
|
from services.image_provider import MODE
|
|
from services.nd_builder_service import NdBuilderService
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def main(**kwargs: dict[str, bool | str]):
|
|
pull: bool = kwargs.get("pull", False) # pyright: ignore[reportAssignmentType]
|
|
image_mode: MODE = kwargs.get("image_mode", "noLookup") # pyright: ignore[reportAssignmentType]
|
|
push: bool = kwargs.get("push", False) # pyright: ignore[reportAssignmentType]
|
|
main_repo_path: str = kwargs.get("main_repo_path", "../exiled-exchange-2") # pyright: ignore[reportAssignmentType]
|
|
|
|
if pull:
|
|
GameDataProvider().run_export()
|
|
for lang in tqdm(LANGUAGES):
|
|
TradeApiProvider(lang).pull()
|
|
|
|
run_on: list[LANG] = [ENGLISH, *[lang for lang in LANGUAGES if lang != ENGLISH]]
|
|
|
|
for index, lang in enumerate(run_on):
|
|
logger.info(
|
|
"================================================================================"
|
|
)
|
|
logger.info(f"Building {lang}: {index + 1}/{len(LANGUAGES)}")
|
|
logger.info(
|
|
"================================================================================"
|
|
)
|
|
nd = NdBuilderService(
|
|
lang,
|
|
image_mode if lang is ENGLISH else "noLookup",
|
|
)
|
|
|
|
stats = nd.build_stats_ndjson()
|
|
stats_o = stats.where(stats.notna(), None)
|
|
with open(f"output/{lang}/stats.ndjson", "w", encoding="utf-8") as f:
|
|
for _, row in stats_o.iterrows():
|
|
obj = {k: v for k, v in row.items() if v is not None}
|
|
f.write(f"{json.dumps(obj, ensure_ascii=False)}\n")
|
|
|
|
items = nd.build_items_ndjson()
|
|
items_o = items.where(items.notna(), None)
|
|
with open(f"output/{lang}/items.ndjson", "w", encoding="utf-8") as f:
|
|
for _, row in items_o.iterrows():
|
|
obj = {k: v for k, v in row.items() if v is not None}
|
|
try:
|
|
f.write(f"{json.dumps(obj, ensure_ascii=False)}\n")
|
|
except Exception as e:
|
|
print(e)
|
|
print(obj)
|
|
raise
|
|
|
|
client_strings_builder = ClientStringsBuilder(lang)
|
|
client_strings_js = client_strings_builder.build(
|
|
base_strings=CLIENT_STRING_KEY_VALUES,
|
|
array_strings=CLIENT_STRING_ARRAYS,
|
|
regex_strings=CLIENT_STRING_REGEX,
|
|
)
|
|
with open(f"output/{lang}/client_strings.js", "w", encoding="utf-8") as f:
|
|
try:
|
|
f.write(client_strings_js)
|
|
except Exception as e:
|
|
print(e)
|
|
print(client_strings_js)
|
|
raise
|
|
|
|
# Copy files to EE2 repo
|
|
if push:
|
|
logger.info("Copying files to EE2 repo")
|
|
_ = shutil.copy(
|
|
f"output/{lang}/stats.ndjson",
|
|
f"{main_repo_path}/renderer/public/data/{lang}/stats.ndjson",
|
|
)
|
|
_ = shutil.copy(
|
|
f"output/{lang}/items.ndjson",
|
|
f"{main_repo_path}/renderer/public/data/{lang}/items.ndjson",
|
|
)
|
|
_ = shutil.copy(
|
|
f"output/{lang}/client_strings.js",
|
|
f"{main_repo_path}/renderer/public/data/{lang}/client_strings.js",
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser()
|
|
_ = parser.add_argument("--pull", action="store_true", help="Pull trade data")
|
|
_ = parser.add_argument(
|
|
"--log", default="info", help="Log level (debug, info, warn, error)"
|
|
)
|
|
_ = parser.add_argument(
|
|
"--image-mode",
|
|
default="noLookup",
|
|
choices=["noLookup", "missing", "new", "all"],
|
|
help="Image mode (noLookup, missing, new, all)",
|
|
)
|
|
_ = parser.add_argument(
|
|
"--push",
|
|
action="store_true",
|
|
help="Push to EE2 repo",
|
|
)
|
|
_ = parser.add_argument(
|
|
"--main-repo-path",
|
|
default="../exiled-exchange-2",
|
|
help="Path to main repo",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
logging.getLogger().setLevel(args.log.upper()) # pyright:ignore [reportAny]
|
|
main(**vars(args)) # pyright:ignore [reportAny]
|