58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
import json
|
|
import logging
|
|
import os
|
|
from typing import Optional
|
|
|
|
import boto3
|
|
|
|
from mev_inspect.text_io import BytesIteratorIO
|
|
|
|
AWS_ENDPOINT_URL_ENV = "AWS_ENDPOINT_URL"
|
|
MEV_SUMMARY_EXPORT_QUERY = """
|
|
SELECT to_json(mev_summary)
|
|
FROM mev_summary
|
|
WHERE
|
|
block_number >= :after_block_number AND
|
|
block_number < :before_block_number
|
|
"""
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def export_block_range(
|
|
inspect_db_session, after_block_number: int, before_block_number
|
|
) -> None:
|
|
export_bucket_name = os.environ["EXPORT_BUCKET_NAME"]
|
|
client = get_s3_client()
|
|
|
|
mev_summary_json_results = inspect_db_session.execute(
|
|
statement=MEV_SUMMARY_EXPORT_QUERY,
|
|
params={
|
|
"after_block_number": after_block_number,
|
|
"before_block_number": before_block_number,
|
|
},
|
|
)
|
|
|
|
mev_summary_json_fileobj = BytesIteratorIO(
|
|
(f"{json.dumps(row)}\n".encode("utf-8") for (row,) in mev_summary_json_results)
|
|
)
|
|
|
|
key = f"mev_summary/{after_block_number}-{before_block_number}.json"
|
|
|
|
client.upload_fileobj(
|
|
mev_summary_json_fileobj,
|
|
Bucket=export_bucket_name,
|
|
Key=key,
|
|
)
|
|
|
|
logger.info(f"Exported to {key}")
|
|
|
|
|
|
def get_s3_client():
|
|
endpoint_url = get_endpoint_url()
|
|
return boto3.client("s3", endpoint_url=endpoint_url)
|
|
|
|
|
|
def get_endpoint_url() -> Optional[str]:
|
|
return os.environ.get(AWS_ENDPOINT_URL_ENV)
|