31 lines
650 B
Python
31 lines
650 B
Python
import json
|
|
from typing import List
|
|
|
|
from mev_inspect.models.punk_snipes import PunkSnipeModel
|
|
from mev_inspect.schemas.punk_snipe import PunkSnipe
|
|
|
|
|
|
def delete_punk_snipes_for_block(
|
|
db_session,
|
|
block_number: int,
|
|
) -> None:
|
|
(
|
|
db_session.query(PunkSnipeModel)
|
|
.filter(PunkSnipeModel.block_number == block_number)
|
|
.delete()
|
|
)
|
|
|
|
db_session.commit()
|
|
|
|
|
|
def write_punk_snipes(
|
|
db_session,
|
|
punk_snipes: List[PunkSnipe],
|
|
) -> None:
|
|
models = [
|
|
PunkSnipeModel(**json.loads(punk_snipe.json())) for punk_snipe in punk_snipes
|
|
]
|
|
|
|
db_session.bulk_save_objects(models)
|
|
db_session.commit()
|