Dynamic configuration, context and rollback improvements, bugfixes

docker-compose.yml

@@ -1,6 +1,7 @@
 version: "3.8"
 
 services:
+
   hic_et_nunc:
     build: .
     depends_on:

pyproject.toml

@@ -32,6 +32,7 @@ pydantic = "^1.8.1"
 aiosignalrcore = "^0.9.2"
 fcache = "^0.4.7"
 click = "^8.0.1"
+pyee = "^8.1.0"
 
 [tool.poetry.dev-dependencies]
 black = "^20.8b1"

src/demo_hic_et_nunc/handlers/on_cancel_swap.py

@@ -1,12 +1,13 @@
 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.models import OperationHandlerContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Transaction
 
 
 async def on_cancel_swap(
     ctx: OperationHandlerContext,
-    cancel_swap: TransactionContext[CancelSwapParameter, HenMinterStorage],
+    cancel_swap: Transaction[CancelSwapParameter, HenMinterStorage],
 ) -> None:
     swap = await models.Swap.filter(id=int(cancel_swap.parameter.__root__)).get()
     swap.status = models.SwapStatus.CANCELED

src/demo_hic_et_nunc/handlers/on_collect.py

@@ -1,12 +1,13 @@
 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.models import OperationHandlerContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Transaction
 
 
 async def on_collect(
     ctx: OperationHandlerContext,
-    collect: TransactionContext[CollectParameter, HenMinterStorage],
+    collect: Transaction[CollectParameter, HenMinterStorage],
 ) -> None:
     swap = await models.Swap.filter(id=collect.parameter.swap_id).get()
     seller = await swap.creator

src/demo_hic_et_nunc/handlers/on_mint.py

