reafactor: formatting and remove uselss libs

Author: S1M0N38

.gitignore

@@ -2,3 +2,4 @@ __pycache__
 Balatro
 gamestate_cache
 .envrc
+.luarc.json

balatrobot.lua

@@ -15,7 +15,7 @@
 ---@type BalatrobotConfig
 BALATRO_BOT_CONFIG = {
 	enabled = true,
-	port = "12345",
+	port = "12346",
 	dt = 1.0 / 60.0,
 	uncap_fps = false,
 	instant_move = false,
@@ -59,4 +59,4 @@ sendDebugMessage("Logger loaded", "BALATROBOT")
 BalatrobotAPI.init()
 sendDebugMessage("API loaded", "BALATROBOT")
 
-sendInfoMessage("BalatroBot loaded - version " .. SMODS.current_mod.version, "BALATROBOT")
+sendInfoMessage("BalatroBot loaded - version " .. SMODS.current_mod.version, "BALATROBOT")

bot.py

@@ -1,13 +1,15 @@
 #!/usr/bin/python3
 
-import sys
 import json
+import random
 import socket
+import subprocess
+import sys
 import time
 from enum import Enum
+from weakref import proxy
+
 from gamestates import cache_state
-import subprocess
-import random
 
 
 class State(Enum):
@@ -72,7 +74,7 @@ def __init__(
 
         self.bot_port = bot_port
 
-        self.addr = ("localhost", self.bot_port)
+        self.addr = ("127.0.0.1", self.bot_port)
         self.running = False
         self.balatro_instance = None
 
@@ -131,7 +133,7 @@ def stop_balatro_instance(self):
 
     def sendcmd(self, cmd, **kwargs):
         msg = bytes(cmd, "utf-8")
-        self.sock.sendto(msg, self.addr)
+        self.sock.send(msg)
 
     def actionToCmd(self, action):
         result = []
@@ -148,15 +150,15 @@ def actionToCmd(self, action):
 
     def verifyimplemented(self):
         try:
-            self.skip_or_select_blind(self, {})
-            self.select_cards_from_hand(self, {})
-            self.select_shop_action(self, {})
-            self.select_booster_action(self, {})
-            self.sell_jokers(self, {})
-            self.rearrange_jokers(self, {})
-            self.use_or_sell_consumables(self, {})
-            self.rearrange_consumables(self, {})
-            self.rearrange_hand(self, {})
+            self.skip_or_select_blind({})
+            self.select_cards_from_hand({})
+            self.select_shop_action({})
+            self.select_booster_action({})
+            self.sell_jokers({})
+            self.rearrange_jokers({})
+            self.use_or_sell_consumables({})
+            self.rearrange_consumables({})
+            self.rearrange_hand({})
         except NotImplementedError as e:
             print(e)
             sys.exit(0)
@@ -168,11 +170,13 @@ def random_seed(self):
         return "".join(random.choices("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ", k=7))
 
     def chooseaction(self):
+        print("Choosing action based on game state:", self.G["state"])
         if self.G["state"] == State.GAME_OVER:
             self.running = False
 
         match self.G["waitingFor"]:
             case "start_run":
+                print("Starting run with deck:", self.deck)
                 seed = self.seed
                 if seed is None:
                     seed = self.random_seed()
@@ -184,23 +188,32 @@ def chooseaction(self):
                     self.challenge,
                 ]
             case "skip_or_select_blind":
-                return self.skip_or_select_blind(self, self.G)
+                print("Choosing action: skip_or_select_blind")
+                return self.skip_or_select_blind(self.G)
             case "select_cards_from_hand":
-                return self.select_cards_from_hand(self, self.G)
+                print("Choosing action: select_cards_from_hand")
+                return self.select_cards_from_hand(self.G)
             case "select_shop_action":
-                return self.select_shop_action(self, self.G)
+                print("Choosing action: select_shop_action")
+                return self.select_shop_action(self.G)
             case "select_booster_action":
-                return self.select_booster_action(self, self.G)
+                print("Choosing action: select_booster_action")
+                return self.select_booster_action(self.G)
             case "sell_jokers":
-                return self.sell_jokers(self, self.G)
+                print("Choosing action: sell_jokers")
+                return self.sell_jokers(self.G)
             case "rearrange_jokers":
-                return self.rearrange_jokers(self, self.G)
+                print("Choosing action: rearrange_jokers")
+                return self.rearrange_jokers(self.G)
             case "use_or_sell_consumables":
-                return self.use_or_sell_consumables(self, self.G)
+                print("Choosing action: use_or_sell_consumables")
+                return self.use_or_sell_consumables(self.G)
             case "rearrange_consumables":
-                return self.rearrange_consumables(self, self.G)
+                print("Choosing action: rearrange_consumables")
+                return self.rearrange_consumables(self.G)
             case "rearrange_hand":
-                return self.rearrange_hand(self, self.G)
+                print("Choosing action: rearrange_hand")
+                return self.rearrange_hand(self.G)
 
     def run_step(self):
         if self.sock is None:

bot_example.py

@@ -94,4 +94,5 @@ def rearrange_hand(self, G):
     mybot.rearrange_consumables = rearrange_consumables
     mybot.rearrange_hand = rearrange_hand
 
+    mybot.running = True
     mybot.run()

my_first_bot.py

@@ -0,0 +1,55 @@
+# my_first_bot.py
+from bot import Actions, Bot
+
+
+class MyFirstBot(Bot):
+    def __init__(self, deck="Red Deck", stake=1):
+        super().__init__(deck=deck, stake=stake, seed="EXAMPLE")
+        self.round_count = 0
+
+    def skip_or_select_blind(self, G):
+        """Always select blinds to play them"""
+        return [Actions.SELECT_BLIND]
+
+    def select_cards_from_hand(self, G):
+        """Simple strategy: play the first card"""
+        return [Actions.PLAY_HAND, [1]]
+
+    def select_shop_action(self, G):
+        """Always leave the shop immediately"""
+        return [Actions.END_SHOP]
+
+    def select_booster_action(self, G):
+        """Skip all booster packs"""
+        return [Actions.SKIP_BOOSTER_PACK]
+
+    def sell_jokers(self, G):
+        """Don't sell any jokers"""
+        return [Actions.SELL_JOKER, []]
+
+    def rearrange_jokers(self, G):
+        """Don't rearrange jokers"""
+        return [Actions.REARRANGE_JOKERS, []]
+
+    def use_or_sell_consumables(self, G):
+        """Don't use consumables"""
+        return [Actions.USE_CONSUMABLE, []]
+
+    def rearrange_consumables(self, G):
+        """Don't rearrange consumables"""
+        return [Actions.REARRANGE_CONSUMABLES, []]
+
+    def rearrange_hand(self, G):
+        """Don't rearrange hand"""
+        return [Actions.REARRANGE_HAND, []]
+
+
+# Run the bot
+if __name__ == "__main__":
+    bot = MyFirstBot()
+    bot.running = True
+    print("Starting MyFirstBot...")
+    print("Running: ", bot.running)
+    while True:
+        input("Press Enter to simulate a game step...")
+        bot.run_step()