12345678910111213141516171819202122232425262728293031 |
- from cassandra.cluster import Cluster
- from cassandra.query import BoundStatement, BatchStatement
- import csv
-
-
- def db_insert_csv_txs(config, tx_file):
- print(" == DB TX INSERTION SCRIPT == ")
-
- print(
- f"Attempting Cassandra connection @ {config['cassandra_addresses']}:{config['cassandra_port']}")
- cluster = Cluster(config['cassandra_addresses'],
- port=config['cassandra_port'])
- session = cluster.connect(config['cassandra_keyspace'])
- print(f"Connection OK")
-
- with open(tx_file, newline='') as tx_csv:
- rowreader = csv.reader(tx_csv, dialect="excel")
- next(rowreader)
-
- statement = session.prepare(
- f"INSERT INTO {config['tx_table_name']} (tx_id,address,value,tx_hash,block_id,timestamp) VALUES(?,?,?,?,?,?);")
- boundStatement = BoundStatement(statement)
- batchStatement = BatchStatement()
-
- for row in rowreader:
- batchStatement.add(boundStatement.bind(
- [int(row[0]), str(row[1]), int(row[2]), str(row[3]), int(row[4]), int(row[5])]))
- session.execute(batchStatement)
-
- print("Done!")
- cluster.shutdown()
|