SEND_GAMESTATE API

Author: phughesion

client.py

@@ -2,10 +2,43 @@
 # WIP
 
 import socket
+import sys
+import signal
+
+def signal_handler(signal, frame):
+    sys.exit(0)
+
+def is_socket_closed(sock: socket.socket):
+    try:
+        data = sock.recv(16, socket.MSG_DONTWAIT | socket.MSG_PEEK)
+        if len(data) == 0:
+            return True
+    except BlockingIOError:
+        return False
+    except ConnectionResetError:
+        return True
+    except Exception as e:
+        return False
+    return False
 
 if __name__ == "__main__":
+    signal.signal(signal.SIGINT, signal_handler)
+
     client_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
     client_socket.settimeout(1.0)
 
     addr = ("127.0.0.1", 12345)
 
+    client_socket.sendto(b'SEND_GAMESTATE', addr)
+
+    running = True
+    while running:
+        
+        try:
+            data, addr = client_socket.recvfrom(1024)
+            print(data.decode())
+        except:
+            print('Request Timed Out')
+
+        running = False
+

src/api.lua

@@ -37,14 +37,19 @@ function BalatrobotAPI.update(dt)
     data, msg_or_ip, port_or_nil = BalatrobotAPI.socket:receivefrom()
 	if data then
 
-        local _action = Utils.parseaction(data)
-
-        if _action and #_action > 1 and #_action > Bot.ACTIONPARAMS[_action[1]].num_args then
-            BalatrobotAPI.respond("Error: Incorrect number of params for action " .. action)
-        elseif not _action then
-            BalatrobotAPI.respond("Error: Incorrect message format. Should be ACTION|arg1|arg2")
+        if data == 'SEND_GAMESTATE' then
+            sendDebugMessage("SEND_GAMESTATE")
+            BalatrobotAPI.notifyapiclient()
         else
-            BalatrobotAPI.queueaction(_action)
+            local _action = Utils.parseaction(data)
+
+            if _action and #_action > 1 and #_action > Bot.ACTIONPARAMS[_action[1]].num_args then
+                BalatrobotAPI.respond("Error: Incorrect number of params for action " .. action)
+            elseif not _action then
+                BalatrobotAPI.respond("Error: Incorrect message format. Should be ACTION|arg1|arg2")
+            else
+                BalatrobotAPI.queueaction(_action)
+            end
         end
 
 	elseif msg_or_ip ~= 'timeout' then

src/bot.lua

@@ -102,7 +102,6 @@ Bot.ACTIONPARAMS[Bot.ACTIONS.START_RUN] = {
     func = "start_run"
 }
 
-
 -- CHANGE ME
 Bot.SETTINGS = {
     stake = 1,
@@ -117,10 +116,10 @@ Bot.SETTINGS = {
     action_delay = 0,
 
     -- Replay actions from file?
-    replay = true,
+    replay = false,
 
     -- Receive commands from the API?
-    api = false,
+    api = true,
 }
 
 --- Skips or selects the current blind

src/utils.lua

@@ -17,9 +17,11 @@ end
 function Utils.getHandData()
     local _hand = { }
 
-    for i = 1, #G.hand.cards do
-        local _card = Utils.getCardData(G.hand.cards[i])
-        _hand[i] = _card
+    if G and G.hand and G.hand.cards then
+        for i = 1, #G.hand.cards do
+            local _card = Utils.getCardData(G.hand.cards[i])
+            _hand[i] = _card
+        end
     end
 
     return _hand
@@ -28,9 +30,11 @@ end
 function Utils.getJokersData()
     local _jokers = { }
 
-    for i = 1, #G.jokers.cards do
-        local _card = Utils.getCardData(G.jokers.cards[i])
-        _jokers[i] = _card
+    if G and G.jokers and G.jokers.cards then
+        for i = 1, #G.jokers.cards do
+            local _card = Utils.getCardData(G.jokers.cards[i])
+            _jokers[i] = _card
+        end
     end
 
     return _jokers
@@ -39,9 +43,11 @@ end
 function Utils.getConsumablesData()
     local _consumables = { }
 
-    for i = 1, #G.consumeables.cards do
-        local _card = Utils.getCardData(G.consumeables.cards[i])
-        _consumables[i] = _card
+    if G and G.consumables and G.consumables.cards then
+        for i = 1, #G.consumeables.cards do
+            local _card = Utils.getCardData(G.consumeables.cards[i])
+            _consumables[i] = _card
+        end
     end
 
     return _consumables
@@ -68,7 +74,7 @@ end
 
 function Utils.getShopData()
     local _shop = { }
-    if not G.shop then return _shop end
+    if not G or not G.shop then return _shop end
 
     _shop.reroll_cost = G.GAME.current_round.reroll_cost
     _shop.cards = { }