Source code for evennia.contrib.game_systems.storage.storage
from evennia import CmdSet
from evennia.utils import list_to_string
from evennia.utils.search import search_object_by_tag
from evennia.commands.default.muxcommand import MuxCommand
SHARED_TAG_PREFIX = "shared"
[docs]class StorageCommand(MuxCommand):
"""
Shared functionality for storage-related commands
"""
[docs] def at_pre_cmd(self):
"""
Check if the current location is tagged as a storage location
Every stored object is tagged on storage, and untagged on retrieval
Returns:
bool: True if the command is to be stopped here
"""
if super().at_pre_cmd():
return True
self.storage_location_id = self.caller.location.tags.get(category="storage_location")
if not self.storage_location_id:
self.caller.msg(f"You cannot {self.cmdstring} anything here.")
return True
self.object_tag = (
SHARED_TAG_PREFIX
if self.storage_location_id.startswith(SHARED_TAG_PREFIX)
else self.caller.pk
)
self.currently_stored = search_object_by_tag(
self.object_tag, category=self.storage_location_id
)
[docs]class CmdStore(StorageCommand):
"""
Store something in a storage location.
Usage:
store <obj>
"""
key = "store"
locks = "cmd:all()"
help_category = "Storage"
[docs] def func(self):
"""
Find the item in question to store, then store it
"""
caller = self.caller
if not self.args:
self.caller.msg("Store what?")
return
obj = caller.search(self.args.strip(), candidates=caller.contents)
if not obj:
return
"""
We first check at_pre_move before setting the location to None, in case
anything should stymie its movement.
"""
if obj.at_pre_move(caller.location):
obj.tags.add(self.object_tag, self.storage_location_id)
obj.location = None
caller.msg(f"You store {obj.get_display_name(caller)} here.")
else:
caller.msg(f"You fail to store {obj.get_display_name(caller)} here.")
[docs]class CmdRetrieve(StorageCommand):
"""
Retrieve something from a storage location.
Usage:
retrieve <obj>
"""
key = "retrieve"
locks = "cmd:all()"
help_category = "Storage"
[docs] def func(self):
"""
Retrieve the item in question if possible
"""
caller = self.caller
if not self.args:
self.caller.msg("Retrieve what?")
return
obj = caller.search(self.args.strip(), candidates=self.currently_stored)
if not obj:
return
if obj.at_pre_move(caller):
obj.tags.remove(self.object_tag, self.storage_location_id)
caller.msg(f"You retrieve {obj.get_display_name(caller)}.")
else:
caller.msg(f"You fail to retrieve {obj.get_display_name(caller)}.")
[docs]class CmdList(StorageCommand):
"""
List items in the storage location.
Usage:
list
"""
key = "list"
locks = "cmd:all()"
help_category = "Storage"
[docs] def func(self):
"""
List items in the storage
"""
caller = self.caller
if not self.currently_stored:
caller.msg("You find nothing stored here.")
return
caller.msg(f"Stored here:\n{list_to_string(self.currently_stored)}")
[docs]class CmdStorage(MuxCommand):
"""
Make the current location a storage room, or delete it as a storage and move all stored objects into the room contents.
Shared storage locations can be used by all interchangeably.
The default storage identifier will be its primary key in the database, but you can supply a new one in case you want linked storages.
Usage:
storage [= [storage identifier]]
storage/shared [= [storage identifier]]
storage/delete
"""
key = "@storage"
locks = "cmd:perm(Builder)"
[docs] def func(self):
"""Set the storage location."""
caller = self.caller
location = caller.location
current_storage_id = location.tags.get(category="storage_location")
storage_id = self.lhs or location.pk
if "delete" in self.switches:
if not current_storage_id:
caller.msg("This is not tagged as a storage location.")
return
# Move the stored objects, if any, into the room
currently_stored_here = search_object_by_tag(category=current_storage_id)
for obj in currently_stored_here:
obj.tags.remove(category=current_storage_id)
obj.location = location
caller.msg("You remove the storage capabilities of the room.")
location.tags.remove(current_storage_id, category="storage_location")
return
if current_storage_id:
caller.msg("This is already a storage location: |wstorage/delete|n to remove the tag.")
return
new_storage_id = (
f"{SHARED_TAG_PREFIX if SHARED_TAG_PREFIX in self.switches else ''}{storage_id}"
)
location.tags.add(new_storage_id, category="storage_location")
caller.msg(f"This is now a storage location with id: {new_storage_id}.")
[docs]class StorageCmdSet(CmdSet):
"""
CmdSet for all storage-related commands
"""
[docs] def at_cmdset_creation(self):
self.add(CmdStore)
self.add(CmdRetrieve)
self.add(CmdList)
self.add(CmdStorage)