from __future__ import annotations

from datetime import datetime

from falkordb import FalkorDB


GRAPH_NAME = "travel_agency_2_0_test"
UPDATED_AT = datetime.now().strftime("%Y-%m-%d %H:%M:%S")


def main() -> dict[str, int | str]:
    graph = FalkorDB(host="localhost", port=6380).select_graph(GRAPH_NAME)
    node_rows = graph.query(
        """
        MATCH (n)
        WHERE n.updated_at IS NULL
        SET n.updated_at = $updated_at,
            n.timestamp_source = 'backfilled_missing_updated_at',
            n.timestamp_note = '原始导入未写入更新时间，本次按修正执行时间补齐'
        RETURN count(n)
        """,
        {"updated_at": UPDATED_AT},
    ).result_set
    relation_rows = graph.query(
        """
        MATCH ()-[r]->()
        WHERE r.updated_at IS NULL
        SET r.updated_at = $updated_at,
            r.timestamp_source = 'backfilled_missing_updated_at',
            r.timestamp_note = '原始导入未写入更新时间，本次按修正执行时间补齐'
        RETURN count(r)
        """,
        {"updated_at": UPDATED_AT},
    ).result_set
    summary = {
        "graph": GRAPH_NAME,
        "updated_at": UPDATED_AT,
        "nodes_backfilled": int(node_rows[0][0] if node_rows else 0),
        "relations_backfilled": int(relation_rows[0][0] if relation_rows else 0),
    }
    print(summary)
    return summary


if __name__ == "__main__":
    main()