dao-treasury 0.0.31__cp312-cp312-macosx_11_0_arm64.whl → 0.0.32__cp312-cp312-macosx_11_0_arm64.whl

dao_treasury/streams/llamapay.py

@@ -1,6 +1,6 @@
 import asyncio
-from datetime import datetime, timedelta, timezone
-from decimal import Decimal
+import datetime as dt
+import decimal
 from logging import getLogger
 from typing import (
     Awaitable,
@@ -21,7 +21,7 @@ from brownie.network.event import _EventItem
 from eth_typing import BlockNumber, ChecksumAddress, HexAddress, HexStr
 from tqdm.asyncio import tqdm_asyncio
 
-from y import Contract, Network, get_block_at_timestamp, get_price
+import y
 from y.time import UnixTimestamp
 from y.utils.events import decode_logs, get_logs_asap
 
@@ -38,18 +38,31 @@ from dao_treasury._wallet import TreasuryWallet
 
 logger: Final = getLogger(__name__)
 
-_UTC: Final = timezone.utc
+_UTC: Final = dt.timezone.utc
 
 _ONE_DAY: Final = 60 * 60 * 24
 
 _STREAMS_THREAD: Final = AsyncThreadPoolExecutor(1)
 
 create_task: Final = asyncio.create_task
+sleep: Final = asyncio.sleep
+
+datetime: Final = dt.datetime
+timedelta: Final = dt.timedelta
+fromtimestamp: Final = datetime.fromtimestamp
+now: Final = datetime.now
+
+Decimal: Final = decimal.Decimal
 
 ObjectNotFound: Final = pony.orm.ObjectNotFound
 commit: Final = pony.orm.commit
 db_session: Final = pony.orm.db_session
 
+Contract: Final = y.Contract
+Network: Final = y.Network
+get_block_at_timestamp: Final = y.get_block_at_timestamp
+get_price: Final = y.get_price
+
 
 networks: Final = [Network.Mainnet]
 
@@ -67,13 +80,13 @@ if yfi_stream_factory := {
 
 
 def _generate_dates(
-    start: datetime, end: datetime, stop_at_today: bool = True
-) -> Iterator[datetime]:
+    start: dt.datetime, end: dt.datetime, stop_at_today: bool = True
+) -> Iterator[dt.datetime]:
     current = start
     while current < end:
         yield current
         current += timedelta(days=1)
-        if stop_at_today and current.date() > datetime.now(_UTC).date():
+        if stop_at_today and current.date() > now(_UTC).date():
             break
 
 
@@ -86,7 +99,7 @@ def _get_streamToStart(stream_id: HexStr) -> _StreamToStart:
     if streamToStart := _streamToStart_cache.get(stream_id):
         return streamToStart
     with db_session:
-        contract: Contract = Stream[stream_id].contract.contract  # type: ignore [misc]
+        contract: y.Contract = Stream[stream_id].contract.contract  # type: ignore [misc]
     streamToStart = contract.streamToStart.coroutine
     _streamToStart_cache[stream_id] = streamToStart
     return streamToStart
@@ -163,7 +176,7 @@ class LlamaPayProcessor:
             for stream_contract in self.stream_contracts
         )
 
-    async def _load_contract_events(self, stream_contract: Contract) -> None:
+    async def _load_contract_events(self, stream_contract: y.Contract) -> None:
         events = decode_logs(
             await get_logs_asap(stream_contract.address, None, sync=False)
         )
@@ -311,7 +324,7 @@ class LlamaPayProcessor:
                 return
 
     async def process_stream_for_date(
-        self, stream_id: HexStr, date_obj: datetime
+        self, stream_id: HexStr, date_obj: dt.datetime
     ) -> Optional[StreamedFunds]:
         entity = await _STREAMS_THREAD.run(
             StreamedFunds.get_entity, stream_id, date_obj
@@ -323,8 +336,10 @@ class LlamaPayProcessor:
             Stream._get_token_and_start_date, stream_id
         )
         check_at = date_obj + timedelta(days=1) - timedelta(seconds=1)
+        if check_at > now(tz=_UTC):
+            await sleep((check_at - now(tz=_UTC)).total_seconds())
         block = await get_block_at_timestamp(check_at, sync=False)
-        price_fut = asyncio.create_task(get_price(stream_token, block, sync=False))
+        price_fut = create_task(get_price(stream_token, block, sync=False))
         start_timestamp = await _get_start_timestamp(stream_id, block)
         if start_timestamp == 0:
             if await _STREAMS_THREAD.run(Stream.check_closed, stream_id):
