121 lines
3.2 KiB
Python
121 lines
3.2 KiB
Python
|
from adventure.models.entity import Attributes, AttributeValue
|
||
|
|
|
||
|
|
|
||
|
|
def add_attribute(attributes: Attributes, name: str, value: int | float) -> Attributes:
|
||
|
|
"""
|
||
|
|
Add an attribute to a set of attributes.
|
||
|
|
"""
|
||
|
|
if name in attributes:
|
||
|
|
previous_value = attributes[name]
|
||
|
|
if isinstance(previous_value, str):
|
||
|
|
raise ValueError(f"Cannot add a number to a string attribute: {name}")
|
||
|
|
|
||
|
|
attributes[name] = value + previous_value
|
||
|
|
else:
|
||
|
|
attributes[name] = value
|
||
|
|
|
||
|
|
return attributes
|
||
|
|
|
||
|
|
|
||
|
|
def subtract_attribute(
|
||
|
|
attributes: Attributes, name: str, value: int | float
|
||
|
|
) -> Attributes:
|
||
|
|
"""
|
||
|
|
Subtract an attribute from a set of attributes.
|
||
|
|
"""
|
||
|
|
if name in attributes:
|
||
|
|
previous_value = attributes[name]
|
||
|
|
if isinstance(previous_value, str):
|
||
|
|
raise ValueError(
|
||
|
|
f"Cannot subtract a number from a string attribute: {name}"
|
||
|
|
)
|
||
|
|
|
||
|
|
attributes[name] = value - previous_value
|
||
|
|
else:
|
||
|
|
attributes[name] = -value
|
||
|
|
|
||
|
|
return attributes
|
||
|
|
|
||
|
|
|
||
|
|
def multiply_attribute(
|
||
|
|
attributes: Attributes, name: str, value: int | float
|
||
|
|
) -> Attributes:
|
||
|
|
"""
|
||
|
|
Multiply an attribute in a set of attributes.
|
||
|
|
"""
|
||
|
|
if name in attributes:
|
||
|
|
previous_value = attributes[name]
|
||
|
|
if isinstance(previous_value, str):
|
||
|
|
raise ValueError(f"Cannot multiply a string attribute: {name}")
|
||
|
|
|
||
|
|
attributes[name] = previous_value * value
|
||
|
|
else:
|
||
|
|
attributes[name] = 0
|
||
|
|
|
||
|
|
return attributes
|
||
|
|
|
||
|
|
|
||
|
|
def divide_attribute(
|
||
|
|
attributes: Attributes, name: str, value: int | float
|
||
|
|
) -> Attributes:
|
||
|
|
"""
|
||
|
|
Divide an attribute in a set of attributes.
|
||
|
|
"""
|
||
|
|
if name in attributes:
|
||
|
|
previous_value = attributes[name]
|
||
|
|
if isinstance(previous_value, str):
|
||
|
|
raise ValueError(f"Cannot divide a string attribute: {name}")
|
||
|
|
|
||
|
|
attributes[name] = previous_value / value
|
||
|
|
else:
|
||
|
|
attributes[name] = 0
|
||
|
|
|
||
|
|
return attributes
|
||
|
|
|
||
|
|
|
||
|
|
def set_attribute(
|
||
|
|
attributes: Attributes, name: str, value: AttributeValue
|
||
|
|
) -> Attributes:
|
||
|
|
"""
|
||
|
|
Set an attribute in a set of attributes.
|
||
|
|
"""
|
||
|
|
attributes[name] = value
|
||
|
|
|
||
|
|
return attributes
|
||
|
|
|
||
|
|
|
||
|
|
def append_attribute(attributes: Attributes, name: str, value: str) -> Attributes:
|
||
|
|
"""
|
||
|
|
Append a string to an attribute in a set of attributes.
|
||
|
|
"""
|
||
|
|
if name in attributes:
|
||
|
|
previous_value = attributes[name]
|
||
|
|
if isinstance(previous_value, str):
|
||
|
|
attributes[name] = previous_value + value
|
||
|
|
else:
|
||
|
|
raise ValueError(
|
||
|
|
f"Cannot append a string to a non-string attribute: {name}"
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
attributes[name] = value
|
||
|
|
|
||
|
|
return attributes
|
||
|
|
|
||
|
|
|
||
|
|
def prepend_attribute(attributes: Attributes, name: str, value: str) -> Attributes:
|
||
|
|
"""
|
||
|
|
Prepend a string to an attribute in a set of attributes.
|
||
|
|
"""
|
||
|
|
if name in attributes:
|
||
|
|
previous_value = attributes[name]
|
||
|
|
if isinstance(previous_value, str):
|
||
|
|
attributes[name] = value + previous_value
|
||
|
|
else:
|
||
|
|
raise ValueError(
|
||
|
|
f"Cannot prepend a string to a non-string attribute: {name}"
|
||
|
|
)
|
||
|
|
else:
|
||
|
|
attributes[name] = value
|
||
|
|
|
||
|
|
return attributes
|