yearn-treasury 0.0.20__cp311-cp311-macosx_11_0_arm64.whl → 0.0.49__cp311-cp311-macosx_11_0_arm64.whl

yearn_treasury/rules/other_expense/misc.py

@@ -1,3 +1,11 @@
+"""
+Other expense rules for miscellaneous cases in Yearn Treasury.
+
+This module defines matching logic for miscellaneous other expenses.
+If it doesn't really fit anywhere else in :mod:`~other_expenses`,
+it will end up in here.
+"""
+
 from dao_treasury import TreasuryTx, other_expense
 from y import Network
 

yearn_treasury/rules/other_income/__init__.py

@@ -1,100 +1,2 @@
-from decimal import Decimal
-from typing import Final
-
-from dao_treasury import TreasuryTx, other_income
-from y import Contract, ContractNotVerified, ERC20, Network  # type: ignore [attr-defined]
-
-from yearn_treasury.rules.constants import ZERO_ADDRESS
-
-
-_POINT_ONE: Final = Decimal("0.1")
-
-
-@other_income("Airdrop", Network.Mainnet)
-def is_airdrop(tx: TreasuryTx) -> bool:
-    txhash = tx.hash
-    return txhash in {
-        "0x327684dab9e3ce61d125b36fe0b59cbfbc8aa5ac7a5b051125ab7cac3b93b90b",
-        "0x3424e8a6688c89f7974968213c8c25f3bd8100f78c54475edb001c11a8ad5d21",  # Gnosis SAFE airdrop
-        "0xb39f2991fdc2c70b43046be3eac36bff35c21c7f66e2888a52afc3956abae451",  # Gnosis SAFE airdrop
-        "0x4923fd32b4eacdc1617700c67176935676ca4d06bbfbb73644730c55534623db",  # Gnosis SAFE airdrop
-        "0x5ba604cae0d355835b182fa23c8a58ae695905e69ed08c7cf8a52f3eca889484",  # Gnosis SAFE airdrop
-        "0x44f7d3b2030799ea45932baf6049528a059aabd6387f3128993d646d01c8e877",  # TKX
-        "0xf2dbe58dffd3bc1476755e9f74e2ae07531579d0a3ea9e2aaac2ef902e080c2a",  # TKX
-        "0x8079e9cae847da196dc5507561bc9d1434f765f05045bc1a82df735ec83bc6ec",  # MTV
-        "0x037a9cc5baa7d63a11d0f0720ee552bbf4ad85118ee5425220a263695fedbe9f",  # Gnosis SAFE airdrop
-        # NOTE: this one was rec'd elsewhere, dumped, and WETH sent to treasury
-        "0xc12ded505ea158717890e4ae6e7ab5eb5cb61edbc13dfd125dd0e6f9b1af9477",  # Gnosis SAFE airdrop
-        "0x7c086a82b43b2f49db93b76a0698cf86a9c620b3bf924f0003175b04a17455ad",  # PRISMA
-    } or (
-        # Gnosis SAFE airdrop
-        txhash == "0xe8b5a4ebf1f04048f6226b22b2865a33621e88ea255dcea0cfd7a975a3a7e387"
-        and tx.log_index == 72
-    )
-
-
-@other_income("aToken Yield", Network.Mainnet)
-def is_atoken_yield(tx: TreasuryTx) -> bool:
-    return (
-        tx.symbol in ("aLEND", "aLINK")
-        and tx.from_address.address == ZERO_ADDRESS
-        and tx.to_nickname in ("Yearn Treasury", "Yearn Treasury V1")
-    )
-
-
-@other_income("RoboVault Thank You", Network.Fantom)
-async def is_robovault_share(tx: TreasuryTx) -> bool:
-    """
-    After Yearn devs helped robovault with a vulnerability, robovault committed to sending Yearn a portion of their fees.
-    """
-    if not tx.symbol.startswith("rv") and tx.from_address.is_contract:
-        return False
-
-    try:
-        strat = await Contract.coroutine(tx.from_address.address)
-    except ContractNotVerified:
-        return False
-
-    if not hasattr(strat, "vault"):
-        return False
-
-    if await strat.vault.coroutine(block_identifier=tx.block) == tx.token:
-        return True
-
-    return (
-        tx.from_nickname == "Contract: Strategy"
-        and tx.symbol == "rv3USDCc"
-        and await ERC20(  # type: ignore [call-overload]
-            await strat.vault.coroutine(block_identifier=tx.block),
-            asynchronous=True,
-        ).symbol
-        == "rv3USDCb"
-    )
-
-
-@other_income("Cowswap Gas Reimbursement", Network.Mainnet)
-def is_cowswap_gas_reimbursement(tx: TreasuryTx) -> bool:
-    return (
-        tx.symbol == "ETH"
-        and tx.from_nickname == "Cowswap Multisig"
-        and tx.to_nickname == "yMechs Multisig"
-    )
-
-
-@other_income("USDS Referral Code", Network.Mainnet)
-def is_usds_referral_code(tx: TreasuryTx) -> bool:
-    """Yearn earns some USDS for referring deposits to Maker"""
-    return (
-        tx.symbol == "USDS"
-        and tx.from_address.address == "0x3C5142F28567E6a0F172fd0BaaF1f2847f49D02F"
-    )
-
-
-@other_income("yETH Application Fee", Network.Mainnet)
-def is_yeth_application_fee(tx: TreasuryTx) -> bool:
-    return tx.symbol == "yETH" and tx.to_nickname == "Yearn Treasury" and tx.amount == _POINT_ONE
-
-
-@other_income("yPRISMA Fees", Network.Mainnet)
-def is_yprisma_fees(tx: TreasuryTx) -> bool:
-    return tx.symbol == "yvmkUSD-A" and tx.from_nickname == "Contract: YPrismaFeeDistributor"
+from yearn_treasury.rules.other_income.airdrops import *
+from yearn_treasury.rules.other_income.misc import *

