# -*- coding: cp1254 -*-
from header_common import *
from header_dialogs import *
from header_operations import *
from header_parties import *
from header_item_modifiers import *
from header_skills import *
from header_triggers import *
from ID_troops import *
from ID_party_templates import *
from module_constants import *
from util_wrappers import *
def modmerge_dialogs(orig_dialogs):
# Start - Allow export and import of companions
# Set troop back to player before opening equipment
dialog = FindDialog(orig_dialogs, ststate="member_trade",
endstate="do_member_trade")
cons_block = dialog.GetConsequenceBlock()
pos = cons_block.FindLineMatching((change_screen_equip_other))
cons_block.InsertBefore(pos, [(set_player_troop, "trp_player")])
# Set troop back to player when ending conversation
dialog = FindDialog(orig_dialogs, ststate="member_talk",
endstate="close_window")
cons_block = dialog.GetConsequenceBlock()
pos = cons_block.FindLastLine()
cons_block.InsertAfter(pos, [(set_player_troop, "trp_player")])
# Set player troop to companion we are talking with for stats screen
dialog = FindDialog(orig_dialogs, ststate="view_member_char_requested",
endstate="do_member_view_char")
cons_block = dialog.GetConsequenceBlock()
pos = cons_block.FindLineMatching((change_screen_view_character))
cons_block.InsertBefore(pos, [(set_player_troop, "$g_talk_troop")])
# End - Allow export and import of companions
def modmerge(var_set):
try:
orig_dialogs = var_set["dialogs"]
modmerge_dialogs(orig_dialogs)
except KeyError:
raise ValueError("Variable set missing dialogs")
