247 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
		
		
			
		
	
	
			247 lines
		
	
	
		
			11 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| 
								 | 
							
								import yaml
							 | 
						||
| 
								 | 
							
								import os
							 | 
						||
| 
								 | 
							
								import Utils
							 | 
						||
| 
								 | 
							
								import zipfile
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								from .Items import item_dictionary_table, CheckDupingItems
							 | 
						||
| 
								 | 
							
								from .Locations import all_locations, SoraLevels, exclusion_table, AllWeaponSlot
							 | 
						||
| 
								 | 
							
								from .Names import LocationName
							 | 
						||
| 
								 | 
							
								from .XPValues import lvlStats, formExp, soraExp
							 | 
						||
| 
								 | 
							
								from worlds.Files import APContainer
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								class KH2Container(APContainer):
							 | 
						||
| 
								 | 
							
								    game: str = 'Kingdom Hearts 2'
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    def __init__(self, patch_data: dict, base_path: str, output_directory: str,
							 | 
						||
| 
								 | 
							
								                 player=None, player_name: str = "", server: str = ""):
							 | 
						||
| 
								 | 
							
								        self.patch_data = patch_data
							 | 
						||
| 
								 | 
							
								        self.file_path = base_path
							 | 
						||
| 
								 | 
							
								        container_path = os.path.join(output_directory, base_path + ".zip")
							 | 
						||
| 
								 | 
							
								        super().__init__(container_path, player, player_name, server)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    def write_contents(self, opened_zipfile: zipfile.ZipFile) -> None:
							 | 
						||
| 
								 | 
							
								        for filename, yml in self.patch_data.items():
							 | 
						||
| 
								 | 
							
								            opened_zipfile.writestr(filename, yml)
							 | 
						||
| 
								 | 
							
								        for root, dirs, files in os.walk(os.path.join(os.path.dirname(__file__), "mod_template")):
							 | 
						||
| 
								 | 
							
								            for file in files:
							 | 
						||
| 
								 | 
							
								                opened_zipfile.write(os.path.join(root, file),
							 | 
						||
| 
								 | 
							
								                                     os.path.relpath(os.path.join(root, file),
							 | 
						||
| 
								 | 
							
								                                                     os.path.join(os.path.dirname(__file__), "mod_template")))
							 | 
						||
| 
								 | 
							
								        # opened_zipfile.writestr(self.zpf_path, self.patch_data)
							 | 
						||
| 
								 | 
							
								        super().write_contents(opened_zipfile)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								def patch_kh2(self, output_directory):
							 | 
						||
| 
								 | 
							
								    def increaseStat(i):
							 | 
						||
| 
								 | 
							
								        if lvlStats[i] == "str":
							 | 
						||
| 
								 | 
							
								            self.strength += 2
							 | 
						||
| 
								 | 
							
								        elif lvlStats[i] == "mag":
							 | 
						||
| 
								 | 
							
								            self.magic += 2
							 | 
						||
| 
								 | 
							
								        elif lvlStats[i] == "def":
							 | 
						||
| 
								 | 
							
								            self.defense += 1
							 | 
						||
| 
								 | 
							
								        elif lvlStats[i] == "ap":
							 | 
						||
| 
								 | 
							
								            self.ap += 3
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    self.formattedTrsr = {}
							 | 
						||
| 
								 | 
							
								    self.formattedBons = []
							 | 
						||
| 
								 | 
							
								    self.formattedLvup = {"Sora": {}}
							 | 
						||
| 
								 | 
							
								    self.formattedBons = {}
							 | 
						||
| 
								 | 
							
								    self.formattedFmlv = {}
							 | 
						||
| 
								 | 
							
								    self.formattedItem = {"Stats": []}
							 | 
						||
| 
								 | 
							
								    self.formattedPlrp = []
							 | 
						||
| 
								 | 
							
								    self.strength = 2
							 | 
						||
| 
								 | 
							
								    self.magic = 6
							 | 
						||
