ae/chainmapper
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: same export filepath for dockerized & non-dockerized deployments

Browse Source
This commit is contained in:
17ms 2024-07-07 15:59:38 +03:00
parent b0f5ab1b83
commit 13f453768c
Signed by untrusted user who does not match committer: ae
GPG Key ID: 995EFD5C1B532B3E
2 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/const.py
View File

@ -2,3 +2,5 @@ import json
WS_ADDR = "wss://ws.blockchain.info/coins" WS_ADDR = "wss://ws.blockchain.info/coins"
SUB_MSG = json.dumps({"coin": "eth", "command": "subscribe", "entity": "confirmed_transaction"}) SUB_MSG = json.dumps({"coin": "eth", "command": "subscribe", "entity": "confirmed_transaction"})
DEFAULT_EXPORT_PATH = "./data/export.json"

14
src/db.py
View File

@ -3,6 +3,8 @@ import json
import logging import logging
import asyncio import asyncio
from src.const import DEFAULT_EXPORT_PATH
class Handler: class Handler:
"""Handle all SQLite connections required to create, update, and export the stored addresses.""" """Handle all SQLite connections required to create, update, and export the stored addresses."""
@ -46,11 +48,11 @@ class Handler:
) )
self.con.commit() self.con.commit()
async def export(self, filename="export.json"): async def export(self, filepath=DEFAULT_EXPORT_PATH):
"""Export the addresses from the SQLite database in descending order based on the transaction counts.""" """Export the addresses from the SQLite database in descending order based on the transaction counts."""
await asyncio.to_thread(self._export, filename) await asyncio.to_thread(self._export, filepath)
def _export(self, filename="export.json"): def _export(self, filepath):
self.cursor.execute( self.cursor.execute(
""" """
SELECT address, total_tx_count SELECT address, total_tx_count
@ -62,12 +64,12 @@ class Handler:
data = [{"address": record[0], "tx_count": record[1]} for record in records] data = [{"address": record[0], "tx_count": record[1]} for record in records]
data_json = json.dumps(data, indent=4) data_json = json.dumps(data, indent=4)
logging.info("Exporting the database's current state to '%s' (overwriting if an old copy exists)...", filename) logging.info("Exporting the database's current state to '%s' (overwriting if an old copy exists)...", filepath)
with open(filename, "w", encoding="utf-8") as f: with open(filepath, "w", encoding="utf-8") as f:
f.write(data_json) f.write(data_json)
logging.info("Data exported to '%s'", filename) logging.info("Data exported to '%s'", filepath)
def periodic_export(loop, handler, interval, shutdown_event): def periodic_export(loop, handler, interval, shutdown_event):