yearn_treasury/rules/other_income/airdrops.py

@@ -0,0 +1,30 @@
+from typing import Final
+
+from dao_treasury import TreasuryTx, other_income
+from y import Network
+
+
+airdrop: Final = other_income("Airdrop")
+
+_SAFE_AIRDROP_CONTRACTS: Final = (
+    "0xA0b937D5c8E32a80E3a8ed4227CD020221544ee6",
+    "0xC0fde70A65C7569Fe919bE57492228DEE8cDb585",
+)
+
+
+@airdrop("SAFE", Network.Mainnet)
+def is_safe_airdrop(tx: TreasuryTx) -> bool:
+    return tx.symbol == "SAFE" and tx.from_address.address in _SAFE_AIRDROP_CONTRACTS  # type: ignore [union-attr]
+
+
+@airdrop("Other", Network.Mainnet)
+def is_airdrop(tx: TreasuryTx) -> bool:
+    return tx.hash in {
+        "0x327684dab9e3ce61d125b36fe0b59cbfbc8aa5ac7a5b051125ab7cac3b93b90b",
+        "0x44f7d3b2030799ea45932baf6049528a059aabd6387f3128993d646d01c8e877",  # TKX
+        "0xf2dbe58dffd3bc1476755e9f74e2ae07531579d0a3ea9e2aaac2ef902e080c2a",  # TKX
+        "0x8079e9cae847da196dc5507561bc9d1434f765f05045bc1a82df735ec83bc6ec",  # MTV
+        # NOTE: this one was rec'd elsewhere, dumped, and WETH sent to treasury
+        "0xc12ded505ea158717890e4ae6e7ab5eb5cb61edbc13dfd125dd0e6f9b1af9477",  # Gnosis SAFE airdrop
+        "0x7c086a82b43b2f49db93b76a0698cf86a9c620b3bf924f0003175b04a17455ad",  # PRISMA
+    }

yearn_treasury/rules/other_income/misc.py

