import json
from typing import List

from mev_inspect.models.classified_traces import ClassifiedTraceModel
from mev_inspect.schemas.classified_traces import ClassifiedTrace


def delete_classified_traces_for_block(
    db_session, block_number: int,
) -> None:
    (db_session.query(ClassifiedTraceModel)
       .filter(ClassifiedTraceModel.block_number == block_number)
       .delete()
    )

    db_session.commit()


def write_classified_traces(
    db_session,
    classified_traces: List[ClassifiedTrace],
) -> None:
    models = [
        ClassifiedTraceModel(**json.loads(trace.json())) for trace in classified_traces
    ]

    db_session.bulk_save_objects(models)
    db_session.commit()