fixed issue with ship singleton manager not always returning the righ…

…t ship
Ctri-The-Third · Apr 11, 2024 · 97d7f88 · 97d7f88
1 parent 2df8671
commit 97d7f88
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/straders_sdk/client_mediator.py b/straders_sdk/client_mediator.py
@@ -228,9 +228,9 @@ def ships_view(self, force=False) -> dict[str, Ship] or SpaceTradersResponse:
             if resp:
                 self.ships = self.ships | resp
                 for ship in self.ships.values():
-                    
+
                     ship = SingletonShips().add_ship(ship)
-                    
+
                 return resp
         start = datetime.now()
         resp = self.api_client.ships_view()
@@ -258,8 +258,7 @@ def ships_view_one(self, symbol: str, force=False):
             resp = self.db_client.ships_view_one(symbol)
             if resp:
                 resp: Ship
-                SingletonShips().add_ship(resp)
-
+                resp = SingletonShips().add_ship(resp)
 
                 return resp
         start = datetime.now()
@@ -270,7 +269,7 @@ def ships_view_one(self, symbol: str, force=False):
         if resp:
             resp: Ship
             resp.dirty = True
-            SingletonShips().add_ship(resp)
+            resp = SingletonShips().add_ship(resp)
 
             self.db_client.update(resp)
         return resp
@@ -289,7 +288,7 @@ def ships_purchase(
         self.set_connections()
         start = datetime.now()
         resp = self.api_client.ships_purchase(ship_type, waypoint)
-        resp[0] = SingletonShips().add_ship(resp[0])
+        resp = (SingletonShips().add_ship(resp[0]), resp[1])
         self.logging_client.ships_purchase(
             ship_type, waypoint, resp, (datetime.now() - start).total_seconds()
         )