algokit-utils 2.3.3b2__py3-none-any.whl → 2.4.0__py3-none-any.whl

algokit_utils/_debugging.py

@@ -27,7 +27,7 @@ SOURCES_FILE = "sources.avm.json"
 TRACES_FILE_EXT = ".trace.avm.json"
 DEBUG_TRACES_DIR = "debug_traces"
 TEAL_FILE_EXT = ".teal"
-TEAL_SOURCEMAP_EXT = ".teal.tok.map"
+TEAL_SOURCEMAP_EXT = ".teal.map"
 
 
 @dataclass
@@ -107,37 +107,6 @@ def _load_or_create_sources(sources_path: Path) -> AVMDebuggerSourceMap:
         return AVMDebuggerSourceMap.from_dict(json.load(f))
 
 
-def _upsert_debug_sourcemaps(sourcemaps: list[AVMDebuggerSourceMapEntry], project_root: Path) -> None:
-    """
-    This function updates or inserts debug sourcemaps. If path in the sourcemap during iteration leads to non
-    existing file, removes it. Otherwise upserts.
-
-    Args:
-        sourcemaps (list[AVMDebuggerSourceMapEntry]): A list of AVMDebuggerSourceMapEntry objects.
-        project_root (Path): The root directory of the project.
-
-    Returns:
-        None
-    """
-
-    sources_path = project_root / ALGOKIT_DIR / SOURCES_DIR / SOURCES_FILE
-    sources = _load_or_create_sources(sources_path)
-
-    for sourcemap in sourcemaps:
-        source_file_path = Path(sourcemap.location)
-        if not source_file_path.exists() and sourcemap in sources.txn_group_sources:
-            sources.txn_group_sources.remove(sourcemap)
-        elif source_file_path.exists():
-            if sourcemap not in sources.txn_group_sources:
-                sources.txn_group_sources.append(sourcemap)
-            else:
-                index = sources.txn_group_sources.index(sourcemap)
-                sources.txn_group_sources[index] = sourcemap
-
-    with sources_path.open("w") as f:
-        json.dump(sources.to_dict(), f)
-
-
 def _write_to_file(path: Path, content: str) -> None:
     path.parent.mkdir(parents=True, exist_ok=True)
     path.write_text(content)