@@ -3,13 +3,14 @@
 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.models import OperationHandlerContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Transaction
 
 
 async def on_mint(
     ctx: OperationHandlerContext,
-    mint_objkt: TransactionContext[MintOBJKTParameter, HenMinterStorage],
-    mint: TransactionContext[MintParameter, HenObjktsStorage],
+    mint_objkt: Transaction[MintOBJKTParameter, HenMinterStorage],
+    mint: Transaction[MintParameter, HenObjktsStorage],
 ) -> None:
     holder, _ = await models.Holder.get_or_create(address=mint.parameter.address)
     token = models.Token(

src/demo_hic_et_nunc/handlers/on_swap.py

@@ -1,12 +1,13 @@
 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.models import OperationHandlerContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Transaction
 
 
 async def on_swap(
     ctx: OperationHandlerContext,
-    swap: TransactionContext[SwapParameter, HenMinterStorage],
+    swap: Transaction[SwapParameter, HenMinterStorage],
 ) -> None:
     holder, _ = await models.Holder.get_or_create(address=swap.data.sender_address)
     swap_model = models.Swap(

src/demo_hic_et_nunc/models.py

@@ -1,10 +1,7 @@
-from enum import Enum, IntEnum
+from enum import IntEnum
 
 from tortoise import Model, fields
 
-# on mint token, holder
-# on_swap new, cancel_swap, collect
-
 
 class SwapStatus(IntEnum):
     ACTIVE = 0

src/demo_quipuswap/handlers/on_fa12_divest_liquidity.py

@@ -5,13 +5,14 @@
 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.models import OperationHandlerContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Transaction
 
 
 async def on_fa12_divest_liquidity(
     ctx: OperationHandlerContext,
-    divest_liquidity: TransactionContext[DivestLiquidityParameter, QuipuFa12Storage],
-    transfer: TransactionContext[TransferParameter, Fa12TokenStorage],
+    divest_liquidity: Transaction[DivestLiquidityParameter, QuipuFa12Storage],
+    transfer: Transaction[TransferParameter, Fa12TokenStorage],
 ) -> None:
     if ctx.template_values is None:
         raise Exception('This index must be templated')
@@ -32,12 +33,12 @@ async def on_fa12_divest_liquidity(
 
     tez_pool = Decimal(storage.storage.tez_pool) / (10 ** 6)
     token_pool = Decimal(storage.storage.token_pool) / (10 ** decimals)
-    if tez_pool and token_pool:
-        price = tez_pool / token_pool
-    else:
-        last_trade = await models.Trade.filter(symbol=symbol).order_by('-id').first()
-        assert last_trade
-        price = last_trade.price
+
+    # NOTE: Empty pools mean exchange is not initialized yet
+    if not tez_pool and not token_pool:
+        return
+
+    price = tez_pool / token_pool
     share_px = (tez_qty + price * token_qty) / shares_qty
 
     position.realized_pl += shares_qty * (share_px - position.avg_share_px)

src/demo_quipuswap/handlers/on_fa12_invest_liquidity.py

@@ -5,13 +5,14 @@
 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.models import OperationHandlerContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Transaction
 
 
 async def on_fa12_invest_liquidity(
     ctx: OperationHandlerContext,
-    invest_liquidity: TransactionContext[InvestLiquidityParameter, QuipuFa12Storage],
-    transfer: TransactionContext[TransferParameter, Fa12TokenStorage],
+    invest_liquidity: Transaction[InvestLiquidityParameter, QuipuFa12Storage],
+    transfer: Transaction[TransferParameter, Fa12TokenStorage],
 ) -> None:
     if ctx.template_values is None:
         raise Exception('This index must be templated')

src/demo_quipuswap/handlers/on_fa12_origination.py

@@ -1,13 +1,12 @@
-from typing import Optional
-
 import demo_quipuswap.models as models
 from demo_quipuswap.types.quipu_fa12.storage import QuipuFa12Storage
-from dipdup.models import OperationData, OperationHandlerContext, OriginationContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Origination
 
 
 async def on_fa12_origination(
     ctx: OperationHandlerContext,
-    quipu_fa12_origination: OriginationContext[QuipuFa12Storage],
+    quipu_fa12_origination: Origination[QuipuFa12Storage],
 ) -> None:
     if ctx.template_values is None:
         raise Exception('This index must be templated')

src/demo_quipuswap/handlers/on_fa12_tez_to_token.py

@@ -5,13 +5,14 @@
 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.models import OperationHandlerContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Transaction
 
 
 async def on_fa12_tez_to_token(
     ctx: OperationHandlerContext,
-    tez_to_token_payment: TransactionContext[TezToTokenPaymentParameter, QuipuFa12Storage],
-    transfer: TransactionContext[TransferParameter, Fa12TokenStorage],
+    tez_to_token_payment: Transaction[TezToTokenPaymentParameter, QuipuFa12Storage],
+    transfer: Transaction[TransferParameter, Fa12TokenStorage],
 ) -> None:
     if ctx.template_values is None:
         raise Exception('This index must be templated')

src/demo_quipuswap/handlers/on_fa12_token_to_tez.py

@@ -5,13 +5,14 @@
 from demo_quipuswap.types.fa12_token.storage import Fa12TokenStorage
 from demo_quipuswap.types.quipu_fa12.parameter.token_to_tez_payment import TokenToTezPaymentParameter
 from demo_quipuswap.types.quipu_fa12.storage import QuipuFa12Storage
-from dipdup.models import OperationHandlerContext, TransactionContext
+from dipdup.context import OperationHandlerContext
+from dipdup.models import Transaction
 
 
 async def on_fa12_token_to_tez(
     ctx: OperationHandlerContext,
-    token_to_tez_payment: TransactionContext[TokenToTezPaymentParameter, QuipuFa12Storage],
-    transfer: TransactionContext[TransferParameter, Fa12TokenStorage],
+    token_to_tez_payment: Transaction[TokenToTezPaymentParameter, QuipuFa12Storage],
+    transfer: Transaction[TransferParameter, Fa12TokenStorage],
 ) -> None:
     if ctx.template_values is None:
         raise Exception('This index must be templated')