strideutils 0.2.2__tar.gz → 0.2.4__tar.gz

{strideutils-0.2.2 → strideutils-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: strideutils
-Version: 0.2.2
+Version: 0.2.4
 Summary: 
 Author: Your Name
 Author-email: you@example.com
@@ -17,6 +17,7 @@ Requires-Dist: google-auth-oauthlib (>=1.1.0)
 Requires-Dist: google-cloud-bigquery (>=3.11.4)
 Requires-Dist: gspread (>=6.0.2)
 Requires-Dist: pandas (>=2.1.1)
+Requires-Dist: pycryptodome (>=3.20.0)
 Requires-Dist: python-dateutil (>=2.8.2)
 Requires-Dist: pytz (>=2023.3)
 Requires-Dist: pyyaml (>=6.0.1)

{strideutils-0.2.2 → strideutils-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "strideutils"
-version = "0.2.2"
+version = "0.2.4"
 description = ""
 authors = ["Your Name <you@example.com>"]
 readme = "README.md"
@@ -21,9 +21,13 @@ pandas = ">=2.1.1"
 twilio = ">=8.10.0"
 tabulate = ">=0.9.0"
 pyyaml = ">=6.0.1"
-python-dateutil= ">=2.8.2"
+python-dateutil = ">=2.8.2"
+pycryptodome = ">=3.20.0"
 
 
+[tool.poetry.group.dev.dependencies]
+pytest = "^8.1.1"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

strideutils-0.2.4/strideutils/cryptography.py

@@ -0,0 +1,37 @@
+import base64
+from typing import Tuple
+
+from Crypto.Cipher import AES
+from Crypto.Random import get_random_bytes
+from Crypto.Util.Padding import pad, unpad
+
+
+def encrypt_message(message: str, key: bytes) -> Tuple[str, str]:
+    """Encrypt the message using AES in CBC mode."""
+    # Generate a random initialization vector
+    iv = get_random_bytes(AES.block_size)
+
+    # Initialize the cipher for encryption
+    cipher_encrypt = AES.new(key, AES.MODE_CBC, iv)
+
+    # Encrypt the message
+    # The message needs to be padded to make sure its length is a multiple of the block size
+    encrypted = cipher_encrypt.encrypt(pad(message.encode(), AES.block_size))
+
+    # Return the encrypted data and the IV
+    return base64.b64encode(encrypted).decode(), base64.b64encode(iv).decode()
+
+
+def decrypt_message(encrypted: str, iv: str, key: bytes) -> str:
+    """Decrypt the message using AES in CBC mode."""
+    # Decode the encrypted message and IV from base64
+    encrypted = base64.b64decode(encrypted)
+    iv = base64.b64decode(iv)
+
+    # Initialize the cipher for decryption
+    cipher_decrypt = AES.new(key, AES.MODE_CBC, iv)
+
+    # Decrypt the message and unpad it
+    decrypted = unpad(cipher_decrypt.decrypt(encrypted), AES.block_size).decode()
+
+    return decrypted

{strideutils-0.2.2 → strideutils-0.2.4}/strideutils/stride_requests.py

@@ -538,7 +538,7 @@ def get_account_info(
     return request(url)
 
 
-def get_txs(event_filters: List[str], api_endpoint: str = config.stride.api_endpoint) -> Dict:
+def get_txs(event_filters: List[str], api_endpoint: str = config.stride.api_endpoint, limit=None) -> Dict:
     """
     Queries the txs endpoint with event filters
     Event filters are of the form `{event_type}.{event_attribute}={attribute_value}`
@@ -559,6 +559,8 @@ def get_txs(event_filters: List[str], api_endpoint: str = config.stride.api_endp
 
     url = f"{api_endpoint}/{TXS_QUERY}"
     params: List[Tuple[str, str]] = [("events", event_filter) for event_filter in event_filters]
+    if limit is not None:
+        params.append(('limit', limit))
     txs = request(url, params=params)
 
     return txs
@@ -625,6 +627,7 @@ def get_block_height_in_past(
     seconds_per_block: int = 6,
     prev_time: Optional[str] = None,
     verbose: bool = False,
+    rpc_endpoint: str = config.stride.library_rpc,
 ) -> int:
     """
     Gets the block height at a specified time by binary searching at different heights
@@ -638,14 +641,14 @@ def get_block_height_in_past(
     Returns:
         The block height at the specified time
     """
-    latest_block = get_latest_block()
+    latest_block = get_latest_block(rpc_endpoint=rpc_endpoint)
 
     # set parameters so we can search for the block height
-    blocks_per_day = 14415
+    blocks_per_day = (24 * 60 * 60) // seconds_per_block
     if prev_time is not None:
         if type(prev_time) is str:
             prev_time: datetime.datetime = pd.to_datetime(prev_time)  # type: ignore
-        ts = (datetime.datetime.now() - cast(datetime.datetime, prev_time)).total_seconds()
+        ts = (datetime.datetime.utcnow() - cast(datetime.datetime, prev_time)).total_seconds()
         num_days_back = ts / (60 * 60 * 24)
         time_in_past = datetime.timedelta(seconds=ts)
     else:
@@ -658,7 +661,7 @@ def get_block_height_in_past(
     # do an iterative search to find the block
     while abs(time_estimate_error) > max_acceptable_error_seconds:
         backwards_looking_height = latest_block - blocks_in_past
-        backwards_looking_time = get_block_time(backwards_looking_height, rpc_endpoint=config.stride.library_rpc)
+        backwards_looking_time = get_block_time(backwards_looking_height, rpc_endpoint=rpc_endpoint)
         time_estimate_error = (current_time - backwards_looking_time - time_in_past).total_seconds()
         error_in_blocks = time_estimate_error / seconds_per_block
         blocks_in_past -= int(error_in_blocks)