mirror of
https://github.com/PaiGramTeam/Gift-Code-Web.git
synced 2024-09-28 03:46:24 +00:00
37 lines
957 B
Python
37 lines
957 B
Python
import json
|
|
from typing import List
|
|
|
|
from models.code import CodeList, Code
|
|
from models.honkai import get_code
|
|
|
|
from pathlib import Path
|
|
|
|
data_path = Path("data")
|
|
custom_path = data_path / "custom.json"
|
|
code_path = data_path / "code.json"
|
|
|
|
|
|
def merge_code(over: List[Code], custom: CodeList) -> CodeList:
|
|
over_codes = [i for i in over]
|
|
custom_over_codes = [i.code for i in custom.over]
|
|
for code in over_codes:
|
|
if code.code in custom_over_codes:
|
|
continue
|
|
custom.over.append(code)
|
|
return custom
|
|
|
|
|
|
def main():
|
|
over = get_code()
|
|
if not over:
|
|
raise ValueError("Get code failed")
|
|
with open(custom_path, "r", encoding="utf-8") as f:
|
|
custom = CodeList.model_validate_json(f.read())
|
|
custom = merge_code(over, custom)
|
|
with open(code_path, "w", encoding="utf-8") as f:
|
|
f.write(json.dumps(custom.model_dump(), indent=4, ensure_ascii=False))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|