@@ -0,0 +1,80 @@
+# mypy: disable-error-code="union-attr"
+from decimal import Decimal
+from typing import Final, Optional
+
+from dao_treasury import TreasuryTx, other_income
+from y import Contract, ContractNotVerified, ERC20, Network  # type: ignore [attr-defined]
+
+from yearn_treasury.rules.constants import ZERO_ADDRESS
+
+
+_POINT_ONE: Final = Decimal("0.1")
+
+
+@other_income("aToken Yield", Network.Mainnet)
+def is_atoken_yield(tx: TreasuryTx) -> bool:
+    return (
+        tx.symbol in ("aLEND", "aLINK")
+        and tx.from_address.address == ZERO_ADDRESS
+        and tx.to_nickname in ("Yearn Treasury", "Yearn Treasury V1")
+    )
+
+
+@other_income("RoboVault Thank You", Network.Fantom)
+async def is_robovault_share(tx: TreasuryTx) -> bool:
+    """
+    After Yearn devs helped robovault with a vulnerability, robovault committed to sending Yearn a portion of their fees.
+    """
+    if not tx.symbol.startswith("rv") and tx.from_address.is_contract:
+        return False
+
+    try:
+        strat = await tx.from_address.contract_coro
+    except ContractNotVerified:
+        return False
+    else:
+        vault: Optional[Contract] = getattr(strat, "vault", None)
+
+    if vault is None:
+        return False
+
+    if await vault.coroutine(block_identifier=tx.block) == tx.token:
+        return True
+
+    return (
+        tx.from_nickname == "Contract: Strategy"
+        and tx.symbol == "rv3USDCc"
+        and await ERC20(  # type: ignore [call-overload]
+            await vault.coroutine(block_identifier=tx.block),
+            asynchronous=True,
+        ).symbol
+        == "rv3USDCb"
+    )
+
+
+@other_income("Cowswap Gas Reimbursement", Network.Mainnet)
+def is_cowswap_gas_reimbursement(tx: TreasuryTx) -> bool:
+    return (
+        tx.symbol == "ETH"
+        and tx.from_nickname == "Cowswap Multisig"
+        and tx.to_nickname == "yMechs Multisig"
+    )
+
+
+@other_income("USDS Referral Code", Network.Mainnet)
+def is_usds_referral_code(tx: TreasuryTx) -> bool:
+    """Yearn earns some USDS for referring deposits to Maker"""
+    return (
+        tx.symbol == "USDS"
+        and tx.from_address.address == "0x3C5142F28567E6a0F172fd0BaaF1f2847f49D02F"
+    )
+
+
+@other_income("yETH Application Fee", Network.Mainnet)
+def is_yeth_application_fee(tx: TreasuryTx) -> bool:
+    return tx.symbol == "yETH" and tx.to_nickname == "Yearn Treasury" and tx.amount == _POINT_ONE
+
+
+@other_income("yPRISMA Fees", Network.Mainnet)
+def is_yprisma_fees(tx: TreasuryTx) -> bool:
+    return tx.symbol == "yvmkUSD-A" and tx.from_nickname == "Contract: YPrismaFeeDistributor"

yearn_treasury/rules/revenue/vaults.py

@@ -1,9 +1,10 @@
+from decimal import Decimal
 from logging import getLogger
-from typing import Final, Optional
+from typing import Final, Optional, cast
 
-from async_lru import alru_cache
 from dao_treasury import TreasuryTx, revenue
 from eth_typing import ChecksumAddress
+from faster_async_lru import alru_cache
 from y import Contract, Network
 
 from yearn_treasury.vaults import v1, v2
@@ -55,7 +56,7 @@ async def _get_controller(vault: Contract) -> Contract:
 
 @fees("Vaults V1", Network.Mainnet)
 async def is_v1_vault_fees(tx: TreasuryTx) -> bool:
-    token = tx.token.address.address
+    token = tx.token_address
 
     # Fees from single-sided strategies are not denominated in `vault.token` but everything else is
     is_single_sided = _is_single_sided(tx)
@@ -85,7 +86,9 @@ async def is_v1_vault_fees(tx: TreasuryTx) -> bool:
     return False
 
 
