strideutils 0.3.4__tar.gz → 0.3.6__tar.gz

{strideutils-0.3.4 → strideutils-0.3.6}/PKG-INFO

@@ -1,13 +1,11 @@
 Metadata-Version: 2.1
 Name: strideutils
-Version: 0.3.4
+Version: 0.3.6
 Summary: 
 Author: Your Name
 Author-email: you@example.com
-Requires-Python: >=3.9
+Requires-Python: >=3.11
 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
 Requires-Dist: bech32 (>=1.2.0)

{strideutils-0.3.4 → strideutils-0.3.6}/pyproject.toml

@@ -1,13 +1,13 @@
 [tool.poetry]
 name = "strideutils"
-version = "0.3.4"
+version = "0.3.6"
 description = ""
 authors = ["Your Name <you@example.com>"]
 readme = "README.md"
 include = ["strideutils", "strideutils-stubs"]
 
 [tool.poetry.dependencies]
-python = ">=3.9"
+python = ">=3.11"
 google-api-python-client = ">=2.100.0"
 google-auth-httplib2 = ">=0.1.1"
 google-auth-oauthlib = ">=1.1.0"

{strideutils-0.3.4 → strideutils-0.3.6}/strideutils/stride_requests.py

@@ -7,7 +7,6 @@ from functools import lru_cache
 from typing import Any, Dict, List, Optional, Tuple, Union, cast
 
 import bech32
-import dateutil.parser
 import pandas as pd
 import pytz
 import requests
@@ -289,6 +288,7 @@ def get_unbonding_records(
     E.g.
     [
         {
+            "epoch_number": "575",
             "st_token_amount": "3218590",
             "native_token_amount": "0",
             "denom": "ucmdx",
@@ -313,17 +313,74 @@ def get_unbonding_records(
     # This function returns the host zone unbonding record
     host_zone_unbondings = []
     for epoch_unbonding in epoch_unbondings:
+        epoch_number = epoch_unbonding["epoch_number"]
         for host_zone_unbonding in epoch_unbonding["host_zone_unbondings"]:
 
             chain_id_match = host_zone_unbonding["host_zone_id"] == chain_id
             status_match = not status or host_zone_unbonding["status"] == status
 
             if chain_id_match and status_match:
+                host_zone_unbonding["epoch_number"] = epoch_number
                 host_zone_unbondings.append(host_zone_unbonding)
 
     return host_zone_unbondings
 
 
+def get_unbonding_queue_records(chain_id: str, block_height: int = 0) -> List[Dict[str, Any]]:
+    """
+    Queries all the unbonding records that are ready to initiate an unbonding
+    This is identified by records in status UNBONDING_QUEUE with a non-zero unbonding amount
+    """
+    unbonding_records = get_unbonding_records(
+        chain_id=chain_id,
+        block_height=block_height,
+        status=types.UnbondingRecordStatus.UNBONDING_QUEUE,
+    )
+    return [ur for ur in unbonding_records if int(ur["native_token_amount"]) > 0]
+
+
+def get_unbonded_records(
+    chain_id: str,
+    block_height: int = 0,
+    filter_unbonded_last_epoch: bool = False,
+) -> List[Dict[str, Any]]:
+    """
+    Queries the list of all unbonding records records that have finished unbonding
+    This is identified by records in status EXIT_TRANSFER_QUEUE with an unbonding time
+    in the past
+    Optionally filter to records that just unbonded last epoch
+    """
+    unbonding_records = get_unbonding_records(
+        chain_id=chain_id,
+        block_height=block_height,
+        status=types.UnbondingRecordStatus.EXIT_TRANSFER_QUEUE,
+    )
+
+    # If we're fetching an older block, use that block's time as the "current time"
+    current_time = int(datetime.datetime.now().timestamp() * 1e9)
+    if block_height != 0:
+        current_time = int(get_block_time(block_height).timestamp() * 1e9)
+
+    # Filter to the records that have finished unbonding, and optionally also
+    # filter to records that just finished during the last epoch
+    day = 24 * 60 * 60 * 1e9
+    filtered_unbonding_records = []
+    for unbonding_record in unbonding_records:
+        unbonding_time = int(unbonding_record["unbonding_time"])
+
+        # Ignore records that haven't finished unbonding yet
+        if unbonding_time > current_time:
+            continue
+
+        # If filtering from last epoch, ignore records that unbonded over a day ago
+        if filter_unbonded_last_epoch and (current_time - unbonding_time) > day:
+            continue
+
+        filtered_unbonding_records.append(unbonding_record)
+
+    return filtered_unbonding_records
+
+
 def get_lsm_deposit_records(chain_id: str) -> List[Dict[str, Any]]:
     """
     Queries all the LSM deposit records
@@ -627,7 +684,7 @@ def get_block_time(height: int, rpc_endpoint: str = config.stride.rpc_endpoint)
     url = f"{rpc_endpoint}/block?height={height}"
     resp = request(url)
     block_time = resp["result"]["block"]["header"]["time"]
-    block_datetime = dateutil.parser.parse(block_time).replace(tzinfo=None)
+    block_datetime = datetime.datetime.fromisoformat(block_time)
     return block_datetime
 
 
@@ -1027,7 +1084,7 @@ def get_stride_epoch_start_time() -> datetime.datetime:
     Returns the epoch start time for the current "stride" epoch
     """
     stride_epoch = get_stride_epoch()
-    return dateutil.parser.parse(stride_epoch["current_epoch_start_time"]).replace(tzinfo=None)
+    return datetime.datetime.fromisoformat(stride_epoch["current_epoch_start_time"])
 
 
 def get_day_epoch_start_time() -> datetime.datetime:
@@ -1035,7 +1092,7 @@ def get_day_epoch_start_time() -> datetime.datetime:
     Returns the epoch start time for the current "day" epoch
     """
     day_epoch = get_day_epoch()
-    return dateutil.parser.parse(day_epoch["current_epoch_start_time"]).replace(tzinfo=None)
+    return datetime.datetime.fromisoformat(day_epoch["current_epoch_start_time"])
 
 
 def get_stride_epoch_start_height() -> int: