85 lines
2.8 KiB
Python
85 lines
2.8 KiB
Python
from taleweave.context import action_context, get_system_data
|
|
from taleweave.errors import ActionError
|
|
from taleweave.systems.quest import (
|
|
QUEST_SYSTEM,
|
|
complete_quest,
|
|
get_active_quest,
|
|
get_quests_for_character,
|
|
set_active_quest,
|
|
)
|
|
from taleweave.utils.search import find_character_in_room
|
|
from taleweave.utils.template import format_prompt
|
|
|
|
|
|
def accept_quest(character: str, quest: str) -> str:
|
|
"""
|
|
Accept and start a quest being given by another character.
|
|
"""
|
|
|
|
with action_context() as (action_room, action_character):
|
|
quests = get_system_data(QUEST_SYSTEM)
|
|
if not quests:
|
|
raise ActionError(
|
|
format_prompt("action_accept_quest_error_none", character=character)
|
|
)
|
|
|
|
target_character = find_character_in_room(action_room, character)
|
|
if not target_character:
|
|
raise ActionError(
|
|
format_prompt("action_accept_quest_error_room", character=character)
|
|
)
|
|
|
|
available_quests = get_quests_for_character(quests, target_character)
|
|
|
|
for available_quest in available_quests:
|
|
if available_quest.name == quest:
|
|
set_active_quest(quests, action_character, available_quest)
|
|
return format_prompt(
|
|
"action_accept_quest_result", character=character, quest=quest
|
|
)
|
|
|
|
raise ActionError(
|
|
format_prompt(
|
|
"action_accept_quest_error_name", character=character, quest=quest
|
|
)
|
|
)
|
|
|
|
|
|
def submit_quest(character: str) -> str:
|
|
"""
|
|
Submit your active quest to the quest giver. If you have completed the quest, you will be rewarded.
|
|
"""
|
|
|
|
with action_context() as (action_room, action_character):
|
|
quests = get_system_data(QUEST_SYSTEM)
|
|
if not quests:
|
|
raise ActionError(
|
|
format_prompt("action_submit_quest_error_none", character=character)
|
|
)
|
|
|
|
active_quest = get_active_quest(quests, action_character)
|
|
if not active_quest:
|
|
raise ActionError(
|
|
format_prompt("action_submit_quest_error_active", character=character)
|
|
)
|
|
|
|
target_character = find_character_in_room(action_room, character)
|
|
if not target_character:
|
|
raise ActionError(
|
|
format_prompt("action_submit_quest_error_room", character=character)
|
|
)
|
|
|
|
if active_quest.giver.character == target_character.name:
|
|
complete_quest(quests, action_character, active_quest)
|
|
return format_prompt(
|
|
"action_submit_quest_result",
|
|
character=character,
|
|
quest=active_quest,
|
|
)
|
|
|
|
return format_prompt(
|
|
"action_submit_quest_error_name",
|
|
character=character,
|
|
quest=active_quest,
|
|
)
|