| 
								 | 
							
								    self.defense = 2
							 | 
						||
| 
								 | 
							
								    self.ap = 0
							 | 
						||
| 
								 | 
							
								    self.dblbonus = 0
							 | 
						||
| 
								 | 
							
								    formexp = None
							 | 
						||
| 
								 | 
							
								    formName = None
							 | 
						||
| 
								 | 
							
								    levelsetting = list()
							 | 
						||
| 
								 | 
							
								    slotDataDuping = set()
							 | 
						||
| 
								 | 
							
								    for values in CheckDupingItems.values():
							 | 
						||
| 
								 | 
							
								        if isinstance(values, set):
							 | 
						||
| 
								 | 
							
								            slotDataDuping |= values
							 | 
						||
| 
								 | 
							
								        else:
							 | 
						||
| 
								 | 
							
								            for inner_values in values.values():
							 | 
						||
| 
								 | 
							
								                slotDataDuping |= inner_values
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    if self.multiworld.Keyblade_Minimum[self.player].value > self.multiworld.Keyblade_Maximum[self.player].value:
							 | 
						||
| 
								 | 
							
								        print(
							 | 
						||
| 
								 | 
							
								                f"{self.multiworld.get_file_safe_player_name(self.player)} has Keyblade Minimum greater than Keyblade Maximum")
							 | 
						||
| 
								 | 
							
								        keyblademin = self.multiworld.Keyblade_Maximum[self.player].value
							 | 
						||
| 
								 | 
							
								        keyblademax = self.multiworld.Keyblade_Minimum[self.player].value
							 | 
						||
| 
								 | 
							
								    else:
							 | 
						||
| 
								 | 
							
								        keyblademin = self.multiworld.Keyblade_Minimum[self.player].value
							 | 
						||
| 
								 | 
							
								        keyblademax = self.multiworld.Keyblade_Maximum[self.player].value
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    if self.multiworld.LevelDepth[self.player] == "level_50":
							 | 
						||
| 
								 | 
							
								        levelsetting.extend(exclusion_table["Level50"])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    elif self.multiworld.LevelDepth[self.player] == "level_99":
							 | 
						||
| 
								 | 
							
								        levelsetting.extend(exclusion_table["Level99"])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    elif self.multiworld.LevelDepth[self.player] in ["level_50_sanity", "level_99_sanity"]:
							 | 
						||
| 
								 | 
							
								        levelsetting.extend(exclusion_table["Level50Sanity"])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								        if self.multiworld.LevelDepth[self.player] == "level_99_sanity":
							 | 
						||
| 
								 | 
							
								            levelsetting.extend(exclusion_table["Level99Sanity"])
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    mod_name = f"AP-{self.multiworld.seed_name}-P{self.player}-{self.multiworld.get_file_safe_player_name(self.player)}"
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    for location in self.multiworld.get_filled_locations(self.player):
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								        data = all_locations[location.name]
							 | 
						||
| 
								 | 
							
								        if location.item.player == self.player:
							 | 
						||
| 
								 | 
							
								            itemcode = item_dictionary_table[location.item.name].kh2id
							 | 
						||
| 
								 | 
							
								            if location.item.name in slotDataDuping and \
							 | 
						||
| 
								 | 
							
								                    location.name not in AllWeaponSlot:
							 | 
						||
| 
								 | 
							
								                self.LocalItems[location.address] = item_dictionary_table[location.item.name].code
							 | 
						||
| 
								 | 
							
								        else:
							 | 
						||
| 
								 | 
							
								            itemcode = 90  # castle map
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								        if data.yml == "Chest":
							 | 
						||
| 
								 | 
							
								            self.formattedTrsr[data.locid] = {"ItemId": itemcode}
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								        elif data.yml in ["Get Bonus", "Double Get Bonus", "Second Get Bonus"]:
							 | 
						||
| 
								 | 
							
								            if data.yml == "Get Bonus":
							 | 
						||
| 
								 | 
							
								                self.dblbonus = 0
							 | 
						||
