Add price mixing

backend/config.yaml

@@ -34,6 +34,7 @@ bots:
   max_price_change: 20
   expiration_ticks: 5
   price_change_coeff: 0.05
+  dataset_price_weight: 0.5
 
 max_energy_per_player: 0.2
 

backend/db/migration.py

@@ -174,6 +174,11 @@ async def run_migrations():
                 biomass INT NOT NULL,
                 gas INT NOT NULL,
                 oil INT NOT NULL,
+                coal_price INT NOT NULL,
+                uranium_price INT NOT NULL,
+                biomass_price INT NOT NULL,
+                gas_price INT NOT NULL,
+                oil_price INT NOT NULL,
                 geothermal INT NOT NULL,
                 wind INT NOT NULL,
                 solar INT NOT NULL,
@@ -230,7 +235,12 @@ async def run_migrations():
                                      solar=row["SOLAR"],
                                      hydro=row["HYDRO"],
                                      energy_demand=row["ENERGY_DEMAND"],
-                                     max_energy_price=row["MAX_ENERGY_PRICE"]
+                                     max_energy_price=row["MAX_ENERGY_PRICE"],
+                                     coal_price=row["COAL_PRICE"],
+                                     uranium_price=row["URANIUM_PRICE"],
+                                     biomass_price=row["BIOMASS_PRICE"],
+                                     gas_price=row["GAS_PRICE"],
+                                     oil_price=row["OIL_PRICE"]
                                      )
             i += 1
         logger.info(f"Added dataset {x}")

backend/game/bots/resource_bot.py

@@ -49,6 +49,9 @@ async def run(self, tick_data: TickData):
         orders = await self.get_last_orders()
 
         for resource in Resource:
+            if resource == Resource.energy:
+                continue
+
             resource_orders = orders[resource]
             resource_sum = resources_sum[resource]
             buy_price = self.buy_prices[resource]
@@ -79,11 +82,19 @@ async def run(self, tick_data: TickData):
             if buy_price == sell_price:
                 buy_price = sell_price - 1
 
+            buy_price = self.mix_dataset_price(
+                tick_data.dataset_row, buy_price, resource)
+            sell_price = self.mix_dataset_price(
+                tick_data.dataset_row, sell_price, resource)
+
             await self.create_orders(tick_data.game.current_tick,
                                      resource, buy_price, sell_price, buy_volume, sell_volume)
             self.buy_prices[resource] = buy_price
             self.sell_prices[resource] = sell_price
 
+    def mix_dataset_price(self, dataset_row, price, resource):
+        return config['bots']['dataset_price_weight'] * dataset_row[resource.name.lower() + "_price"] + (1 - config['bots']['dataset_price_weight']) * price
+
     def get_filled_perc(self, orders: List[Order]):
         size = {side: 0 for side in OrderSide}
         filled_size = {side: 0 for side in OrderSide}

backend/model/dataset_data.py

@@ -22,6 +22,11 @@ class DatasetData(Table):
     wind: int
     solar: int
     hydro: int
+    coal_price: int
+    uranium_price: int
+    biomass_price: int
+    gas_price: int
+    oil_price: int
     energy_demand: int
     max_energy_price: int