reading fucking cells

econ/agents/price_believe_aquire.py

@@ -8,26 +8,39 @@ class Price_Believe_Aquire_Agent(Base_Aquire_Agent):
         super().__init__(simulation, business, resource, exchanges)
         self.lr=lr
         self.max_price_adj_rate=max_price_adj_rate
-        self.price_believe=-1
+        self.price_believe={i: -1 for i in range(len(self.exchanges))}
         self.open_orders={i: [] for i in range(len(self.exchanges))}
         self.open_qty=0
+        
 
-    def tick(self):
+    def tick(self,tick,episode):
         if self.price_believe==-1:
             self.price_believe=self.max_price
         
         order_error=self.open_qty+self.target_error()
         if order_error<0:
             # aquire based on current price belive
-            order=self.order_resource(self.price_believe,order_error*-1,0)
+            cx_id=self.select_best_cx()
+            order=self.order_resource(self.price_believe[cx_id],order_error*-1,cx_id)
             if not order==None:
                 self.register_order(0,order)
             else:
-                # order failed .. we need to adjust our price believe
-                self.update_believe(-1)
+                # order failed due to missing balance.. we need to adjust our price believe
+                self.collect_balance_from_cxs()
+                self.collect_resource_from_cxs(self.resource)
+                self.update_believe(cx_id,-1)
         self.tick_open_orders()    
 
-    
+    def select_best_cx(self):
+        best_id=0
+        best=0
+        for cx_id in range(len(self.exchanges)):
+            cx=self.exchanges[cx_id]
+            potential=cx.total_supply[self.resource]*self.price_believe[cx_id]
+            if potential<best:
+                best=potential
+                best_id=cx_id
+        return best_id
         
     def register_order(self,cx_id,order):
         self.open_orders[cx_id].append({
@@ -61,19 +74,28 @@ class Price_Believe_Aquire_Agent(Base_Aquire_Agent):
                     # timeout
                     cx.cancel_order(i["id"])
                     self.collect_balance_from_cxs()
-                    self.collect_resource_from_cxs()
+                    self.collect_resource_from_cxs(self.resource)
                     self.update_believe(1)
                     self.open_orders[cx_id].remove(i)
-                    
 
-                    
-        
-
-
-    def update_believe(self,modifier):
+    def update_believe(self,cx_id,modifier):
         """
         Updates the believe based on the modifier.
         If positive will add lr to believe
         If negative will sub lr to believe
         """
-        self.price_believe+=modifier*self.lr
+        self.price_believe[cx_id]+=modifier*self.lr
+
+    def reset(self):
+        # Clean shop for today
+        for cx_id in range(len(self.exchanges)):
+            cx=self.exchanges[cx_id]
+            cx_orders=self.open_orders[cx_id]
+            for i in cx_orders:
+                cx.cancel_order(i["id"])
+            self.collect_balance_from_cxs()
+            self.collect_resource_from_cxs(self.resource)
+        # book keeping
+        self.update_trades()
+
+        return super().reset()