BigMapIndex handlers get single diff at once, logging improvements (#60)

Author: droserasprout

src/demo_hic_et_nunc/handlers/on_cancel_swap.py

@@ -1,12 +1,12 @@
 import demo_hic_et_nunc.models as models
 from demo_hic_et_nunc.types.hen_minter.parameter.cancel_swap import CancelSwapParameter
 from demo_hic_et_nunc.types.hen_minter.storage import HenMinterStorage
-from dipdup.context import OperationHandlerContext
+from dipdup.context import HandlerContext
 from dipdup.models import Transaction
 
 
 async def on_cancel_swap(
-    ctx: OperationHandlerContext,
+    ctx: HandlerContext,
     cancel_swap: Transaction[CancelSwapParameter, HenMinterStorage],
 ) -> None:
     swap = await models.Swap.filter(id=int(cancel_swap.parameter.__root__)).get()

src/demo_hic_et_nunc/handlers/on_collect.py

@@ -1,12 +1,12 @@
 import demo_hic_et_nunc.models as models
 from demo_hic_et_nunc.types.hen_minter.parameter.collect import CollectParameter
 from demo_hic_et_nunc.types.hen_minter.storage import HenMinterStorage
-from dipdup.context import OperationHandlerContext
+from dipdup.context import HandlerContext
 from dipdup.models import Transaction
 
 
 async def on_collect(
-    ctx: OperationHandlerContext,
+    ctx: HandlerContext,
     collect: Transaction[CollectParameter, HenMinterStorage],
 ) -> None:
     swap = await models.Swap.filter(id=collect.parameter.swap_id).get()

src/demo_hic_et_nunc/handlers/on_mint.py

@@ -3,12 +3,12 @@
 from demo_hic_et_nunc.types.hen_minter.storage import HenMinterStorage
 from demo_hic_et_nunc.types.hen_objkts.parameter.mint import MintParameter
 from demo_hic_et_nunc.types.hen_objkts.storage import HenObjktsStorage
-from dipdup.context import OperationHandlerContext
+from dipdup.context import HandlerContext
 from dipdup.models import Transaction
 
 
 async def on_mint(
-    ctx: OperationHandlerContext,
+    ctx: HandlerContext,
     mint_objkt: Transaction[MintOBJKTParameter, HenMinterStorage],
     mint: Transaction[MintParameter, HenObjktsStorage],
 ) -> None:

src/demo_hic_et_nunc/handlers/on_rollback.py

@@ -1,10 +1,6 @@
-import logging
-
 from dipdup.context import RollbackHandlerContext
 
-_logger = logging.getLogger(__name__)
-
 
 async def on_rollback(ctx: RollbackHandlerContext) -> None:
-    _logger.warning('Datasource `%s` rolled back from level %s to level %s, reindexing', ctx.datasource, ctx.from_level, ctx.to_level)
+    ctx.logger.warning('Datasource `%s` rolled back from level %s to level %s, reindexing', ctx.datasource, ctx.from_level, ctx.to_level)
     await ctx.reindex()

src/demo_hic_et_nunc/handlers/on_swap.py

@@ -1,12 +1,12 @@
 import demo_hic_et_nunc.models as models
 from demo_hic_et_nunc.types.hen_minter.parameter.swap import SwapParameter
 from demo_hic_et_nunc.types.hen_minter.storage import HenMinterStorage
-from dipdup.context import OperationHandlerContext
-from dipdup.models import Transaction
+from dipdup.context import HandlerContext, RollbackHandlerContext
+from dipdup.models import BigMapAction, BigMapData, BigMapDiff, OperationData, Origination, Transaction
 
 
 async def on_swap(
-    ctx: OperationHandlerContext,
+    ctx: HandlerContext,
     swap: Transaction[SwapParameter, HenMinterStorage],
 ) -> None:
     holder, _ = await models.Holder.get_or_create(address=swap.data.sender_address)

src/demo_quipuswap/dipdup.yml

@@ -46,6 +46,8 @@ templates:
           - type: transaction
             destination: <token_contract>
             entrypoint: transfer
+          - type: transaction
+            source: <dex_contract>
       - callback: on_fa12_tez_to_token
         pattern:
           - type: transaction
@@ -75,6 +77,8 @@ templates:
           - type: transaction
             destination: <token_contract>
             entrypoint: transfer
+          - type: transaction
+            source: <dex_contract>
       - callback: on_fa12_withdraw_profit
         pattern:
           - type: transaction
@@ -105,6 +109,8 @@ templates:
           - type: transaction
             destination: <token_contract>
             entrypoint: transfer
+          - type: transaction
+            source: <dex_contract>
       - callback: on_fa2_tez_to_token
         pattern:
           - type: transaction
@@ -134,6 +140,8 @@ templates:
           - type: transaction
             destination: <token_contract>
             entrypoint: transfer
+          - type: transaction
+            source: <dex_contract>
       - callback: on_fa2_withdraw_profit
         pattern:
           - type: transaction

src/demo_quipuswap/handlers/on_fa12_divest_liquidity.py

@@ -5,14 +5,15 @@
 from demo_quipuswap.types.fa12_token.storage import Fa12TokenStorage
 from demo_quipuswap.types.quipu_fa12.parameter.divest_liquidity import DivestLiquidityParameter
 from demo_quipuswap.types.quipu_fa12.storage import QuipuFa12Storage
-from dipdup.context import OperationHandlerContext
-from dipdup.models import Transaction
+from dipdup.context import HandlerContext
+from dipdup.models import OperationData, Transaction
 
 
 async def on_fa12_divest_liquidity(
-    ctx: OperationHandlerContext,
+    ctx: HandlerContext,
     divest_liquidity: Transaction[DivestLiquidityParameter, QuipuFa12Storage],
     transfer: Transaction[TransferParameter, Fa12TokenStorage],
+    transaction_1: OperationData,
 ) -> None:
     if ctx.template_values is None:
         raise Exception('This index must be templated')
@@ -24,10 +25,9 @@ async def on_fa12_divest_liquidity(
     trader = divest_liquidity.data.sender_address
 
     position, _ = await models.Position.get_or_create(trader=trader, symbol=symbol)
-    transaction = next(op for op in ctx.operations if op.amount)
 
-    assert transaction.amount is not None
-    tez_qty = Decimal(transaction.amount) / (10 ** 6)
+    assert transaction_1.amount is not None
+    tez_qty = Decimal(transaction_1.amount) / (10 ** 6)
     token_qty = Decimal(transfer.parameter.value) / (10 ** decimals)
     shares_qty = int(divest_liquidity.parameter.shares)
 

src/demo_quipuswap/handlers/on_fa12_invest_liquidity.py

@@ -5,12 +5,12 @@
 from demo_quipuswap.types.fa12_token.storage import Fa12TokenStorage
 from demo_quipuswap.types.quipu_fa12.parameter.invest_liquidity import InvestLiquidityParameter
 from demo_quipuswap.types.quipu_fa12.storage import QuipuFa12Storage
-from dipdup.context import OperationHandlerContext
+from dipdup.context import HandlerContext
 from dipdup.models import Transaction
 
 
 async def on_fa12_invest_liquidity(
-    ctx: OperationHandlerContext,
+    ctx: HandlerContext,
     invest_liquidity: Transaction[InvestLiquidityParameter, QuipuFa12Storage],
     transfer: Transaction[TransferParameter, Fa12TokenStorage],
 ) -> None:

src/demo_quipuswap/handlers/on_fa12_origination.py

@@ -1,11 +1,11 @@
 import demo_quipuswap.models as models
 from demo_quipuswap.types.quipu_fa12.storage import QuipuFa12Storage
-from dipdup.context import OperationHandlerContext
+from dipdup.context import HandlerContext
 from dipdup.models import Origination
 
 
 async def on_fa12_origination(
-    ctx: OperationHandlerContext,
+    ctx: HandlerContext,
     quipu_fa12_origination: Origination[QuipuFa12Storage],
 ) -> None:
     if ctx.template_values is None:

src/demo_quipuswap/handlers/on_fa12_tez_to_token.py

@@ -5,12 +5,12 @@
 from demo_quipuswap.types.fa12_token.storage import Fa12TokenStorage
 from demo_quipuswap.types.quipu_fa12.parameter.tez_to_token_payment import TezToTokenPaymentParameter
 from demo_quipuswap.types.quipu_fa12.storage import QuipuFa12Storage
-from dipdup.context import OperationHandlerContext
+from dipdup.context import HandlerContext
 from dipdup.models import Transaction
 
 
 async def on_fa12_tez_to_token(
-    ctx: OperationHandlerContext,
+    ctx: HandlerContext,
     tez_to_token_payment: Transaction[TezToTokenPaymentParameter, QuipuFa12Storage],
     transfer: Transaction[TransferParameter, Fa12TokenStorage],
 ) -> None: