diff --git a/aurora/importers/aurora.py b/aurora/importers/aurora.py index ced2566..462a7bd 100644 --- a/aurora/importers/aurora.py +++ b/aurora/importers/aurora.py @@ -1,6 +1,5 @@ # pylint: disable=duplicate-code import json -from datetime import timedelta from time import time from typing import Dict @@ -10,6 +9,7 @@ from redbot.core.utils.chat_formatting import box, warning from ..models.moderation import Moderation from ..utilities.database import connect, create_guild_table +from ..utilities.utils import timedelta_from_string class ImportAuroraView(ui.View): @@ -91,8 +91,7 @@ class ImportAuroraView(ui.View): metadata.update({"imported_timestamp": int(time())}) if case["duration"] != "NULL" and case["duration"] is not None: - hours, minutes, seconds = map(int, case["duration"].split(":")) - duration = timedelta(hours=hours, minutes=minutes, seconds=seconds) + duration = timedelta_from_string(case["duration"]) else: duration = None diff --git a/aurora/utilities/utils.py b/aurora/utilities/utils.py index bd130d4..834035d 100644 --- a/aurora/utilities/utils.py +++ b/aurora/utilities/utils.py @@ -194,6 +194,13 @@ def timedelta_from_relativedelta(relativedelta: rd) -> timedelta: then = now - relativedelta return now - then +def timedelta_from_string(string: str) -> timedelta: + """Converts a string to a timedelta object.""" + from .logger import logger + hours, minutes, seconds = map(int, string.split(":")) + logger.debug("%s | hours: %s, minutes: %s, seconds: %s", string, hours, minutes, seconds) + return timedelta(hours=hours, minutes=minutes, seconds=seconds) + def timedelta_to_string(timedelta: timedelta) -> str: """Converts a timedelta object to a string.""" hours, remainder = divmod(timedelta.seconds, 3600)