Kayne Ruse
f831019148
When more than one item can be picked up, and options window is shown. Stubs for "using" an item are in place.
32 lines
699 B
Python
32 lines
699 B
Python
from __future__ import annotations
|
|
from typing import Optional, Set, TYPE_CHECKING
|
|
|
|
if TYPE_CHECKING:
|
|
from entity import Entity
|
|
|
|
class Inventory:
|
|
"""Handles inventory for an Entity"""
|
|
_contents: Set[Entity]
|
|
|
|
def __init__(self, contents: Set[Entity] = set()):
|
|
self._contents = contents
|
|
|
|
def insert(self, entity: Entity) -> bool:
|
|
if entity in self._contents:
|
|
return False
|
|
|
|
self._contents.add(entity)
|
|
return True
|
|
|
|
def access(self, key: str) -> Optional[Entity]:
|
|
return self._contents[key]
|
|
|
|
def remove(self, key: str) -> Optional[Entity]:
|
|
item = self._contents[key]
|
|
self._contents.remove(key)
|
|
return item
|
|
|
|
@property
|
|
def contents(self) -> Set[Entity]:
|
|
return self._contents
|