| 
								 | 
							
								            # if double bonus then addresses dbl bonus so the next check gets 2 items on it
							 | 
						||
| 
								 | 
							
								            if data.yml == "Double Get Bonus":
							 | 
						||
| 
								 | 
							
								                self.dblbonus = itemcode
							 | 
						||
| 
								 | 
							
								                continue
							 | 
						||
| 
								 | 
							
								            if data.locid not in self.formattedBons.keys():
							 | 
						||
| 
								 | 
							
								                self.formattedBons[data.locid] = {}
							 | 
						||
| 
								 | 
							
								            self.formattedBons[data.locid][data.charName] = {
							 | 
						||
| 
								 | 
							
								                "RewardId":             data.locid,
							 | 
						||
| 
								 | 
							
								                "CharacterId":          data.charNumber,
							 | 
						||
| 
								 | 
							
								                "HpIncrease":           0,
							 | 
						||
| 
								 | 
							
								                "MpIncrease":           0,
							 | 
						||
| 
								 | 
							
								                "DriveGaugeUpgrade":    0,
							 | 
						||
| 
								 | 
							
								                "ItemSlotUpgrade":      0,
							 | 
						||
| 
								 | 
							
								                "AccessorySlotUpgrade": 0,
							 | 
						||
| 
								 | 
							
								                "ArmorSlotUpgrade":     0,
							 | 
						||
| 
								 | 
							
								                "BonusItem1":           itemcode,
							 | 
						||
| 
								 | 
							
								                "BonusItem2":           self.dblbonus,
							 | 
						||
| 
								 | 
							
								                "Padding":              0
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								            }
							 | 
						||
| 
								 | 
							
								            # putting dbl bonus at 0 again, so we don't have the same item placed multiple time
							 | 
						||
| 
								 | 
							
								            self.dblbonus = 0
							 | 
						||
| 
								 | 
							
								        elif data.yml == "Keyblade":
							 | 
						||
| 
								 | 
							
								            self.formattedItem["Stats"].append({
							 | 
						||
| 
								 | 
							
								                "Id":                  data.locid,
							 | 
						||
| 
								 | 
							
								                "Attack":              self.multiworld.per_slot_randoms[self.player].randint(keyblademin, keyblademax),
							 | 
						||
| 
								 | 
							
								                "Magic":               self.multiworld.per_slot_randoms[self.player].randint(keyblademin, keyblademax),
							 | 
						||
| 
								 | 
							
								                "Defense":             0,
							 | 
						||
| 
								 | 
							
								                "Ability":             itemcode,
							 | 
						||
| 
								 | 
							
								                "AbilityPoints":       0,
							 | 
						||
| 
								 | 
							
								                "Unknown08":           100,
							 | 
						||
| 
								 | 
							
								                "FireResistance":      100,
							 | 
						||
| 
								 | 
							
								                "IceResistance":       100,
							 | 
						||
| 
								 | 
							
								                "LightningResistance": 100,
							 | 
						||
| 
								 | 
							
								                "DarkResistance":      100,
							 | 
						||
| 
								 | 
							
								                "Unknown0d":           100,
							 | 
						||
| 
								 | 
							
								                "GeneralResistance":   100,
							 | 
						||
| 
								 | 
							
								                "Unknown":             0
							 | 
						||
| 
								 | 
							
								            })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								        elif data.yml == "Forms":
							 | 
						||
| 
								 | 
							
								            # loc id is form lvl
							 | 
						||
| 
								 | 
							
								            # char name is the form name number :)
							 | 
						||
| 
								 | 
							
								            if data.locid == 2:
							 | 
						||
| 
								 | 
							
								                formDict = {1: "Valor", 2: "Wisdom", 3: "Limit", 4: "Master", 5: "Final"}
							 | 
						||
