reading fucking cells

econ/agents/price_believe_distribute.py

@@ -19,11 +19,21 @@ class Price_Believe_Distribiute_Agent(Base_Distribution_Agent):
         order_error=self.target_error()
         if order_error>0:
             # aquire based on current price belive
-            order=self.distribute_resource(self.price_believe,order_error,0)
+            cx_id=self.select_best_cx()
+            order=self.distribute_resource(self.price_believe,order_error,cx_id)
             self.register_order(0,order)
         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({
@@ -66,10 +76,24 @@ class Price_Believe_Distribiute_Agent(Base_Distribution_Agent):
         
 
 
-    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()