@@ -174,8 +143,29 @@ def _build_avm_sourcemap(  # noqa: PLR0913
     return AVMDebuggerSourceMapEntry(str(source_map_output_path), program_hash)
 
 
+def cleanup_old_trace_files(output_dir: Path, buffer_size_mb: float) -> None:
+    """
+    Cleanup old trace files if total size exceeds buffer size limit.
+
+    Args:
+        output_dir (Path): Directory containing trace files
+        buffer_size_mb (float): Maximum allowed size in megabytes
+    """
+    total_size = sum(f.stat().st_size for f in output_dir.glob("*") if f.is_file())
+    if total_size > buffer_size_mb * 1024 * 1024:
+        sorted_files = sorted(output_dir.glob("*"), key=lambda p: p.stat().st_mtime)
+        while total_size > buffer_size_mb * 1024 * 1024 and sorted_files:
+            oldest_file = sorted_files.pop(0)
+            total_size -= oldest_file.stat().st_size
+            oldest_file.unlink()
+
+
 def persist_sourcemaps(
-    *, sources: list[PersistSourceMapInput], project_root: Path, client: "AlgodClient", with_sources: bool = True
+    *,
+    sources: list[PersistSourceMapInput],
+    project_root: Path,
+    client: "AlgodClient",
+    with_sources: bool = True,
 ) -> None:
     """
     Persist the sourcemaps for the given sources as an AlgoKit AVM Debugger compliant artifacts.
@@ -187,7 +177,7 @@ def persist_sourcemaps(
         Default is True, as needed by an AlgoKit AVM debugger.
     """
 
-    sourcemaps = [
+    for source in sources:
         _build_avm_sourcemap(
             raw_teal=source.raw_teal,
             compiled_teal=source.compiled_teal,
@@ -197,10 +187,6 @@ def persist_sourcemaps(
             client=client,
             with_sources=with_sources,
         )
-        for source in sources
-    ]
-
-    _upsert_debug_sourcemaps(sourcemaps, project_root)
 
 
 def simulate_response(atc: AtomicTransactionComposer, algod_client: "AlgodClient") -> SimulateAtomicTransactionResponse:
@@ -257,24 +243,20 @@ def simulate_and_persist_response(
     response = simulate_response(atc_to_simulate, algod_client)
     txn_results = response.simulate_response["txn-groups"]
 
-    txn_types = [txn_result["txn-results"][0]["txn-result"]["txn"]["txn"]["type"] for txn_result in txn_results]
-    txn_types_count = {txn_type: txn_types.count(txn_type) for txn_type in set(txn_types)}
-    txn_types_str = "_".join([f"{count}#{txn_type}" for txn_type, count in txn_types_count.items()])
+    txn_types = [
+        txn["txn-result"]["txn"]["txn"]["type"] for txn_result in txn_results for txn in txn_result["txn-results"]
+    ]
+    txn_types_count = {}
+    for txn_type in txn_types:
+        if txn_type not in txn_types_count:
+            txn_types_count[txn_type] = txn_types.count(txn_type)
+    txn_types_str = "_".join([f"{count}{txn_type}" for txn_type, count in txn_types_count.items()])
 
     last_round = response.simulate_response["last-round"]
     timestamp = datetime.now(tz=timezone.utc).strftime("%Y%m%d_%H%M%S")
     output_file = project_root / DEBUG_TRACES_DIR / f"{timestamp}_lr{last_round}_{txn_types_str}{TRACES_FILE_EXT}"
 
     output_file.parent.mkdir(parents=True, exist_ok=True)
-
-    # cleanup old files if buffer size is exceeded
-    total_size = sum(f.stat().st_size for f in output_file.parent.glob("*") if f.is_file())
-    if total_size > buffer_size_mb * 1024 * 1024:
-        sorted_files = sorted(output_file.parent.glob("*"), key=lambda p: p.stat().st_mtime)
-        while total_size > buffer_size_mb * 1024 * 1024:
-            oldest_file = sorted_files.pop(0)
-            total_size -= oldest_file.stat().st_size
-            oldest_file.unlink()
-
+    cleanup_old_trace_files(output_file.parent, buffer_size_mb)
     output_file.write_text(json.dumps(response.simulate_response, indent=2))
     return response

{algokit_utils-2.3.3b2.dist-info → algokit_utils-2.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: algokit-utils
-Version: 2.3.3b2
+Version: 2.4.0
 Summary: Utilities for Algorand development for use by AlgoKit
 License: MIT
 Author: Algorand Foundation

{algokit_utils-2.3.3b2.dist-info → algokit_utils-2.4.0.dist-info}/RECORD

@@ -1,5 +1,5 @@
 algokit_utils/__init__.py,sha256=Xc9NVy3cETI2qzbkkAea2YrZ9-CLtZLHnuS1UHQ13wQ,4909
-algokit_utils/_debugging.py,sha256=HlO_mOsVTaz2aqci_4tOpGT-rAd-xZ1RxFphnTRgDLQ,10751
+algokit_utils/_debugging.py,sha256=ACNayBEjA5_Y91YAIg-4_h5ld0IyKfW7vqoxo1xcxGE,9873
 algokit_utils/_ensure_funded.py,sha256=ZdEdUB43QGIQrg7cSSgNrDmWaLSUhli9x9I6juwKfgo,6786
 algokit_utils/_transfer.py,sha256=R9q8RoMHiwtqcwQjuGHEluMxIzmYqAsI5WrTsQd24Ds,6021
 algokit_utils/account.py,sha256=JYovI84sxiU2mIbwNwWVuoEJBMuAPJauHx5jkKFrR7E,7067
@@ -18,7 +18,7 @@ algokit_utils/logic_error.py,sha256=I9fJJ09zfpPlKgcJJ7fqC77BBPTz37QsSlGfZwXDdPQ,
 algokit_utils/models.py,sha256=KMpSUv7XGjZ9_50U6qUjcPjGh9_7lsRj3ZlO9Fb2zyE,8421
 algokit_utils/network_clients.py,sha256=TlGRZ4l62_vi__AKg9zyVGiAawTrA0ca6AfPDzRL44E,5136
 algokit_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-algokit_utils-2.3.3b2.dist-info/LICENSE,sha256=J5i7U1Q9Q2c7saUzlvFRmrCCFhQyXb5Juz_LO5omNUw,1076
-algokit_utils-2.3.3b2.dist-info/METADATA,sha256=6yh9fbbT3vGZbBQMb8vwWeoKn-XO-3WiFp5AMNrlsTs,2258
-algokit_utils-2.3.3b2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-algokit_utils-2.3.3b2.dist-info/RECORD,,
+algokit_utils-2.4.0.dist-info/LICENSE,sha256=J5i7U1Q9Q2c7saUzlvFRmrCCFhQyXb5Juz_LO5omNUw,1076
+algokit_utils-2.4.0.dist-info/METADATA,sha256=cDRvMEzxKtLUWbnctWZP93OfdMUAwgk-rMlreSPikjI,2256
+algokit_utils-2.4.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+algokit_utils-2.4.0.dist-info/RECORD,,