-def is_inverse_fees_from_stash_contract(from_address: str, to_nickname: str) -> bool:
+def is_inverse_fees_from_stash_contract(
+    from_address: ChecksumAddress, to_nickname: str | None
+) -> bool:
     return (
         from_address == "0xE376e8e8E3B0793CD61C6F1283bA18548b726C2e"
         and to_nickname == "Token: Curve stETH Pool yVault"
@@ -94,18 +97,19 @@ def is_inverse_fees_from_stash_contract(from_address: str, to_nickname: str) ->
 
 @fees("Vaults V2")
 async def is_v2_vault_fees(tx: TreasuryTx) -> bool:
-    token = tx.token.address.address
-    from_address = tx.from_address.address
+    token = tx.token_address
+    from_address = cast(ChecksumAddress, tx.from_address.address)
     if (
         from_address == token
-        and (vault := v2.get(from_address))  # type: ignore [arg-type]
+        and (vault := v2.get(from_address))
         and tx.to_address == await vault.rewards.coroutine(block_identifier=tx.block)
     ):
         return True
 
-    if is_inverse_fees_from_stash_contract(from_address, tx.to_nickname):  # type: ignore [arg-type]
-        if tx.value_usd > 0:  # type: ignore [operator]
-            tx.value_usd *= -1  # type: ignore [operator]
+    if is_inverse_fees_from_stash_contract(from_address, tx.to_nickname):
+        value_usd = cast(Decimal, tx.value_usd)
+        if value_usd > 0:
+            tx.value_usd = -value_usd
         return True
 
     return False

yearn_treasury/shitcoins.py

@@ -1,8 +1,24 @@
+"""
+List of "shitcoins" (but like, actual shit shitcoins) to ignore in all
+Yearn Treasury analytics.
+
+This module defines, for each blockchain network, a set of token addresses
+known to be unpricable, considered as spam, or otherwise unwanted for
+reporting and analytics. These tokens are passed in to :mod:`eth-portfolio`,
+which contains the logic that prevents these shitcoins from being included in
+any Yearn Treasury outputs.
+
+Since these tokens do nothing but add noise to the outputs, transactions
+involving them are excluded from treasury calculations, reports, and dashboards.
+"""
+
+from typing import Final
+
 from y import Network, convert
 from y.constants import CHAINID
 
 
-_SHITCOINS = {
+_SHITCOINS: Final = {
     Network.Mainnet: (
         "0xC36442b4a4522E871399CD717aBDD847Ab11FE88",  # UNI-V3 NFT, not shitcoin but not pricable
         "0x0329b631464C43f4e8132df7B4ac29a2D89FFdC7",
@@ -34,6 +50,13 @@ _SHITCOINS = {
         "0xe5868468Cb6Dd5d6D7056bd93f084816c6eF075f",
         "0x0a24Bb4842c301276c65086B5d78D5C872993c72",
         "0x63125c0d5Cd9071de9A1ac84c400982f41C697AE",
+        "0x4d22921215cF37e8d49e2Ac6d1F5e9672f63A7c6",
+        "0xe2549E429B78458fa60BC7B1b284d4411E1D5105",
+        "0xCfdD747d041397bcE08B0Fe6ebF7Ef65E9F46795",
+        "0x9745969171a38B40db05c506fe2DA2C36f317627",
+        "0x6051C1354Ccc51b4d561e43b02735DEaE64768B8",
+        "0xf0814d0E47F2390a8082C4a1BD819FDDe50f9bFc",
+        "0x2DBd330bC9B7f3A822a9173aB52172BdDDcAcE2A",
         # just andre tinkering
         "0x5cB5e2d7Ab9Fd32021dF8F1D3E5269bD437Ec3Bf",
         # these arent shitcoins per se but we can't price them and dont expect to in the future, lets save cpu cycles
@@ -50,6 +73,18 @@ _SHITCOINS = {
         "0x57f19E7e1A9066a741f59484481C4D2E9150e9E2",  # MOCO
         "0xb092D8E13Ba50963D57bEcB17a037728D883D02d",  # BABYLABUBU
         "0x4E51960bd33A6edA71a8B24A76311834BD98DD9f",  # AICC
+        "0xB85485421d8a41C4648AB80bE1A725fA6b5Bc86d",  # MEGA
+        "0x46D0Fb47b1e91130D53498EbeE7A968e7e6599f9",  # Ghibli
+        "0x4FbB350052Bca5417566f188eB2EBCE5b19BC964",  # GRG
+        "0x1f6DEb07E1a19bAfF90EC4075447eeF6eb96c0BA",  # BABYMANYU
+        "0xD10EFABA11A51237fa120b15153DD432958bbDE3",  # JIFFPOM
+        "0xCd9594cd25ED2a166362b6F76c523da08c4Ef2e5",  # ESTHER
+        "0x16B907b5d1208Ae6086dE983a5EF45E7890eF272",  # JUNFOX
+        "0xdE56173463d6461001B0891bC90DB224965f5762",  # MAGNUS
+        "0x922824A5b3B3D5f4485fF52F4Ab7Cba5eA491874",  # POSEIDON
+        "0x84F7D2f6FB447Bb11d3E7Ce64D83e1c02c2F3078",  # VIRTUAL
+        "0x5C6Ed14E1017cf75C237A4A4b93Ce1D2f83EB002",  # GRVT
+        "0xf76E6eFf109258fd5F52823d9F9feE7c90C97251",  # wkeyDAO
         # test token?
         "0x372d5d02c6b4075bd58892f80300cA590e92d29E",  # tOUSG
         # dETH? don't think this is needed
@@ -69,6 +104,16 @@ _SHITCOINS = {
         "0xD2F89F59fBC7125b406e3F60A992DFa9FdB76524",  # MISTRAL AI
         "0xa0CCdBCeB5Da30F9d62F7A727F2B35C69dF08760",  # CHUNK AI
         "0x7CE31075d7450Aff4A9a82DdDF69D451B1e0C4E9",  # DEEPSEEK AI
+        "0xf0f9C021AF9B6431FA59DAB75C8e6cB80c0dEa37",  # TESLA AI
+        "0x635eeC65a7Ef10dCF96Bfe051D8A6e5960efe180",  # KLING AI
+        "0xa3Efa0929569c15c20f89B591931899Fb05B4663",  # GPT-5
+        "0x0A953979fdCfD82B08C77dB29705327BeC39ff13",  # GROK4 AI
+        "0xc83377b9eE3CEe4Cc03CCd58AfdE1FB12864aEE3",  # E AI
+        "0x927402ab67c0CDA3c187E9DFE34554AC581441f2",  # SAITABIT
+        "0x691539810DF6e879A377C24CfEE130BBE92708d8",  # NVIDIA AI
+        "0xdC82aC0A89197854cb2240FaBF7E7760a4fF4d9e",  # NVIDIA
+        "0x5Fba8ea5A559CF5c99BA6dd884Ae17C1d621fE5B",  # OSCAR AI
+        "0x57b055656460055192c8EAf616F90Ab76a32CC20",  # Openx
         # matt furry spam
         "0x73228b3D33cC71cB721Fc62950577bE63bd9c8C9",  # Maskman by Matt Furie
         "0x7c28e66436C93BB9F657dDF2BA0eeeCf61369b92",  # Bloodboy by Matt Furie
@@ -78,6 +123,20 @@ _SHITCOINS = {
         "0x2F375Ce83EE85e505150d24E85A1742fd03cA593",  # TEST
     ),
 }
+"""
+Mapping of blockchain networks to tuples of token addresses that should be
+ignored in Yearn Treasury analytics. These tokens are considered unpricable,
+spam, or otherwise unwanted for reporting and analytics purposes.
+
+Each tuple contains token contract addresses that will be excluded from
+treasury calculations, reports, and dashboards for the corresponding network.
+"""
+
 
+SHITCOINS: Final = {convert.to_address(shitcoin) for shitcoin in _SHITCOINS.get(CHAINID, ())}  # type: ignore [call-overload]
+"""Set of checksummed token addresses to ignore for the current chain.
 
-SHITCOINS = {convert.to_address(shitcoin) for shitcoin in _SHITCOINS.get(CHAINID, ())}  # type: ignore [call-overload]
+This set is derived from the _SHITCOINS mapping for the current CHAINID,
+and is used to filter out unpricable, spam, or otherwise unwanted tokens
+from all Yearn Treasury analytics and reporting.
+"""

yearn_treasury/vaults.py

@@ -1,4 +1,18 @@
-from typing import Dict, Final, List
+"""
+Vault discovery and tracking utilities for Yearn Treasury.
+
+This module discovers Yearn vault contracts and maps them to their
+underlying assets using Yearn's on-chain registry contracts. It
+provides dictionaries for v1 and v2 vaults, supporting transaction
+classification, analytics, and reporting across the Yearn Treasury
+system.
+
+Key Responsibilities:
+    - Discover and map all v1 and v2 vault contracts to underlying assets at startup.
+    - Provide lookup tables for use in vault deposit/withdrawal sort rules.
+"""
+
+from typing import Dict, Final
 
 from brownie import chain
 from eth_typing import ChecksumAddress
@@ -27,11 +41,10 @@ if chain.id == Network.Mainnet:
 
     # TODO: make resolve_ens util in eth-port and refactor this out
     v2_registries = [
-        event["newAddress"].hex()  # type: ignore [attr-defined]
-        for event in Events(addresses=resolver, topics=topics).events(now)
+        event["newAddress"].hex() for event in Events(addresses=resolver, topics=topics).events(now)
     ]
 
     for event in Events(addresses=list(map(str, v2_registries))).events(now):
         if event.name == "NewVault":
             vault_address = event["vault"]
-            v2[vault_address] = Contract(vault_address)  # type: ignore [index]
+            v2[vault_address] = Contract(vault_address)

yearn_treasury/yteams.py

@@ -0,0 +1,208 @@
+# This is a loose copy of an old script and will not likely be refactored into something pretty
+import os
+from datetime import datetime, timedelta, timezone
+from decimal import Decimal
+from functools import lru_cache
+from logging import getLogger
+from typing import Any, Dict, Final, List, Set, Tuple
+
+import a_sync
+from brownie import chain
+from eth_portfolio.structs import TokenTransfer
+from eth_portfolio._ydb.token_transfers import InboundTokenTransfers
+from pandas import DataFrame, MultiIndex
+from y import Contract, Network, get_block_at_timestamp
+from y.exceptions import ContractNotVerified
+
+from yearn_treasury.constants import ZERO_ADDRESS
+
+
+DATA_FOLDER: Final = os.path.join(".", "data")
+OUTPUT_FILE: Final = os.path.join(DATA_FOLDER, f"teams_revenue_{chain.id}.csv")
+NUMBER_OF_MONTHS_TO_INCLUDE_IN_REPORT: Final = 36
+
+if not os.path.exists(DATA_FOLDER):
+    os.makedirs(DATA_FOLDER)
+
+
+# TODO: maybe move this into constants for reuse
+yteams_addresses = {
+    Network.Mainnet: {
+        "v3": {
+            "ms": "0x33333333D5eFb92f19a5F94a43456b3cec2797AE",
+            "splits": {"0x2A12CAA2c13Af03c117D836CA3811a5Ca946133B": 12.5},
+        },
+        "dinobots": {
+            "ms": "0x2C01B4AD51a67E2d8F02208F54dF9aC4c0B778B6",
+            "splits": {"0xC4f238633A85A854C4702d2c66264771D1fa7904": 17.5},
+        },
+        "ylockers": {
+            "ms": "0x4444AAAACDBa5580282365e25b16309Bd770ce4a",
+            "splits": {
+                "0xac580302548FCCBBf00020de20C3A8AA516821AD": 2.5,
+                "0x794f80E899c772de9E326eC83cCfD8D94e208B49": 6.25,
+                "0x5FF0f87b05806ce89967638CA727Af8309d92A89": 12.5,
+                "0x5A7575368797695BefD785f546C6b8B7e9D37f8c": 15.625,
+            },
+        },
+        # "ylockers others": {"ms": "0x4444AAAACDBa5580282365e25b16309Bd770ce4a","splits":{"0x5FF0f87b05806ce89967638CA727Af8309d92A89":12.5, "0x5A7575368797695BefD785f546C6b8B7e9D37f8c":15.625}},
+        "yaudit": {
+            "ms": "0x8973B848775a87a0D5bcf262C555859b87E6F7dA",
+            "splits": {
+                "0xd7A1DBe236A38528D54317415a530b2326068373": 35,
+                "0xF104F38592287e25868BD8C3dcCCa1a311916f88": 35,
+                "0x1a9D272C3b7fE427639702A332D51348213B0bC1": 20,
+            },
+        },
+        "yeth": {
+            "ms": "0xeEEEEeeeEe274C3CCe13f77C85d8eBd9F7fd4479",
+            "splits": {"0x14EFe6390C6758E3fE4379A14e3B329274b1b072": 25},
+        },
+        "yfarm": {
+            "ms": "0x55157997cb324a374cCd7c40914ff879Fd9D515C",
+            "splits": {"0x0B3cCe59E038373F6008D9266B6D6eB4d21689b1": 50},
+        },
+        "sms": {
+            "ms": "0x16388463d60FFE0661Cf7F1f31a7D658aC790ff7",
+            "splits": {"0xd6748776CF06a80EbE36cd83D325B31bb916bf54": 25},
+        },
+    }
+}[Network(chain.id)]
+
+
+logger: Final = getLogger(__name__)
+
+_not_verified: Final[Set[str]] = set()
+_warned: Final[Set[TokenTransfer]] = set()
+
+_known_tokens_without_prices: Final = frozenset({"SAFE", "vCOW"})
+"""When there is a PriceError for these tokens, no logs will be emitted."""
+
+
+@lru_cache(maxsize=None)
+def transfers_for(wallet: str) -> InboundTokenTransfers:
+    return InboundTokenTransfers(wallet, 0, load_prices=True)  # type: ignore [arg-type]
+
+
+async def calculate_teams_revenue_expenses() -> None:
+    logger.info("Starting process to calculate teams revenues and expenses")
+    timestamps = get_timestamps_for_report()
+
+    async def get_coros_for_timestamp(dt: datetime) -> Dict[str, Dict[str, Decimal]]:
+        return await a_sync.gather(
+            {
+                label: total(label, wallet_info, dt)
+                for label, wallet_info in yteams_addresses.items()
+            }
+        )
+
+    all_data = await a_sync.gather({dt: get_coros_for_timestamp(dt) for dt in timestamps})
+
+    result = {
+        (dt, teams, movement): values
+        for dt, data in all_data.items()
+        for teams, info in data.items()
+        for movement, values in info.items()
+    }
+    df = DataFrame.from_dict(result, orient="index")
+    print("------------")
+    # print(df.index)
+    # print(df.head(10))
+    df.index = MultiIndex.from_tuples(df.index)
+    print("********")
+    # print(df.index)
+    df.reset_index(inplace=True)
+    df.columns = ["datetime", "team", "label", "value"]
+    df.to_csv(OUTPUT_FILE)
+    logger.info(
+        f"Finished processing yteams calculations and saved file to {os.path.abspath(OUTPUT_FILE)}"
+    )
+
+
+def get_timestamps_for_report() -> List[datetime]:
+    now = datetime.now(tz=timezone.utc)
+    prev_month_end = datetime(
+        year=now.year,
+        month=now.month,
+        day=1,
+        hour=0,
+        minute=0,
+        second=0,
+        microsecond=0,
+        tzinfo=timezone.utc,
+    ) - timedelta(microseconds=1)
+    datetimes = []
+    print("Exporting report for timestamps:")
+    for _ in range(NUMBER_OF_MONTHS_TO_INCLUDE_IN_REPORT):
+        print(f" - {prev_month_end}")
+        datetimes.append(prev_month_end)
+        prev_month_end -= timedelta(days=prev_month_end.day)
+    return datetimes
+
+
+async def total(label: str, wallet_info: Dict[str, Any], timestamp: datetime) -> Dict[str, Decimal]:
+    rev = await sum_revenue_transfers.sum(wallet_info["splits"].items(), timestamp=timestamp)
+    grants = await sum_grants_received(wallet_info["ms"], timestamp)
+    if rev > 10_000_000:
+        raise ValueError(rev)
+    if grants > 10_000_000:
+        raise ValueError(grants)
+    net = rev - grants
+    if label == "yaudit":
+        logger.info("--- %s thru %s ---", label, timestamp)
+        logger.info("inbound  %s", rev)
+        logger.info("grants  -%s", grants)
+        logger.info("net      %s", net)
+    return {"revenue": rev, "grants": grants, "total": net}
+
+
+@a_sync.a_sync(default="async")
+async def sum_revenue_transfers(params: Tuple[str, Decimal], timestamp: datetime) -> Decimal:
+    wallet, rev_share = params
+    block = await get_block_at_timestamp(timestamp)
+    total = Decimal(0)
+    async for transfer in transfers_for(wallet).yield_thru_block(block):
+        transfer = await transfer
+        if transfer is None:
+            # failed to decode, probably shitcoin
+            continue
+        if not transfer.value:
+            # zero value transfer
+            continue
+        if transfer.price:
+            total += transfer.value * transfer.price
+        elif transfer not in _warned and transfer.token not in _known_tokens_without_prices:
+            logger.warning(f"BAD: {transfer}")
+            _warned.add(transfer)
+    return round(total * Decimal((100 - rev_share) / 100), 8)
+
+
+async def sum_grants_received(wallet: str, timestamp: datetime) -> Decimal:
+    grants = Decimal(0)
+    block = await get_block_at_timestamp(timestamp)
+    async for transfer in transfers_for(wallet).yield_thru_block(block):
+        transfer = await transfer
+        if transfer is None:
+            # failed to decode, probably shitcoin
+            continue
+        if not transfer.value:
+            # zero value transfer
+            continue
+        if transfer.price:
+            if transfer.from_address != ZERO_ADDRESS:
+                try:
+                    contract = await Contract.coroutine(transfer.from_address)
+                    if (
+                        hasattr(contract, "recipient") and await contract.recipient == wallet
+                    ) or transfer.from_address == "0xFEB4acf3df3cDEA7399794D0869ef76A6EfAff52":
+                        grants += transfer.value * transfer.price
+                except ContractNotVerified as e:
+                    if str(e) not in _not_verified:
+                        _not_verified.add(str(e))
+                        logger.debug(f"{e.__class__.__name__}: {e}")
+                except Exception as e:
+                    logger.warning(f"{e.__class__.__name__}: {e}")
+        elif transfer not in _warned and transfer.token not in _known_tokens_without_prices:
+            logger.warning(f"BAD: {transfer}")
+            _warned.add(transfer)
+    return round(grants, 8)

{yearn_treasury-0.0.20.dist-info → yearn_treasury-0.0.49.dist-info}/METADATA

@@ -1,13 +1,12 @@
 Metadata-Version: 2.4
 Name: yearn_treasury
-Version: 0.0.20
+Version: 0.0.49
 Summary: This WIP library runs the eth-portfolio and dao-treasury exporters for the Yearn Finance treasury.
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: Intended Audience :: End Users/Desktop
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
@@ -16,7 +15,8 @@ Classifier: Operating System :: OS Independent
 Classifier: Topic :: Software Development :: Libraries
 Requires-Python: >=3.10,<3.13
 Description-Content-Type: text/markdown
-Requires-Dist: dao-treasury<0.1,>=0.0.39
+Requires-Dist: dao-treasury==0.0.72
+Requires-Dist: faster-async-lru<3,>=2.0.5
 Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type