@@ -335,9 +350,7 @@ class LlamaPayProcessor:
                 block -= 1
                 start_timestamp = await _get_start_timestamp(stream_id, block)
 
-            block_datetime = datetime.fromtimestamp(
-                await _get_block_timestamp(block), tz=_UTC
-            )
+            block_datetime = fromtimestamp(await _get_block_timestamp(block), tz=_UTC)
             assert block_datetime.date() == date_obj.date()
             seconds_active = (check_at - block_datetime).seconds
             is_last_day = True

{dao_treasury-0.0.31.dist-info → dao_treasury-0.0.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dao_treasury
-Version: 0.0.31
+Version: 0.0.32
 Summary: Produce comprehensive financial reports for your on-chain org
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers

{dao_treasury-0.0.31.dist-info → dao_treasury-0.0.32.dist-info}/RECORD

@@ -1,20 +1,24 @@
-17ebe61b88bd37338d0a__mypyc.cpython-312-darwin.so,sha256=1_l12kolnmYAe85UxowJWPd5fFr-kUBhjcYDKDP0wVE,812912
+bf2b4fe1f86ad2ea158b__mypyc.cpython-312-darwin.so,sha256=tuSJfvY3HqkqdslUakK3CmaEjKybsWtvATjEG9QIiVU,813808
+dao_treasury-0.0.32.dist-info/RECORD,,
+dao_treasury-0.0.32.dist-info/WHEEL,sha256=V1loQ6TpxABu1APUg0MoTRBOzSKT5xVc3skizX-ovCU,136
+dao_treasury-0.0.32.dist-info/top_level.txt,sha256=Gw-ri_26lZA_3hwhnOX48mffUbmBGr8NhtoQ0XoMeF8,41
+dao_treasury-0.0.32.dist-info/METADATA,sha256=CtChVdcnb51eL9ngQP9PEA_D4uYWnelEtQuAUkD9G_g,7030
 dao_treasury/_wallet.py,sha256=sIeUsyu9Y8MmOm-9bfwIEYjJ9Owgv1GXdTLZpcXMnCg,10338
 dao_treasury/db.py,sha256=pGz4C2OmxHI9CyrphbFzs04aiVP_VQeViaavCiJuR3o,44831
 dao_treasury/docker-compose.yaml,sha256=Q6NPNT9iKkKOehDcKtxltvQjbxZuXmb51Tjwb1M21_I,1376
 dao_treasury/ENVIRONMENT_VARIABLES.py,sha256=yKvgOMUyU7-fF_-1zSMXML7ah7-r2f0oHrOV0p10LPc,195
-dao_treasury/constants.cpython-312-darwin.so,sha256=Rwq9iJx5bxsaeLJL2diX9VTb5wJOqNwjpCVm6wdosYk,50656
-dao_treasury/_docker.cpython-312-darwin.so,sha256=W5-W-GDBfzPgXKu2q3cJX7n-tI44wdLC7LyKW2rigDU,50640
+dao_treasury/constants.cpython-312-darwin.so,sha256=jsaBEneejCpG06SBcfxRGa6y6rddxBNhrPqgLJu3PS4,50656
+dao_treasury/_docker.cpython-312-darwin.so,sha256=QWiX9DvKus4LHMm02I2kHjR_FqHhIypWriohs4dMpxk,50640
 dao_treasury/constants.py,sha256=IFg_CyVnfPhPjKyFX1Ix4QaB8ghbOI5QZGSy2TNsUSA,507
 dao_treasury/__init__.py,sha256=ZQ0ZROQzSyJdYuQS1Z_VD0c7CiV8BWZcLTX66wO4slc,1004
 dao_treasury/_nicknames.py,sha256=deSpcDMtC_78C07tW0fZ_NzHIfHXsoHBVXfb_Dvlzsw,480
 dao_treasury/types.py,sha256=dJDx0hYfDDL3slcUzRnCgrzFFsgCzoqg8uaTtLaT4kk,3685
-dao_treasury/_wallet.cpython-312-darwin.so,sha256=Ibb1I6T-68O1dYpM--2uu5UL6zNTni9NUDQVXL9wml4,50640
+dao_treasury/_wallet.cpython-312-darwin.so,sha256=DSYsuOQXq_w4jLG--yr15XXVXEJWcGC6tHN-jl0A2-c,50640
 dao_treasury/_docker.py,sha256=7MnPA5xhRCM8Kqiogk87ltUVn8i_hpkNgO_hFC-7sPw,5430
-dao_treasury/types.cpython-312-darwin.so,sha256=JTnQ2hiIXHqjOf5k1seqb316w-o5AKi36dJomJkvxJM,50616
+dao_treasury/types.cpython-312-darwin.so,sha256=J7FcDMlPn6NgW3vQ-uLegY6gZW3fX7Fb0elZ6cU6o1s,50616
 dao_treasury/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dao_treasury/treasury.py,sha256=7xLVQPqnDd3cexbYuF1opTjCiHgkPqHFOanRom8wiLE,6506
-dao_treasury/_nicknames.cpython-312-darwin.so,sha256=RTr9TybXLLhCGTbFMdPiFqt7tu5X7TxekTweqIZmoGI,50656
+dao_treasury/_nicknames.cpython-312-darwin.so,sha256=gefMDqMNyIg788Nha-Ga6Pet822S85hGHFvGpNLSyvs,50656
 dao_treasury/main.py,sha256=31EA10Fpaj9y-cRJPnNnvCBI38UM0848xlzn7A0BzC4,8047
 dao_treasury/.grafana/provisioning/datasources/datasources.yaml,sha256=mjQs8or651NgD_bZnKBm8vSghzjw_g3enL01YPt0dik,327
 dao_treasury/.grafana/provisioning/dashboards/dashboards.yaml,sha256=ecu7lHpcrYFTSEHbZErX63mn2CFPqEXcprtcYDcHPSo,713
@@ -23,27 +27,23 @@ dao_treasury/.grafana/provisioning/dashboards/transactions/Treasury Transactions
 dao_treasury/.grafana/provisioning/dashboards/treasury/Treasury.json,sha256=p9bupL06-CKqkqCOb6KkgP1LCa6P1VMkq3yhkyqMAzI,70178
 dao_treasury/.grafana/provisioning/dashboards/treasury/Cashflow.json,sha256=52LHxaxfPj5NoPB1guTvayd08ETIfFjlydBsoywnxjI,42349
 dao_treasury/.grafana/provisioning/dashboards/summary/Monthly.json,sha256=y7t4O0ujImcfVULGsV_0WxcZCm7sz1jtAB5ZXUfPww4,7911
-dao_treasury/streams/__init__.cpython-312-darwin.so,sha256=WQ8zAnMqHPmQgI9otvpL36El_BKFniraSGDHvkT6-FU,50640
-dao_treasury/streams/llamapay.py,sha256=jg6oiFWy5WmDnHXJucG-hsAEeOHQQcMOa5cv7A7QcsA,12465
-dao_treasury/streams/llamapay.cpython-312-darwin.so,sha256=es8nHWzqy5BV--sC5xdRxKeZvAIMi60HxX8CdrWjeLI,50656
+dao_treasury/streams/__init__.cpython-312-darwin.so,sha256=93qtHMptO0cv1DyLb7Rk2U65WPne4mRkq2w5-l5ec5o,50640
+dao_treasury/streams/llamapay.py,sha256=xc2sgmTV4EnQFFemea87QoUdIeGtO9OJQCwJG9c-SJQ,12774
+dao_treasury/streams/llamapay.cpython-312-darwin.so,sha256=sgArj2A8M7XWFIFEuN3zcpzYd73Q91XnnTQ3TPz2R7I,50656
 dao_treasury/streams/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 dao_treasury/sorting/_rules.py,sha256=vWkqJVA8yCHKRv9sw872Jq6G0zpM-v_J1jCz-Mgleec,8994
-dao_treasury/sorting/__init__.cpython-312-darwin.so,sha256=DwtRX6stQ3CT2_xsY2Go56O1UBlfB4wwrLf3ZoKqy88,50640
-dao_treasury/sorting/_rules.cpython-312-darwin.so,sha256=KYf4OnMfYjFmn8y2R4kjZez_l58mUqYRErpZU1uzx6I,50640
+dao_treasury/sorting/__init__.cpython-312-darwin.so,sha256=eNZ51uU9thTUGpPByCC1k-THvNmL2xZV7gYVqUBjbFg,50640
+dao_treasury/sorting/_rules.cpython-312-darwin.so,sha256=3poAPJgHtWhKYWRQLGCfkQCQIpKeK369AFlWjafQQ2M,50640
 dao_treasury/sorting/__init__.py,sha256=F_yKKdpdHKGH5RRt-8cZ-M1CuHFI_XvSypnE0YHdovo,5785
 dao_treasury/sorting/factory.py,sha256=pFD6luc0bKxOTT1hlAp3ZwJw6R0wN8pjIZrqMZwhsko,10185
-dao_treasury/sorting/factory.cpython-312-darwin.so,sha256=eGPJw9Mlpr3Aua2tctnRpGRhPDlmWQHjGoMTuQASHJI,50656
+dao_treasury/sorting/factory.cpython-312-darwin.so,sha256=BGmMmdUG8QjFPJ0Y-kTWjUNpNY7PviXrtubzVh-9s1M,50656
 dao_treasury/sorting/rule.py,sha256=-4dokQbRJ2GPJLEwtpgGG0aXAM1pW4jI17aezo--Z5g,11491
-dao_treasury/sorting/rule.cpython-312-darwin.so,sha256=mI0srxt31uPwH6koqwh-3PpfHPeqZz6SH_XHQOGpzgo,50632
-dao_treasury/sorting/_matchers.cpython-312-darwin.so,sha256=qk2-w6A2fffVuOOJ8sDYlNR58l5IE1-iwIh7NZ8YIDs,50656
+dao_treasury/sorting/rule.cpython-312-darwin.so,sha256=EVjBIm7x7Y2zhOh7zm5CVcLqXZjGwR4wLK5NlCKRLDA,50632
+dao_treasury/sorting/_matchers.cpython-312-darwin.so,sha256=0D9-fDLZQzJhXADb2NQRtPXQ1qArh29NXwWkY_2WDrM,50656
 dao_treasury/sorting/_matchers.py,sha256=u0GfPYiODr_NtzbfyLxEKDuXmWRiaeuZulDj9FPDOH8,14005
-dao_treasury/sorting/rules/__init__.cpython-312-darwin.so,sha256=XK-_uR-n8SO5fjoaCV6YahdW4ffmKr2oANYZp0KuE68,50640
+dao_treasury/sorting/rules/__init__.cpython-312-darwin.so,sha256=Zc7wA5GVx2pVutZkytSMJehgchojR1j45_DrcLyNT_A,50640
 dao_treasury/sorting/rules/__init__.py,sha256=ofzIsW0P74kuKr6DlScZIF8p7hBh6guRpe1uOq6oLLA,48
-dao_treasury/sorting/rules/ignore/__init__.cpython-312-darwin.so,sha256=iEGW2MM1DKuyOD3OgdrO-VM3rBfRoauw5t4LyePaYNY,50656
+dao_treasury/sorting/rules/ignore/__init__.cpython-312-darwin.so,sha256=N-ZqgILCm0a2TH-ruc5988OYZD0BB8t3nfcSpmKfAME,50656
 dao_treasury/sorting/rules/ignore/llamapay.py,sha256=NgiVt9MRtI-f4mw9U5_gqttdPqDa3vt_z5XWJP_UYWs,763
-dao_treasury/sorting/rules/ignore/llamapay.cpython-312-darwin.so,sha256=MWrU4v255ExqMowOxQY8NBzFborECeJ3tDOOnmax_PA,50672
+dao_treasury/sorting/rules/ignore/llamapay.cpython-312-darwin.so,sha256=kkVk1DQYQEoGY1sHoBCwr3Qhd94M5bzoZtQTTIzlqQw,50672
 dao_treasury/sorting/rules/ignore/__init__.py,sha256=dJ4oKSLe7fal1Ct7XtMCfVJAtvZXcyIomf8HF3gWUhQ,57
-dao_treasury-0.0.31.dist-info/RECORD,,
-dao_treasury-0.0.31.dist-info/WHEEL,sha256=V1loQ6TpxABu1APUg0MoTRBOzSKT5xVc3skizX-ovCU,136
-dao_treasury-0.0.31.dist-info/top_level.txt,sha256=VtgTZeHElDnBUEn2XLN6Hy_nev5HfhK5c-OnfU04bUs,41
-dao_treasury-0.0.31.dist-info/METADATA,sha256=l2XgIiKMWCNWUnBiNIZ8Igc8GjHMKOuaJLaJYfZClr4,7030

dao_treasury-0.0.32.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+bf2b4fe1f86ad2ea158b__mypyc
+dao_treasury

dao_treasury-0.0.31.dist-info/top_level.txt

@@ -1,2 +0,0 @@
-17ebe61b88bd37338d0a__mypyc
-dao_treasury