balatrobot 0.7.3__py3-none-any.whl → 0.7.4__py3-none-any.whl

balatrobot/__init__.py

@@ -5,7 +5,7 @@ from .enums import Actions, Decks, Stakes, State
 from .exceptions import BalatroError
 from .models import G
 
-__version__ = "0.7.3"
+__version__ = "0.7.4"
 __all__ = [
     # Main client
     "BalatroClient",

balatrobot/client.py

@@ -6,6 +6,7 @@ import platform
 import re
 import shutil
 import socket
+import time
 from pathlib import Path
 from typing import Self
 
@@ -186,6 +187,9 @@ class BalatroClient:
         logger.debug(f"Sending API request: {name}")
 
         try:
+            # Start timing measurement
+            start_time = time.perf_counter()
+
             # Send request
             message = request.model_dump_json() + "\n"
             self._socket.send(message.encode())
@@ -216,6 +220,18 @@ class BalatroClient:
             logger.debug(f"API request {name} completed successfully")
             return response_data
 
+        except socket.timeout as e:
+            # Calculate elapsed time and log timeout
+            elapsed_time = time.perf_counter() - start_time
+            logger.warning(
+                f"Timeout on API request {name}: took {elapsed_time:.3f}s, "
+                f"exceeded timeout of {self.timeout}s (port: {self.port})"
+            )
+            raise ConnectionFailedError(
+                f"Socket timeout during communication: {e}",
+                error_code="E008",
+                context={"error": str(e), "elapsed_time": elapsed_time},
+            ) from e
         except socket.error as e:
             logger.error(f"Socket error during API request {name}: {e}")
             raise ConnectionFailedError(

{balatrobot-0.7.3.dist-info → balatrobot-0.7.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: balatrobot
-Version: 0.7.3
+Version: 0.7.4
 Summary: A framework for Balatro bot development
 Project-URL: Homepage, https://github.com/coder/balatrobot
 Project-URL: Issues, https://github.com/coder/balatrobot/issues

balatrobot-0.7.4.dist-info/RECORD

@@ -0,0 +1,10 @@
+balatrobot/__init__.py,sha256=sn-gdztWJAELq4WCWhtOQVIu3r4fguVwb9YP6JEn4TU,406
+balatrobot/client.py,sha256=pXPKmTh3hQpt0wMQG9wZSNFD47x-pwHbqspndjApAcE,18664
+balatrobot/enums.py,sha256=UzlrljX1x_UlbuqR87c-KrCQjS74DGpG8A0zH0G-h98,22463
+balatrobot/exceptions.py,sha256=fRupkYMw4PuMCJd8SFjfBDlla8wiKwhVzLj8AxmVL9I,4046
+balatrobot/models.py,sha256=me9-qN1LQtnWfReC8tF8DFR38d5hBBSJD1SfUF3d2b4,14996
+balatrobot/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+balatrobot-0.7.4.dist-info/METADATA,sha256=iFVGpa27v74dlCNe9ix5qEHTVq9WDjYCL-6GpFUp_EM,2801
+balatrobot-0.7.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+balatrobot-0.7.4.dist-info/licenses/LICENSE,sha256=71EXhU7CSe-Cihhj_VVxLtgVnSOaavHqVoixPKtE7Bk,1064
+balatrobot-0.7.4.dist-info/RECORD,,

balatrobot-0.7.3.dist-info/RECORD

@@ -1,10 +0,0 @@
-balatrobot/__init__.py,sha256=VUudB2uHb5OSd1vqnKW17Ir6iU0ecIWe20LtXHLQDqw,406
-balatrobot/client.py,sha256=qx9EMkZEbZahd6wirontmxslmwejgdMTUW87GCzcYh8,17993
-balatrobot/enums.py,sha256=UzlrljX1x_UlbuqR87c-KrCQjS74DGpG8A0zH0G-h98,22463
-balatrobot/exceptions.py,sha256=fRupkYMw4PuMCJd8SFjfBDlla8wiKwhVzLj8AxmVL9I,4046
-balatrobot/models.py,sha256=me9-qN1LQtnWfReC8tF8DFR38d5hBBSJD1SfUF3d2b4,14996
-balatrobot/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-balatrobot-0.7.3.dist-info/METADATA,sha256=k791DV7ZmTe_qpj5k62bSqLD2G7dfdLI4db3k8B1Uwg,2801
-balatrobot-0.7.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-balatrobot-0.7.3.dist-info/licenses/LICENSE,sha256=71EXhU7CSe-Cihhj_VVxLtgVnSOaavHqVoixPKtE7Bk,1064
-balatrobot-0.7.3.dist-info/RECORD,,