companions.py
"""Companion character database: 20x3★, 10x4★, 5x5★."""
from __future__ import annotations
import random
from data_types import *
from character import Character, JOB_BASE_STATS, JOB_GROWTH
from skills_db import SKILL_DB, JOB_SKILL_POOL
def _make_companion(name, job, weapon, element, rarity,
stat_mult=1.0, growth_mult=1.0) -> Character:
base = JOB_BASE_STATS[job].copy()
g = JOB_GROWTH[job]
# Apply rarity scaling
base.hp = int(base.hp * stat_mult)
base.mp = int(base.mp * stat_mult)
base.patk = int(base.patk * stat_mult)
base.matk = int(base.matk * stat_mult)
base.pdef = int(base.pdef * stat_mult)
base.mdef = int(base.mdef * stat_mult)
base.spd = int(base.spd * stat_mult)
# Scale growth rates
scaled_g = GrowthRates(
hp = min(0.99, g.hp * growth_mult),
mp = min(0.99, g.mp * growth_mult),
patk = min(0.99, g.patk * growth_mult),
matk = min(0.99, g.matk * growth_mult),
pdef = min(0.99, g.pdef * growth_mult),
mdef = min(0.99, g.mdef * growth_mult),
spd = min(0.99, g.spd * growth_mult),
eva = min(0.99, g.eva * growth_mult),
acc = min(0.99, g.acc * growth_mult),
crit = min(0.99, g.crit * growth_mult),
)
pool = JOB_SKILL_POOL[job]
basics = [sid for sid in pool if SKILL_DB[sid].tier == SkillTier.BASIC][:4]
inters = [sid for sid in pool if SKILL_DB[sid].tier == SkillTier.INTERMEDIATE][:4]
ultims = [sid for sid in pool if SKILL_DB[sid].tier == SkillTier.ULTIMATE][:2]
chosen_basic = random.sample(basics, min(2, len(basics)))
chosen_inter = random.sample(inters, min(2, len(inters)))
chosen_ultim = random.sample(ultims, min(1, len(ultims)))
skill_pool = chosen_basic + chosen_inter + chosen_ultim
unlocked = chosen_basic[:]
return Character(
name=name, job=job, weapon=weapon, element=element, rarity=rarity,
base_stats=base, growth=scaled_g,
skill_pool=skill_pool, unlocked_skills=unlocked
)
# ══════════════════════════════════════════════════════════════════════════
# COMPANION TEMPLATES (factories — called fresh each game)
# ══════════════════════════════════════════════════════════════════════════
# stat_mult: 3★=0.90~0.95, 4★=1.10~1.20, 5★=1.35~1.50
# growth_mult: 3★=0.85~0.90, 4★=1.05~1.15, 5★=1.25~1.40
THREE_STAR_TEMPLATES = [
# name, job, weapon, element, stat_mult, growth_mult
("Rook", JobClass.WARRIOR, WeaponType.SWORD, Element.FIRE, 0.90, 0.88),
("Gara", JobClass.KNIGHT, WeaponType.SWORD, Element.EARTH, 0.92, 0.87),
("Tifa", JobClass.MONK, WeaponType.AXE, Element.WIND, 0.91, 0.88),
("Sera", JobClass.CLERIC, WeaponType.STAFF, Element.LIGHT, 0.93, 0.88),
("Rex", JobClass.RANGER, WeaponType.BOW, Element.ICE, 0.90, 0.87),
("Bram", JobClass.BERSERKER, WeaponType.AXE, Element.FIRE, 0.89, 0.86),
("Nira", JobClass.WIND_MAGE, WeaponType.STAFF, Element.WIND, 0.92, 0.87),
("Dag", JobClass.SPEARMAN, WeaponType.SPEAR, Element.LIGHTNING, 0.91, 0.88),
("Ella", JobClass.FIRE_MAGE, WeaponType.STAFF, Element.FIRE, 0.90, 0.87),
("Wynn", JobClass.HUNTER, WeaponType.BOW, Element.EARTH, 0.93, 0.88),
("Bart", JobClass.EARTH_MAGE, WeaponType.STAFF, Element.EARTH, 0.91, 0.86),
("Yuna", JobClass.PRIEST, WeaponType.STAFF, Element.LIGHT, 0.92, 0.87),
("Kage", JobClass.ASSASSIN, WeaponType.DAGGER, Element.DARK, 0.90, 0.86),
("Coral", JobClass.ICE_MAGE, WeaponType.STAFF, Element.ICE, 0.91, 0.87),
("Miles", JobClass.WARRIOR, WeaponType.AXE, Element.WIND, 0.90, 0.86),
("Dora", JobClass.CLERIC, WeaponType.STAFF, Element.FIRE, 0.92, 0.87),
("Oryn", JobClass.STORM_MAGE, WeaponType.STAFF, Element.LIGHTNING, 0.90, 0.86),
("Fenn", JobClass.PALADIN, WeaponType.SWORD, Element.LIGHT, 0.93, 0.88),
("Zell", JobClass.DRAGOON, WeaponType.SPEAR, Element.WIND, 0.91, 0.87),
("Mira", JobClass.DARK_MAGE, WeaponType.STAFF, Element.DARK, 0.90, 0.86),
]
FOUR_STAR_TEMPLATES = [
("Cael", JobClass.KNIGHT, WeaponType.SWORD, Element.LIGHT, 1.12, 1.10),
("Lyra", JobClass.ARCANE_SAGE,WeaponType.STAFF, Element.LIGHTNING, 1.15, 1.12),
("Vance", JobClass.BERSERKER, WeaponType.AXE, Element.DARK, 1.18, 1.14),
("Shira", JobClass.ASSASSIN, WeaponType.DAGGER, Element.ICE, 1.12, 1.10),
("Bael", JobClass.DRAGOON, WeaponType.SPEAR, Element.FIRE, 1.14, 1.11),
("Noel", JobClass.PRIEST, WeaponType.STAFF, Element.LIGHT, 1.15, 1.12),
("Kira", JobClass.STORM_MAGE, WeaponType.STAFF, Element.LIGHTNING, 1.16, 1.13),
("Roan", JobClass.PALADIN, WeaponType.SWORD, Element.LIGHT, 1.14, 1.10),
("Zara", JobClass.HUNTER, WeaponType.BOW, Element.WIND, 1.12, 1.10),
("Drax", JobClass.MONK, WeaponType.AXE, Element.EARTH, 1.18, 1.14),
]
FIVE_STAR_TEMPLATES = [
("Seraph", JobClass.LIGHT_MAGE, WeaponType.STAFF, Element.LIGHT, 1.45, 1.38),
("Abyss", JobClass.DARK_MAGE, WeaponType.STAFF, Element.DARK, 1.42, 1.35),
("Titan", JobClass.BERSERKER, WeaponType.AXE, Element.EARTH, 1.50, 1.40),
("Oracle", JobClass.ARCANE_SAGE,WeaponType.STAFF, Element.LIGHTNING, 1.45, 1.40),
("Valkyrie",JobClass.PALADIN, WeaponType.SWORD, Element.LIGHT, 1.48, 1.38),
]
COMPANION_GACHA_POOL = {
3: THREE_STAR_TEMPLATES,
4: FOUR_STAR_TEMPLATES,
5: FIVE_STAR_TEMPLATES,
}
GACHA_RATES = {3: 0.65, 4: 0.30, 5: 0.05}
def summon_companion(target_level: int = 2) -> Character:
"""Roll a random companion via gacha, leveled to target_level."""
roll = random.random()
cumulative = 0.0
rarity = 3
for r, rate in GACHA_RATES.items():
cumulative += rate
if roll < cumulative:
rarity = r
break
templates = COMPANION_GACHA_POOL[rarity]
tmpl = random.choice(templates)
name, job, weapon, element, stat_mult, growth_mult = tmpl
char = _make_companion(name, job, weapon, element, rarity,
stat_mult, growth_mult)
# Level the character up silently
for _ in range(1, target_level):
char._level_up()
char.level = target_level
char.current_hp = char.max_hp
char.current_mp = char.max_mp
return char
