diff --git a/sizebot/lib/quake.py b/sizebot/lib/quake.py index 4653022d..46baa2c6 100644 --- a/sizebot/lib/quake.py +++ b/sizebot/lib/quake.py @@ -58,17 +58,17 @@ def mag_to_name(mag: float) -> str: def scale_to_joules(user: User, g: float, factor: float) -> Decimal: return (Decimal(user.weight / 1000) / 2) * (Decimal(g) * Decimal(user.scale)) * Decimal(factor) -def step_joules(user: User) -> int: +def step_joules(user: User) -> Decimal: return scale_to_joules(user, G, STEP_FACTOR) -def stomp_joules(user: User) -> int: +def stomp_joules(user: User) -> Decimal: return scale_to_joules(user, STOMP_G, STOMP_FACTOR) -def jump_joules(user: User) -> int: +def jump_joules(user: User) -> Decimal: return scale_to_joules(user, G, JUMP_FACTOR) -def breath_joules(user: User) -> int: - return int(BREATH_JOULES * (user.scale ** 4)) +def breath_joules(user: User) -> Decimal: + return Decimal(BREATH_JOULES * (user.scale ** 4)) -def poke_joules(user: User) -> int: - return int(POKE_JOULES * (user.scale ** 4)) +def poke_joules(user: User) -> Decimal: + return Decimal(POKE_JOULES * (user.scale ** 4))