ee9bcb84b7
* create tool progression feature and unwrap option * replace option usage with calling feature * add comment explaining why some logic is a weird place * replace item creation logic with feature * self review and add unit tests * rename test cuz I named them too long * add a test for the trash can useful stuff cuz I thought there was a bug but turns out it works * self review again * remove price_multiplier, turns out it's unused during generation * damn it 3.11 why are you like this * use blacksmith region when checking vanilla tools * fix rule * move can mine using in tool logic * remove changes to performance test * properly set the option I guess * properly set options 2 * that's what happen when you code too late
69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
from abc import ABC
|
|
from collections import Counter
|
|
from collections.abc import Mapping
|
|
from dataclasses import dataclass, field
|
|
from functools import cache
|
|
from types import MappingProxyType
|
|
from typing import ClassVar
|
|
|
|
from ...strings.tool_names import Tool
|
|
|
|
|
|
def to_progressive_item(tool: str) -> str:
|
|
"""Return the name of the progressive item."""
|
|
return f"Progressive {tool}"
|
|
|
|
|
|
# The golden scythe is always randomized
|
|
VANILLA_TOOL_DISTRIBUTION = MappingProxyType({
|
|
Tool.scythe: 1,
|
|
})
|
|
|
|
PROGRESSIVE_TOOL_DISTRIBUTION = MappingProxyType({
|
|
Tool.axe: 4,
|
|
Tool.hoe: 4,
|
|
Tool.pickaxe: 4,
|
|
Tool.pan: 4,
|
|
Tool.trash_can: 4,
|
|
Tool.watering_can: 4,
|
|
Tool.fishing_rod: 4,
|
|
})
|
|
|
|
# Masteries add another tier to the scythe and the fishing rod
|
|
SKILL_MASTERIES_TOOL_DISTRIBUTION = MappingProxyType({
|
|
Tool.scythe: 1,
|
|
Tool.fishing_rod: 1,
|
|
})
|
|
|
|
|
|
@cache
|
|
def get_tools_distribution(progressive_tools_enabled: bool, skill_masteries_enabled: bool) -> Mapping[str, int]:
|
|
distribution = Counter(VANILLA_TOOL_DISTRIBUTION)
|
|
|
|
if progressive_tools_enabled:
|
|
distribution += PROGRESSIVE_TOOL_DISTRIBUTION
|
|
|
|
if skill_masteries_enabled:
|
|
distribution += SKILL_MASTERIES_TOOL_DISTRIBUTION
|
|
|
|
return MappingProxyType(distribution)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class ToolProgressionFeature(ABC):
|
|
is_progressive: ClassVar[bool]
|
|
tool_distribution: Mapping[str, int]
|
|
|
|
to_progressive_item = staticmethod(to_progressive_item)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class ToolProgressionVanilla(ToolProgressionFeature):
|
|
is_progressive = False
|
|
# FIXME change the default_factory to a simple default when python 3.11 is no longer supported
|
|
tool_distribution: Mapping[str, int] = field(default_factory=lambda: VANILLA_TOOL_DISTRIBUTION)
|
|
|
|
|
|
class ToolProgressionProgressive(ToolProgressionFeature):
|
|
is_progressive = True
|