is_finalized property

dipdup-io · Oct 27, 2024 · d8eb4be · d8eb4be
1 parent 3001086
commit d8eb4be
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 10 deletions.
diff --git a/src/demo_substrate_events/handlers/on_transfer.py b/src/demo_substrate_events/handlers/on_transfer.py
@@ -47,8 +47,9 @@ async def on_transfer(
     if not amount:
         return
 
-    await sql_update(ctx, event.payload['from'], -amount, event.data.level)
-    await sql_update(ctx, event.payload['to'], amount, event.data.level)
-
-    # await orm_update(ctx, event.payload['from'], -amount, event.data.level)
-    # await orm_update(ctx, event.payload['to'], amount, event.data.level)
+    if ctx.is_finalized:
+        await sql_update(ctx, event.payload['from'], -amount, event.data.level)
+        await sql_update(ctx, event.payload['to'], amount, event.data.level)
+    else:
+        await orm_update(ctx, event.payload['from'], -amount, event.data.level)
+        await orm_update(ctx, event.payload['to'], amount, event.data.level)
diff --git a/src/dipdup/context.py b/src/dipdup/context.py
@@ -878,3 +878,8 @@ def _wrap(
         )
         ctx._link(new_ctx)
         return new_ctx
+
+    @property
+    def is_finalized(self) -> bool:
+        # FIXME: check the datasource
+        return 1
diff --git a/src/dipdup/projects/demo_substrate_events/handlers/on_transfer.py.j2 b/src/dipdup/projects/demo_substrate_events/handlers/on_transfer.py.j2
@@ -47,8 +47,9 @@ async def on_transfer(
     if not amount:
         return
 
-    await sql_update(ctx, event.payload['from'], -amount, event.data.level)
-    await sql_update(ctx, event.payload['to'], amount, event.data.level)
-
-    # await orm_update(ctx, event.payload['from'], -amount, event.data.level)
-    # await orm_update(ctx, event.payload['to'], amount, event.data.level)
+    if ctx.is_finalized:
+        await sql_update(ctx, event.payload['from'], -amount, event.data.level)
+        await sql_update(ctx, event.payload['to'], amount, event.data.level)
+    else:
+        await orm_update(ctx, event.payload['from'], -amount, event.data.level)
+        await orm_update(ctx, event.payload['to'], amount, event.data.level)