from model.building import Building
from model.trainer import Trainer


class CastleBuilding(Building, Trainer):
    def __init__(self, *args, **kwargs):
        super(CastleBuilding, self).__init__(*args, **kwargs)

    def give_name(self):
        return "{0}'s Castle".format(self.owner)

    def max_health(self):
        return 100

    def trainable_units(self):
        return {
            'slave': {
                'trainspeed': 20,
                'resources': {
                    'gold': 50,
                    'wood': 0,
                    'stone': 0
                }
            }
        }