| 
								 | 
							
								                formDictExp = {
							 | 
						||
| 
								 | 
							
								                    1: self.multiworld.Valor_Form_EXP[self.player].value,
							 | 
						||
| 
								 | 
							
								                    2: self.multiworld.Wisdom_Form_EXP[self.player].value,
							 | 
						||
| 
								 | 
							
								                    3: self.multiworld.Limit_Form_EXP[self.player].value,
							 | 
						||
| 
								 | 
							
								                    4: self.multiworld.Master_Form_EXP[self.player].value,
							 | 
						||
| 
								 | 
							
								                    5: self.multiworld.Final_Form_EXP[self.player].value}
							 | 
						||
| 
								 | 
							
								                formexp = formDictExp[data.charName]
							 | 
						||
| 
								 | 
							
								                formName = formDict[data.charName]
							 | 
						||
| 
								 | 
							
								                self.formattedFmlv[formName] = []
							 | 
						||
| 
								 | 
							
								                self.formattedFmlv[formName].append({
							 | 
						||
| 
								 | 
							
								                    "Ability":            1,
							 | 
						||
| 
								 | 
							
								                    "Experience":         int(formExp[data.charName][data.locid] / formexp),
							 | 
						||
| 
								 | 
							
								                    "FormId":             data.charName,
							 | 
						||
| 
								 | 
							
								                    "FormLevel":          1,
							 | 
						||
| 
								 | 
							
								                    "GrowthAbilityLevel": 0,
							 | 
						||
| 
								 | 
							
								                })
							 | 
						||
| 
								 | 
							
								            # row is form column is lvl
							 | 
						||
| 
								 | 
							
								            self.formattedFmlv[formName].append({
							 | 
						||
| 
								 | 
							
								                "Ability":            itemcode,
							 | 
						||
| 
								 | 
							
								                "Experience":         int(formExp[data.charName][data.locid] / formexp),
							 | 
						||
| 
								 | 
							
								                "FormId":             data.charName,
							 | 
						||
| 
								 | 
							
								                "FormLevel":          data.locid,
							 | 
						||
| 
								 | 
							
								                "GrowthAbilityLevel": 0,
							 | 
						||
| 
								 | 
							
								            })
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    # Summons have no checks on them so done fully locally
							 | 
						||
| 
								 | 
							
								    self.formattedFmlv["Summon"] = []
							 | 
						||
| 
								 | 
							
								    for x in range(1, 7):
							 | 
						||
| 
								 | 
							
								        self.formattedFmlv["Summon"].append({
							 | 
						||
| 
								 | 
							
								            "Ability":            123,
							 | 
						||
| 
								 | 
							
								            "Experience":         int(formExp[0][x] / self.multiworld.Summon_EXP[self.player].value),
							 | 
						||
| 
								 | 
							
								            "FormId":             0,
							 | 
						||
| 
								 | 
							
								            "FormLevel":          x,
							 | 
						||
| 
								 | 
							
								            "GrowthAbilityLevel": 0,
							 | 
						||
| 
								 | 
							
								        })
							 | 
						||
| 
								 | 
							
								    # levels done down here because of optional settings that can take locations out of the pool.
							 | 
						||
| 
								 | 
							
								    self.i = 1
							 | 
						||
| 
								 | 
							
								    for location in SoraLevels:
							 | 
						||
| 
								 | 
							
								        increaseStat(self.multiworld.per_slot_randoms[self.player].randint(0, 3))
							 | 
						||
| 
								 | 
							
								        if location in levelsetting:
							 | 
						||
| 
								 | 
							
								            data = self.multiworld.get_location(location, self.player)
							 | 
						||
| 
								 | 
							
								            if data.item.player == self.player:
							 | 
						||
| 
								 | 
							
								                itemcode = item_dictionary_table[data.item.name].kh2id
							 | 
						||
| 
								 | 
							
								            else:
							 | 
						||
| 
								 | 
							
								                itemcode = 461
							 | 
						||
| 
								 | 
							
								        else:
							 | 
						||
| 
								 | 
							
								            increaseStat(self.multiworld.per_slot_randoms[self.player].randint(0, 3))
							 | 
						||
