FoBot/modules/deeptownOptimizer/optimizer.py

import copy
import datetime
import json
import os


class Optimizer():
    def __init__(self):
        self.mines = {}
        self.items = {}
        # get mine stats
        with open(os.path.join("modules", "deeptownOptimizer", "mines.json")) as mines:
            self.mines = json.load(mines)
        # get items stats
        with open(os.path.join("modules", "deeptownOptimizer", "items.json")) as items:
            self.items = json.load(items)
        # Add zero values to mine stat
        ores = self.mines["0"].keys()
        for area, stats in self.mines.items():
            for ore in ores:
                if self.mines[area].get(ore) is None:
                    self.mines[area].update({ore: 0})

    def best_mines(self, ore):
        if ore not in self.mines["0"].keys():
            raise ValueError("{ore} is not a correct ore.".format(ore=ore))
        ordered_mines = [(k, v) for k, v in self.mines.items()]
        ordered_mines.sort(key=lambda x: x[1][ore], reverse=True)
        return ordered_mines

    def to_make(self, item, quantity=1):
        if item not in self.items.keys():
            raise ValueError("{item} is not a correct item.".format(item=item))
        if self.items[item]["quantity"] != 0:
            number_of_craft = int(quantity / self.items[item]["quantity"])
        else:
            number_of_craft = int(quantity)
        if number_of_craft % 1 != 0:
            number_of_craft = int((number_of_craft // 1) + 1)
        time = self.items[item]["time"] * number_of_craft
        value = self.items[item]["value"] * number_of_craft * self.items[item]["quantity"]
        needed = {}
        for resource, quantity in self.items[item]["needed"].items():
            needed.update({resource: quantity * number_of_craft})
        return {"time": time, "value": value, "needed": needed}

    def recursive_to_make(self, item, quantity=1):
        if item in self.items.keys():
            needed = self.to_make(item, quantity)
            results = [(self.items[item]["type"], item, quantity, needed["time"])]
            for needed_item, needed_quantity in needed["needed"].items():
                needed_result = self.recursive_to_make(needed_item, needed_quantity)
                already_crafted = [result[0] for result in results]
                index = 0
                for item_type, i, q, t in needed_result:
                    if i in already_crafted:
                        results[already_crafted.index(i)] = (
                            results[already_crafted.index(i)][0],
                            results[already_crafted.index(i)][1],
                            results[already_crafted.index(i)][2] + q,
                            results[already_crafted.index(i)][3] + t
                        )
                    else:
                        results.append((item_type, i, q, t))
                    index += 1
            return results
        else:
            return [(self.items[item]["type"], item, quantity, 0)]