@@ -26,37 +26,42 @@ def get_address_reward(
2626
2727 The `epoch_from` and `epoch_to` are epochs where the reward can be spent.
2828 """
29- rewards = []
30- for db_row in dbsync_queries .query_address_reward (
31- address = address , epoch_from = epoch_from , epoch_to = epoch_to
32- ):
33- rewards .append (
34- dbsync_types .RewardEpochRecord (
35- amount = int (db_row .amount ),
36- earned_epoch = db_row .earned_epoch ,
37- spendable_epoch = db_row .spendable_epoch ,
38- type = db_row .type ,
39- pool_id = db_row .pool_id or "" ,
40- )
29+ rewards : list [dbsync_types .RewardEpochRecord ] = []
30+ rewards .extend (
31+ dbsync_types .RewardEpochRecord (
32+ amount = int (db_row .amount ),
33+ earned_epoch = db_row .earned_epoch ,
34+ spendable_epoch = db_row .spendable_epoch ,
35+ type = db_row .type ,
36+ pool_id = db_row .pool_id or "" ,
37+ )
38+ for db_row in dbsync_queries .query_address_reward (
39+ address = address , epoch_from = epoch_from , epoch_to = epoch_to
4140 )
41+ )
4242
43- for db_row in dbsync_queries .query_address_reward_rest (
44- address = address , epoch_from = epoch_from , epoch_to = epoch_to
45- ):
46- rewards .append (
47- dbsync_types .RewardEpochRecord (
48- amount = int (db_row .amount ),
49- earned_epoch = db_row .earned_epoch ,
50- spendable_epoch = db_row .spendable_epoch ,
51- type = db_row .type ,
52- pool_id = db_row .pool_id or "" ,
53- )
43+ reward_rest = list (
44+ dbsync_queries .query_address_reward_rest (
45+ address = address , epoch_from = epoch_from , epoch_to = epoch_to
5446 )
47+ )
48+ rewards .extend (
49+ dbsync_types .RewardEpochRecord (
50+ amount = int (db_row .amount ),
51+ earned_epoch = db_row .earned_epoch ,
52+ spendable_epoch = db_row .spendable_epoch ,
53+ type = db_row .type ,
54+ pool_id = db_row .pool_id or "" ,
55+ )
56+ for db_row in reward_rest
57+ )
5558 if not rewards :
5659 return dbsync_types .RewardRecord (address = address , reward_sum = 0 , rewards = [])
5760
5861 reward_sum = functools .reduce (lambda x , y : x + y .amount , rewards , 0 )
59- return dbsync_types .RewardRecord (address = db_row .address , reward_sum = reward_sum , rewards = rewards )
62+ return dbsync_types .RewardRecord (
63+ address = reward_rest [- 1 ].address , reward_sum = reward_sum , rewards = rewards
64+ )
6065
6166
6267def check_address_reward (
@@ -796,7 +801,7 @@ def check_plutus_costs(
796801 for db_record , cost_record in zip (sorted_db , sorted_costs ):
797802 try :
798803 check_plutus_cost (redeemer_record = db_record , cost_record = cost_record )
799- except AssertionError as err :
804+ except AssertionError as err : # noqa: PERF203
800805 errors .append (f"{ db_record .script_hash } :\n { err } " )
801806
802807 if errors :
0 commit comments