from aoe.types import *



class Production():


    # assumes for now each unit can be built in only one type of building
    def __init__(self):
        self.building = {
                         Units.VILLAGER: Buildings.TOWNCENTER,
                         Units.TRADECART: Buildings.MARKET,
                         Units.SPEARMAN: Buildings.BARRACKS,
                         Units.MILITIA: Buildings.BARRACKS,
                         Units.SCOUT: Buildings.STABLE,
                         Units.KNIGHT: Buildings.STABLE,
                         Units.SKIRMISHER: Buildings.ARCHERYRANGE,
                         Units.ARCHER: Buildings.ARCHERYRANGE,
                        }