| 
								 | 
							
								            itemcode = 0
							 | 
						||
| 
								 | 
							
								        self.formattedLvup["Sora"][self.i] = {
							 | 
						||
| 
								 | 
							
								            "Exp":           int(soraExp[self.i] / self.multiworld.Sora_Level_EXP[self.player].value),
							 | 
						||
| 
								 | 
							
								            "Strength":      self.strength,
							 | 
						||
| 
								 | 
							
								            "Magic":         self.magic,
							 | 
						||
| 
								 | 
							
								            "Defense":       self.defense,
							 | 
						||
| 
								 | 
							
								            "Ap":            self.ap,
							 | 
						||
| 
								 | 
							
								            "SwordAbility":  itemcode,
							 | 
						||
| 
								 | 
							
								            "ShieldAbility": itemcode,
							 | 
						||
| 
								 | 
							
								            "StaffAbility":  itemcode,
							 | 
						||
| 
								 | 
							
								            "Padding":       0,
							 | 
						||
| 
								 | 
							
								            "Character":     "Sora",
							 | 
						||
| 
								 | 
							
								            "Level":         self.i
							 | 
						||
| 
								 | 
							
								        }
							 | 
						||
| 
								 | 
							
								        self.i += 1
							 | 
						||
| 
								 | 
							
								    # averaging stats for the struggle bats
							 | 
						||
| 
								 | 
							
								    for x in {122, 144, 145}:
							 | 
						||
| 
								 | 
							
								        self.formattedItem["Stats"].append({
							 | 
						||
| 
								 | 
							
								            "Id":                  x,
							 | 
						||
| 
								 | 
							
								            "Attack":              int((keyblademin + keyblademax) / 2),
							 | 
						||
| 
								 | 
							
								            "Magic":               int((keyblademin + keyblademax) / 2),
							 | 
						||
| 
								 | 
							
								            "Defense":             0,
							 | 
						||
| 
								 | 
							
								            "Ability":             405,
							 | 
						||
| 
								 | 
							
								            "AbilityPoints":       0,
							 | 
						||
| 
								 | 
							
								            "Unknown08":           100,
							 | 
						||
| 
								 | 
							
								            "FireResistance":      100,
							 | 
						||
| 
								 | 
							
								            "IceResistance":       100,
							 | 
						||
| 
								 | 
							
								            "LightningResistance": 100,
							 | 
						||
| 
								 | 
							
								            "DarkResistance":      100,
							 | 
						||
| 
								 | 
							
								            "Unknown0d":           100,
							 | 
						||
| 
								 | 
							
								            "GeneralResistance":   100,
							 | 
						||
| 
								 | 
							
								            "Unknown":             0
							 | 
						||
| 
								 | 
							
								        })
							 | 
						||
| 
								 | 
							
								    mod_dir = os.path.join(output_directory, mod_name + "_" + Utils.__version__)
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    openkhmod = {
							 | 
						||
| 
								 | 
							
								        "TrsrList.yml": yaml.dump(self.formattedTrsr, line_break="\n"),
							 | 
						||
| 
								 | 
							
								        "LvupList.yml": yaml.dump(self.formattedLvup, line_break="\n"),
							 | 
						||
| 
								 | 
							
								        "BonsList.yml": yaml.dump(self.formattedBons, line_break="\n"),
							 | 
						||
| 
								 | 
							
								        "ItemList.yml": yaml.dump(self.formattedItem, line_break="\n"),
							 | 
						||
| 
								 | 
							
								        "FmlvList.yml": yaml.dump(self.formattedFmlv, line_break="\n"),
							 | 
						||
| 
								 | 
							
								    }
							 | 
						||
| 
								 | 
							
								
							 | 
						||
| 
								 | 
							
								    mod = KH2Container(openkhmod, mod_dir, output_directory, self.player,
							 | 
						||
| 
								 | 
							
								                       self.multiworld.get_file_safe_player_name(self.player))
							 | 
						||
| 
								 | 
							
								    mod.write()
							 |