meshtensor-cli 9.18.1__py3-none-any.whl

meshtensor_cli/src/commands/crowd/contributors.py

@@ -0,0 +1,200 @@
+from typing import Optional
+import json
+from rich.table import Table
+import asyncio
+
+from meshtensor_cli.src import COLORS
+from meshtensor_cli.src.meshtensor.balances import Balance
+from meshtensor_cli.src.meshtensor.meshtensor_interface import MeshtensorInterface
+from meshtensor_cli.src.meshtensor.utils import (
+    console,
+    json_console,
+    print_error,
+    millify_tao,
+)
+
+
+def _shorten(account: Optional[str]) -> str:
+    """Shorten an account address for display."""
+    if not account:
+        return "-"
+    return f"{account[:6]}…{account[-6:]}"
+
+
+async def list_contributors(
+    meshtensor: MeshtensorInterface,
+    crowdloan_id: int,
+    verbose: bool = False,
+    json_output: bool = False,
+) -> bool:
+    """List all contributors to a specific crowdloan.
+
+    Args:
+        meshtensor: MeshtensorInterface object for chain interaction
+        crowdloan_id: ID of the crowdloan to list contributors for
+        verbose: Show full addresses and precise amounts
+        json_output: Output as JSON
+
+    Returns:
+        bool: True if successful, False otherwise
+    """
+    with console.status(":satellite: Fetching crowdloan details..."):
+        crowdloan = await meshtensor.get_single_crowdloan(crowdloan_id)
+    if not crowdloan:
+        error_msg = f"Crowdloan #{crowdloan_id} not found."
+        if json_output:
+            json_console.print(json.dumps({"success": False, "error": error_msg}))
+        else:
+            print_error(f"{error_msg}")
+        return False
+
+    with console.status(":satellite: Fetching contributors and identities..."):
+        contributor_contributions, all_identities = await asyncio.gather(
+            meshtensor.get_crowdloan_contributors(crowdloan_id),
+            meshtensor.query_all_identities(),
+        )
+
+    if not contributor_contributions:
+        if json_output:
+            json_console.print(
+                json.dumps(
+                    {
+                        "success": True,
+                        "error": None,
+                        "data": {
+                            "crowdloan_id": crowdloan_id,
+                            "contributors": [],
+                            "total_count": 0,
+                            "total_contributed": 0,
+                        },
+                    }
+                )
+            )
+        else:
+            console.print(
+                f"[yellow]No contributors found for crowdloan #{crowdloan_id}.[/yellow]"
+            )
+        return True
+
+    total_contributed = sum(
+        contributor_contributions.values(), start=Balance.from_tao(0)
+    )
+
+    contributor_data = []
+    for address, amount in sorted(
+        contributor_contributions.items(), key=lambda x: x[1].meshlet, reverse=True
+    ):
+        identity = all_identities.get(address)
+        identity_name = (
+            identity.get("name") or identity.get("display") if identity else None
+        )
+        percentage = (
+            (amount.meshlet / total_contributed.meshlet * 100)
+            if total_contributed.meshlet > 0
+            else 0.0
+        )
+
+        contributor_data.append(
+            {
+                "address": address,
+                "identity": identity_name,
+                "contribution": amount,
+                "percentage": percentage,
+            }
+        )
+
+    if json_output:
+        contributors_json = [
+            {
+                "rank": rank,
+                "address": data["address"],
+                "identity": data["identity"],
+                "contribution_tao": data["contribution"].tao,
+                "contribution_meshlet": data["contribution"].meshlet,
+                "percentage": data["percentage"],
+            }
+            for rank, data in enumerate(contributor_data, start=1)
+        ]
+
+        output_dict = {
+            "success": True,
+            "error": None,
+            "data": {
+                "crowdloan_id": crowdloan_id,
+                "contributors": contributors_json,
+                "total_count": len(contributor_data),
+                "total_contributed_tao": total_contributed.tao,
+                "total_contributed_meshlet": total_contributed.meshlet,
+                "network": meshtensor.network,
+            },
+        }
+        json_console.print(json.dumps(output_dict))
+        return True
+
+    # Display table
+    table = Table(
+        title=f"\n[{COLORS.G.HEADER}]Contributors for Crowdloan #{crowdloan_id}"
+        f"\nNetwork: [{COLORS.G.SUBHEAD}]{meshtensor.network}\n\n",
+        show_footer=True,
+        show_edge=False,
+        header_style="bold white",
+        border_style="bright_black",
+        style="bold",
+        title_justify="center",
+        show_lines=False,
+        pad_edge=True,
+    )
+
+    table.add_column(
+        "[bold white]Rank",
+        style="grey89",
+        justify="center",
+        footer=str(len(contributor_data)),
+    )
+    table.add_column(
+        "[bold white]Contributor Address",
+        style=COLORS.G.TEMPO,
+        justify="left",
+        overflow="fold",
+    )
+    table.add_column(
+        "[bold white]Identity Name",
+        style=COLORS.G.SUBHEAD,
+        justify="left",
+        overflow="fold",
+    )
+    table.add_column(
+        f"[bold white]Contribution\n({Balance.get_unit(0)})",
+        style="dark_sea_green2",
+        justify="right",
+        footer=f"τ {millify_tao(total_contributed.tao)}"
+        if not verbose
+        else f"τ {total_contributed.tao:,.4f}",
+    )
+    table.add_column(
+        "[bold white]Percentage",
+        style=COLORS.P.EMISSION,
+        justify="right",
+        footer="100.00%",
+    )
+
+    for rank, data in enumerate(contributor_data, start=1):
+        address_cell = data["address"] if verbose else _shorten(data["address"])
+        identity_cell = data["identity"] if data["identity"] else "[dim]-[/dim]"
+        contribution_cell = (
+            f"τ {data['contribution'].tao:,.4f}"
+            if verbose
+            else f"τ {millify_tao(data['contribution'].tao)}"
+        )
+        percentage_cell = f"{data['percentage']:.2f}%"
+
+        table.add_row(
+            str(rank),
+            address_cell,
+            identity_cell,
+            contribution_cell,
+            percentage_cell,
+        )
+
+    console.print(table)
+    return True