split rewards/withdrawal

pixelsoup42 · Sep 16, 2023 · b45e750 · b45e750
1 parent c501baf
commit b45e750
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/cardano_account_pandas_dumper/cardano_account_pandas_dumper.py b/src/cardano_account_pandas_dumper/cardano_account_pandas_dumper.py
@@ -350,17 +350,21 @@ def _transaction_balance(self, transaction: blockfrost.utils.Namespace) -> Any:
         result[
             self._asset_tuple(self.data.LOVELACE_ASSET) + ("", " deposit", True)
         ] = np.longlong(transaction.deposit)
-        result[self._asset_tuple(self.data.LOVELACE_ASSET) + ("", " rewards", True)] = (
-            np.negative(
+        if transaction.reward_amount:
+            result[
+                self._asset_tuple(self.data.LOVELACE_ASSET) + ("", " rewards", True)
+            ] = np.longlong(transaction.reward_amount)
+        if transaction.withdrawals:
+            result[
+                self._asset_tuple(self.data.LOVELACE_ASSET)
+                + ("", " rewards withdrawal", True)
+            ] = np.negative(
                 functools.reduce(
                     np.add,
                     [np.longlong(w.amount) for w in transaction.withdrawals],
                     np.longlong(0),
                 )
             )
-            if not transaction.reward_amount
-            else np.longlong(transaction.reward_amount)
-        )
         for utxo in transaction.utxos.nonref_inputs:
             if not utxo.collateral or not transaction.valid_contract:
                 for amount in utxo.amount: