Hic Et Nunc demo (#11)

Author: droserasprout

README.md

@@ -18,6 +18,12 @@ $ pip install dipdup
 
 ## Creating indexer
 
+If you want to see dipdup in action before diving into details you can run a demo project and use it as reference:
+
+```shell
+$ dipdup -c src/demo_hic_et_nunc/dipdup.yml run
+```
+
 ### Write configuration file
 
 Create a new YAML file and adapt the following example to your needs:

src/demo_hic_et_nunc/handlers/on_cancel_swap.py

@@ -1,4 +1,4 @@
-from demo_hic_et_nunc.models import *
+import demo_hic_et_nunc.models as models
 from demo_hic_et_nunc.types.hen_minter.parameter.cancel_swap import CancelSwap
 from dipdup.models import HandlerContext, OperationContext
 
@@ -7,4 +7,6 @@ async def on_cancel_swap(
     ctx: HandlerContext,
     cancel_swap: OperationContext[CancelSwap],
 ) -> None:
-    ...
+    swap = await models.Swap.filter(id=int(cancel_swap.parameter.__root__)).get()
+    swap.status = models.SwapStatus.CANCELED
+    await swap.save()

src/demo_hic_et_nunc/handlers/on_collect.py

@@ -1,4 +1,4 @@
-from demo_hic_et_nunc.models import *
+import demo_hic_et_nunc.models as models
 from demo_hic_et_nunc.types.hen_minter.parameter.collect import Collect
 from dipdup.models import HandlerContext, OperationContext
 
@@ -7,4 +7,21 @@ async def on_collect(
     ctx: HandlerContext,
     collect: OperationContext[Collect],
 ) -> None:
-    ...
+    swap = await models.Swap.filter(id=collect.parameter.swap_id).get()
+    seller = await swap.creator
+    buyer, _ = await models.Holder.get_or_create(address=collect.data.sender_address)
+
+    trade = models.Trade(
+        swap=swap,
+        seller=seller,
+        buyer=buyer,
+        amount=int(collect.parameter.objkt_amount),
+        level=collect.data.level,
+        timestamp=collect.data.timestamp,
+    )
+    await trade.save()
+
+    swap.amount_left -= int(collect.parameter.objkt_amount)  # type: ignore
+    if swap.amount_left == 0:
+        swap.status = models.SwapStatus.FINISHED
+    await swap.save()

src/demo_hic_et_nunc/handlers/on_mint.py

@@ -1,4 +1,4 @@
-from demo_hic_et_nunc.models import *
+from demo_hic_et_nunc.models import Holder, Token
 from demo_hic_et_nunc.types.hen_minter.parameter.mint_objkt import MintOBJKT
 from demo_hic_et_nunc.types.hen_objkts.parameter.mint import Mint
 from dipdup.models import HandlerContext, OperationContext
@@ -9,4 +9,12 @@ async def on_mint(
     mint_objkt: OperationContext[MintOBJKT],
     mint: OperationContext[Mint],
 ) -> None:
-    ...
+    holder, _ = await Holder.get_or_create(address=mint.parameter.address)
+    token = Token(
+        id=mint.parameter.token_id,
+        creator=holder,
+        supply=mint.parameter.amount,
+        level=mint.data.level,
+        timestamp=mint.data.timestamp,
+    )
+    await token.save()

src/demo_hic_et_nunc/handlers/on_swap.py

@@ -1,4 +1,4 @@
-from demo_hic_et_nunc.models import Swap as SwapModel
+import demo_hic_et_nunc.models as models
 from demo_hic_et_nunc.types.hen_minter.parameter.swap import Swap
 from dipdup.models import HandlerContext, OperationContext
 
@@ -7,4 +7,15 @@ async def on_swap(
     ctx: HandlerContext,
     swap: OperationContext[Swap],
 ) -> None:
-    ...
+    holder, _ = await models.Holder.get_or_create(address=swap.data.sender_address)
+    swap_model = models.Swap(
+        id=int(swap.storage.swap_id) - 1,  # type: ignore
+        creator=holder,
+        price=swap.parameter.xtz_per_objkt,
+        amount=swap.parameter.objkt_amount,
+        amount_left=swap.parameter.objkt_amount,
+        status=models.SwapStatus.ACTIVE,
+        level=swap.data.level,
+        timestamp=swap.data.timestamp,
+    )
+    await swap_model.save()

src/demo_hic_et_nunc/models.py

@@ -1,5 +1,16 @@
+from enum import Enum, IntEnum
+
 from tortoise import Model, fields
 
+# on mint token, holder
+# on_swap new, cancel_swap, collect
+
+
+class SwapStatus(IntEnum):
+    ACTIVE = 0
+    FINISHED = 1
+    CANCELED = 2
+
 
 class Holder(Model):
     address = fields.CharField(58, pk=True)
@@ -16,9 +27,11 @@ class Token(Model):
 class Swap(Model):
     id = fields.BigIntField(pk=True)
     creator = fields.ForeignKeyField('models.Holder', 'swaps')
+    price = fields.IntField()
     amount = fields.IntField()
     amount_left = fields.IntField()
     level = fields.BigIntField()
+    status = fields.IntEnumField(SwapStatus)
     timestamp = fields.DatetimeField()
 
 

src/demo_quipuswap/handlers/on_fa12_divest_liquidity.py

@@ -1,4 +1,4 @@
-from demo_quipuswap.models import *
+import demo_quipuswap.models as models
 from demo_quipuswap.types.quipu_fa12.parameter.divest_liquidity import DivestLiquidity
 from dipdup.models import HandlerContext, OperationContext
 

src/demo_quipuswap/handlers/on_fa12_invest_liquidity.py

@@ -1,4 +1,4 @@
-from demo_quipuswap.models import *
+import demo_quipuswap.models as models
 from demo_quipuswap.types.quipu_fa12.parameter.invest_liquidity import InvestLiquidity
 from dipdup.models import HandlerContext, OperationContext
 

src/demo_quipuswap/handlers/on_fa12_tez_to_token.py

@@ -1,4 +1,4 @@
-from demo_quipuswap.models import *
+import demo_quipuswap.models as models
 from demo_quipuswap.types.fa12_token.parameter.transfer import Transfer
 from demo_quipuswap.types.quipu_fa12.parameter.tez_to_token_payment import TezToTokenPayment
 from dipdup.models import HandlerContext, OperationContext

src/demo_quipuswap/handlers/on_fa12_token_to_tez.py

@@ -1,4 +1,4 @@
-from demo_quipuswap.models import *
+import demo_quipuswap.models as models
 from demo_quipuswap.types.fa12_token.parameter.transfer import Transfer
 from demo_quipuswap.types.quipu_fa12.parameter.token_to_tez_payment import TokenToTezPayment
 from dipdup.models import HandlerContext, OperationContext