squirrel-battle/squirrelbattle/entities/friendly.py

51 lines
1.5 KiB
Python
Raw Normal View History

from ..interfaces import FriendlyEntity, InventoryHolder
2020-12-01 16:12:22 +00:00
from ..translations import gettext as _
from .player import Player
from .items import Item
from random import choice
2020-12-01 16:12:22 +00:00
class Merchant(InventoryHolder, FriendlyEntity):
"""
The class for merchants in the dungeon
"""
def keys(self) -> list:
"""
Returns a friendly entitie's specific attributes
"""
2020-12-09 14:32:37 +00:00
return super().keys() + ["inventory", "hazel"]
2020-12-07 20:13:55 +00:00
def __init__(self, name: str = "merchant", inventory: list = None,
hazel: int = 75, *args, **kwargs):
super().__init__(name=name, *args, **kwargs)
self.inventory = self.translate_inventory(inventory or [])
self.hazel = hazel
2020-12-07 20:48:56 +00:00
if not self.inventory:
for i in range(5):
self.inventory.append(choice(Item.get_all_items())())
2020-12-07 20:22:06 +00:00
def talk_to(self, player: Player) -> str:
"""
This function is used to open the merchant's inventory in a menu,
and allow the player to buy/sell objects
"""
2020-12-07 20:13:55 +00:00
return _("I don't sell any squirrel")
2020-12-07 20:22:06 +00:00
def change_hazel_balance(self, hz: int) -> None:
"""
Change the number of hazel the merchant has by hz.
"""
self.hazel += hz
2020-12-07 20:48:56 +00:00
2020-12-07 20:22:06 +00:00
class Sunflower(FriendlyEntity):
"""
A friendly sunflower
"""
2020-12-01 16:12:22 +00:00
dialogue_option = [_("Flower power!!"), _("The sun is warm today")]
2020-12-07 20:22:06 +00:00
def __init__(self, maxhealth: int = 15,
*args, **kwargs) -> None:
2020-12-07 20:22:06 +00:00
super().__init__(name="sunflower", maxhealth=maxhealth, *args, **kwargs)