charli3_dendrite 1.4.6.dev0__tar.gz → 1.4.7.dev0__tar.gz

{charli3_dendrite-1.4.6.dev0 → charli3_dendrite-1.4.7.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: charli3_dendrite
-Version: 1.4.6.dev0
+Version: 1.4.7.dev0
 Summary: 
 License-File: LICENSE
 Author: Elder Millenial

{charli3_dendrite-1.4.6.dev0 → charli3_dendrite-1.4.7.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "charli3_dendrite"
-version = "1.4.6-dev0"
+version = "1.4.7-dev0"
 description = ""
 authors = ["Elder Millenial <eldermillenial@protonmail.com>"]
 readme = "README.md"
@@ -49,7 +49,7 @@ module = ["pycardano", "pycardano.*", "dotenv", "nox"]
 ignore_missing_imports = true
 
 [tool.bumpversion]
-current_version = "1.4.6-dev0"
+current_version = "1.4.7-dev0"
 parse = """(?x)
     (?P<major>\\d+)\\.
     (?P<minor>\\d+)\\.

{charli3_dendrite-1.4.6.dev0 → charli3_dendrite-1.4.7.dev0}/src/charli3_dendrite/__init__.py

@@ -1,5 +1,6 @@
 # noqa
 from charli3_dendrite.dexs.amm.cswap import CSwapCPPState
+from charli3_dendrite.dexs.amm.dano import DanoCLMMState
 from charli3_dendrite.dexs.amm.minswap import MinswapCPPState
 from charli3_dendrite.dexs.amm.minswap import MinswapDJEDiUSDStableState
 from charli3_dendrite.dexs.amm.minswap import MinswapDJEDUSDCStableState

{charli3_dendrite-1.4.6.dev0 → charli3_dendrite-1.4.7.dev0}/src/charli3_dendrite/backend/backend_base.py

@@ -99,6 +99,24 @@ class AbstractBackend(ABC):
         """
         pass
 
+    def get_stake_rewards(self, address: Address) -> int:
+        """Withdrawable staking-reward balance (lovelace) for a reward address.
+
+        Direct-spend DEXs whose pools stake their ADA (e.g. Dano) must withdraw
+        the pool's accrued rewards as part of a swap; the withdrawal amount must
+        equal the account's current withdrawable balance. Not every backend can
+        answer this — those raise ``NotImplementedError``.
+
+        Args:
+            address (Address): A stake/reward address (delegation part only).
+
+        Returns:
+            int: Withdrawable reward balance in lovelace (0 if none).
+        """
+        raise NotImplementedError(
+            f"{type(self).__name__} does not implement get_stake_rewards",
+        )
+
     @abstractmethod
     def get_historical_order_utxos(
         self,

{charli3_dendrite-1.4.6.dev0 → charli3_dendrite-1.4.7.dev0}/src/charli3_dendrite/backend/dbsync/__init__.py

@@ -5,6 +5,7 @@ import logging
 import os
 from datetime import datetime
 from threading import Lock
+from typing import Any
 
 import psycopg_pool  # type: ignore
 from dotenv import load_dotenv  # type: ignore
@@ -296,7 +297,7 @@ OFFSET %(offset)s"""
         tx_hash: str,
         addresses: list[str],
         assets: list[str] | None = None,
-    ) -> tuple[str, dict[str, list]]:
+    ) -> tuple[str, dict[str, Any]]:
         """Get transactions by policy or address."""
         # Use the pool selector to format the output
         datum_selector = PoolSelector.select()
@@ -399,7 +400,7 @@ AND ma.policy = ANY(%(policies)b) AND ma.name = ANY(%(names)b)"""
     def _get_pool_utxos_in_block(self, block_no: int) -> tuple[str, dict]:
         # Use this for gathering all assets for multiple addresses
         datum_selector = (
-            PoolSelector.select()
+            PoolSelector.select()  # noqa: S608
             + """
     FROM (
         SELECT txo.*, address.address, address.payment_cred
@@ -469,6 +470,39 @@ LIMIT 1"""
 
         return UTxOSelector.parse(r[0])
 
+    def _get_stake_rewards(self, address: Address) -> tuple[str, dict]:
+        # Withdrawable balance = spendable rewards (member/leader + treasury/
+        # reserves "rest") minus everything already withdrawn. spendable_epoch
+        # gates rewards by the ~2-epoch maturation delay.
+        query = """
+SELECT (
+    COALESCE((SELECT SUM(r.amount) FROM reward r
+              JOIN stake_address sa ON sa.id = r.addr_id
+              WHERE sa.hash_raw = %(raw)b
+                AND r.spendable_epoch <= (SELECT MAX(epoch_no) FROM block)), 0)
+  + COALESCE((SELECT SUM(rr.amount) FROM reward_rest rr
+              JOIN stake_address sa ON sa.id = rr.addr_id
+              WHERE sa.hash_raw = %(raw)b
+                AND rr.spendable_epoch <= (SELECT MAX(epoch_no) FROM block)), 0)
+  - COALESCE((SELECT SUM(w.amount) FROM withdrawal w
+              JOIN stake_address sa ON sa.id = w.addr_id
+              WHERE sa.hash_raw = %(raw)b), 0)
+) AS balance
+"""
+        return query, {"raw": bytes(address)}
+
+    def get_stake_rewards(self, address: Address) -> int:
+        """Withdrawable staking-reward balance (lovelace) for a reward address."""
+        query, values = self._get_stake_rewards(address)
+        r = self.db_query(query, values)
+        return int((r[0]["balance"] if r else 0) or 0)
+
+    async def get_stake_rewards_async(self, address: Address) -> int:
+        """Withdrawable staking-reward balance (lovelace) for a reward address."""
+        query, values = self._get_stake_rewards(address)
+        r = await self.db_query_async(query, values)
+        return int((r[0]["balance"] if r else 0) or 0)
+
     def _get_datum_from_address(
         self,
         address: Address,
@@ -631,7 +665,7 @@ OFFSET %(offset)s"""
 
         import pprint
 
-        pprint.pprint(r, indent=4)
+        pprint.pprint(r, indent=4)  # noqa: T203
 
         return OrderSelector.parse(r)