squirrel-battle/squirrelbattle/entities/items.py

308 lines
9.0 KiB
Python
Raw Normal View History

2020-11-27 15:33:17 +00:00
# Copyright (C) 2020 by ÿnérant, eichhornchen, nicomarg, charlse
# SPDX-License-Identifier: GPL-3.0-or-later
2020-12-05 13:20:58 +00:00
from random import choice, randint
2020-11-11 15:47:19 +00:00
from typing import Optional
from ..interfaces import Entity, FightingEntity, Map, InventoryHolder
from ..translations import gettext as _
2020-11-06 14:33:26 +00:00
2020-10-23 14:51:48 +00:00
class Item(Entity):
"""
A class for items
"""
2020-11-06 14:33:26 +00:00
held: bool
held_by: Optional[InventoryHolder]
price: int
2020-10-23 14:51:48 +00:00
def __init__(self, held: bool = False,
held_by: Optional[InventoryHolder] = None,
2020-12-11 17:17:08 +00:00
price: int = 2, *args, **kwargs):
2020-11-06 14:33:26 +00:00
super().__init__(*args, **kwargs)
self.held = held
self.held_by = held_by
self.price = price
2020-11-06 14:33:26 +00:00
2020-12-04 15:53:27 +00:00
def drop(self) -> None:
"""
The item is dropped from the inventory onto the floor
"""
if self.held:
self.held_by.remove_from_inventory(self)
2020-12-11 17:33:47 +00:00
self.held_by.map.add_entity(self)
2020-12-04 15:53:27 +00:00
self.move(self.held_by.y, self.held_by.x)
self.held = False
self.held_by = None
2020-12-04 15:53:27 +00:00
def use(self) -> None:
"""
Indicates what should be done when the item is used.
"""
def equip(self, armor: bool = False) -> None:
2020-12-04 15:53:27 +00:00
"""
Indicates what should be done when the item is equipped.
"""
if armor:
if self.held_by.equipped_armor:
self.held_by.equipped_armor.unequip()
self.held_by.remove_from_inventory(self)
self.held_by.equipped_armor = self
else:
if self.held_by.equipped_item:
self.held_by.equipped_item.unequip()
self.held_by.remove_from_inventory(self)
self.held_by.equipped_item = self
2020-11-06 14:33:26 +00:00
2020-12-18 16:30:03 +00:00
def unequip(self) -> None:
"""
Indicates what should be done when the item is unequipped.
"""
self.held_by.add_to_inventory(self)
self.held_by.equipped_item = None
def hold(self, holder: InventoryHolder) -> None:
"""
The item is taken from the floor and put into the inventory
"""
2020-10-23 14:51:48 +00:00
self.held = True
2020-12-18 16:30:03 +00:00
self.held_by = holder
2020-12-11 17:33:47 +00:00
self.held_by.map.remove_entity(self)
2020-12-18 16:30:03 +00:00
holder.add_to_inventory(self)
2020-10-23 16:02:57 +00:00
2020-11-18 23:10:37 +00:00
def save_state(self) -> dict:
"""
Saves the state of the entity into a dictionary
"""
d = super().save_state()
d["held"] = self.held
2020-11-18 23:10:37 +00:00
return d
2020-12-07 20:13:55 +00:00
@staticmethod
def get_all_items() -> list:
return [BodySnatchPotion, Bomb, Heart, Shield, Sword]
def be_sold(self, buyer: InventoryHolder, seller: InventoryHolder) -> bool:
"""
Does all necessary actions when an object is to be sold.
2020-12-11 16:06:30 +00:00
Is overwritten by some classes that cannot exist in the player's
inventory
"""
2020-12-11 16:06:30 +00:00
if buyer.hazel >= self.price:
self.hold(buyer)
seller.remove_from_inventory(self)
buyer.change_hazel_balance(-self.price)
seller.change_hazel_balance(self.price)
return True
2020-12-11 16:06:30 +00:00
else:
return False
2020-11-06 14:33:26 +00:00
2020-11-11 15:23:27 +00:00
class Heart(Item):
"""
A heart item to return health to the player
"""
healing: int
2020-12-11 16:06:30 +00:00
def __init__(self, name: str = "heart", healing: int = 5, price: int = 3,
*args, **kwargs):
super().__init__(name=name, price=price, *args, **kwargs)
self.healing = healing
2020-11-11 15:47:19 +00:00
def hold(self, entity: InventoryHolder) -> None:
2020-11-11 15:47:19 +00:00
"""
When holding a heart, heal the player and don't put item in inventory.
"""
entity.health = min(entity.maxhealth, entity.health + self.healing)
entity.map.remove_entity(self)
2020-11-11 15:23:27 +00:00
2020-11-19 00:11:11 +00:00
def save_state(self) -> dict:
"""
Saves the state of the header into a dictionary
"""
d = super().save_state()
d["healing"] = self.healing
return d
2020-11-11 15:23:27 +00:00
2020-10-23 16:02:57 +00:00
class Bomb(Item):
"""
A bomb item intended to deal damage to enemies at long range
"""
2020-11-06 14:33:26 +00:00
damage: int = 5
exploding: bool
2020-12-18 16:30:03 +00:00
owner: Optional["InventoryHolder"]
2020-12-04 17:16:46 +00:00
tick: int
2020-10-23 16:02:57 +00:00
def __init__(self, name: str = "bomb", damage: int = 5,
exploding: bool = False, price: int = 4, *args, **kwargs):
super().__init__(name=name, price=price, *args, **kwargs)
self.damage = damage
self.exploding = exploding
self.tick = 4
2020-12-04 17:16:46 +00:00
self.owner = None
2020-10-23 16:02:57 +00:00
2020-12-04 15:53:27 +00:00
def use(self) -> None:
"""
When the bomb is used, throw it and explodes it.
"""
if self.held:
self.owner = self.held_by
2020-12-04 15:53:27 +00:00
super().drop()
self.exploding = True
2020-11-06 14:33:26 +00:00
def act(self, m: Map) -> None:
"""
Special exploding action of the bomb
"""
2020-10-23 16:02:57 +00:00
if self.exploding:
2020-12-04 15:53:27 +00:00
if self.tick > 0:
# The bomb will explode in <tick> moves
2020-12-04 15:53:27 +00:00
self.tick -= 1
else:
# The bomb is exploding.
# Each entity that is close to the bomb takes damages.
# The player earn XP if the entity was killed.
log_message = _("Bomb is exploding.")
2020-12-04 15:53:27 +00:00
for e in m.entities.copy():
if abs(e.x - self.x) + abs(e.y - self.y) <= 3 and \
2020-12-04 15:53:27 +00:00
isinstance(e, FightingEntity):
log_message += " " + e.take_damage(self, self.damage)
if e.dead:
self.owner.add_xp(randint(3, 7))
m.logs.add_message(log_message)
2020-12-04 15:53:27 +00:00
m.entities.remove(self)
2020-11-19 00:11:11 +00:00
# Add sparkles where the bomb exploded.
explosion = Explosion(y=self.y, x=self.x)
self.map.add_entity(explosion)
2020-11-19 00:11:11 +00:00
def save_state(self) -> dict:
"""
Saves the state of the bomb into a dictionary
"""
d = super().save_state()
d["exploding"] = self.exploding
d["damage"] = self.damage
return d
2020-12-05 13:20:58 +00:00
class Explosion(Item):
"""
When a bomb explodes, the explosion is displayed.
"""
def __init__(self, *args, **kwargs):
super().__init__(name="explosion", *args, **kwargs)
def act(self, m: Map) -> None:
"""
The explosion instant dies.
"""
m.remove_entity(self)
def hold(self, player: InventoryHolder) -> None:
"""
The player can't hold any explosion.
"""
pass
class Weapon(Item):
"""
Non-throwable items that improve player damage
"""
damage: int
def __init__(self, damage: int = 3, *args, **kwargs):
super().__init__(*args, **kwargs)
self.damage = damage
def save_state(self) -> dict:
"""
Saves the state of the weapon into a dictionary
"""
d = super().save_state()
d["damage"] = self.damage
return d
2020-12-07 20:22:06 +00:00
class Sword(Weapon):
"""
A basic weapon
"""
def __init__(self, name: str = "sword", price: int = 20,
2020-12-18 16:39:11 +00:00
*args, **kwargs):
super().__init__(name=name, price=price, *args, **kwargs)
self.name = name
2020-12-18 16:39:11 +00:00
def equip(self, armor: bool = False) -> None:
2020-12-18 16:39:11 +00:00
"""
When a sword is equipped, the player gains strength.
"""
super().equip()
self.held_by.strength += self.damage
2020-12-18 16:39:11 +00:00
def unequip(self) -> None:
"""
Remove the strength earned by the sword.
:return:
"""
super().unequip()
self.held_by.strength -= self.damage
class Shield(Item):
constitution: int
def __init__(self, constitution: int = 2, *args, **kwargs):
super().__init__(name="shield", *args, **kwargs)
self.constitution = constitution
def equip(self, armor: bool = True) -> None:
super().equip(armor)
self.held_by.constitution += self.constitution
def unequip(self) -> None:
super().unequip()
self.held_by.constitution -= self.constitution
def save_state(self) -> dict:
d = super().save_state()
d["constitution"] = self.constitution
return d
2020-12-07 20:22:06 +00:00
2020-12-05 13:20:58 +00:00
class BodySnatchPotion(Item):
"""
The body-snatch potion allows to exchange all characteristics with a random
other entity.
"""
2020-12-11 16:06:30 +00:00
def __init__(self, name: str = "body_snatch_potion", price: int = 14,
*args, **kwargs):
super().__init__(name=name, price=price, *args, **kwargs)
2020-12-05 13:20:58 +00:00
def use(self) -> None:
"""
Find a valid random entity, then exchange characteristics.
"""
valid_entities = self.held_by.map.find_entities(FightingEntity)
valid_entities.remove(self.held_by)
entity = choice(valid_entities)
entity_state = entity.save_state()
player_state = self.held_by.save_state()
self.held_by.__dict__.update(entity_state)
entity.__dict__.update(player_state)
self.held_by.map.currenty, self.held_by.map.currentx = self.held_by.y,\
self.held_by.x
self.held_by.map.logs.add_message(
_("{player} exchanged its body with {entity}.").format(
player=self.held_by.translated_name.capitalize(),
entity=entity.translated_name))
self.held_by.recalculate_paths()
2020-12-05 13:20:58 +00:00
self.held_by.inventory.remove(self)