Add timestamp column to csv output
pull/2694/head
KPrasch 2021-05-07 00:50:02 -07:00 committed by GitHub
commit 7e9cbcaac8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 8 additions and 5 deletions

View File

@ -0,0 +1 @@
Added timestamp and date columns to csv output of "nucypher status events" command.

View File

@ -34,7 +34,7 @@ class EventRecord:
except BlockchainInterfaceFactory.NoRegisteredInterfaces:
self.timestamp = None
else:
self.timestamp = blockchain.client.w3.eth.getBlock(self.block_number)['timestamp'],
self.timestamp = blockchain.client.w3.eth.getBlock(self.block_number)['timestamp']
def __repr__(self):
pairs_to_show = dict(self.args.items())

View File

@ -52,6 +52,8 @@ def write_events_to_csv_file(csv_file: str,
event_row = OrderedDict()
event_row['event_name'] = event_name
event_row['block_number'] = event_record.block_number
event_row['unix_timestamp'] = event_record.timestamp
event_row['date'] = maya.MayaDT(event_record.timestamp).iso8601()
event_row.update(dict(event_record.args.items()))
if events_writer is None:
events_writer = csv.DictWriter(events_file, fieldnames=event_row.keys())

View File

@ -205,13 +205,13 @@ def test_nucypher_status_events(click_runner, testerchain, agency_local_registry
line_count = 0
for row in csv_reader:
if line_count == 0:
assert ",".join(row) == 'event_name,block_number,staker,period,value' # specific to CommitmentMade
assert ",".join(row) == 'event_name,block_number,unix_timestamp,date,staker,period,value' # specific to CommitmentMade
else:
row_data = f'{row}'
assert row[0] == 'CommitmentMade', row_data
# skip block number
assert row[2] == first_staker.checksum_address, row_data
assert row[3] == f'{committed_period}', row_data
# skip block_number, unix_timestamp, date
assert row[4] == first_staker.checksum_address, row_data
assert row[5] == f'{committed_period}', row_data
# skip value
line_count += 1
assert line_count == 2, 'column names and single event row in csv file'