merge_update_infos.py

"""
更新履歴をマージする。
"""

import argparse
import json
from collections import OrderedDict
from pathlib import Path


def merge_json_string(src: str, dst: str) -> str:
    """
    バージョンが同じ場合は要素を結合する
    >>> src = '[{"version": "0.0.1", "a": ["a1"], "b": ["b1", "b2"]}]'
    >>> dst = '[{"version": "0.0.1", "a": ["a2"], "b": ["b1", "b3"]}]'
    >>> merge_json_string(src, dst)
    '[{"version": "0.0.1", "a": ["a1", "a2"], "b": ["b1", "b2", "b3"]}]'

    バージョンが無かった場合は無視される
    >>> src = '[{"version": "1"}]'
    >>> dst = '[{"version": "1"}, {"version": "2"}]'
    >>> merge_json_string(src, dst)
    '[{"version": "1"}]'
    """
    # FIXME: バリデーションする
    # TODO: `str | list[str]`だけど`str`が来るとエラーになるのでならないようにしたい
    src_json: list[dict[str, str | list[str]]] = json.loads(src)
    dst_json: list[dict[str, str | list[str]]] = json.loads(dst)

    for src_item in src_json:
        for dst_item in dst_json:
            if src_item["version"] == dst_item["version"]:
                for key in src_item:
                    if key == "version":
                        continue

                    src_value = src_item[key]
                    dst_value = dst_item[key]
                    assert isinstance(src_value, list)
                    assert isinstance(dst_value, list)

                    # 異なるものがあった場合だけ後ろに付け足す
                    src_item[key] = list(OrderedDict.fromkeys(src_value + dst_value))

    return json.dumps(src_json, ensure_ascii=False)


def merge_update_infos(src_path: Path, dst_path: Path, output_path: Path) -> None:
    src = src_path.read_text(encoding="utf-8")
    dst = dst_path.read_text(encoding="utf-8")
    merged = merge_json_string(src, dst)
    output_path.write_text(merged, encoding="utf-8")


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("src_path", type=Path)
    parser.add_argument("dst_path", type=Path)
    parser.add_argument("output_path", type=Path)
    args = parser.parse_args()
    merge_update_infos(args.src_path, args.dst_path, args.output_path)