import re
from datetime import timedelta

from discord.ext.commands import BadArgument


def time_pars(s: str) -> timedelta:
    match = re.fullmatch(r"(?:([0-9]+)W)*(?:([0-9]+)D)*(?:([0-9]+)H)*(?:([0-9]+)M)*(?:([0-9]+)S)*",
                         s.upper().replace(" ", "").strip())
    if match:
        w, d, h, m, s = match.groups()
        if any([w, d, h, m, s]):
            w, d, h, m, s = [i if i else 0 for i in [w, d, h, m, s]]
            return timedelta(weeks=int(w), days=int(d), hours=int(h), minutes=int(m), seconds=int(s))
    raise BadArgument()


def seconds_to_time_string(seconds: float) -> str:
    days, seconds = divmod(seconds, 86400)
    hours, seconds = divmod(seconds, 3600)
    minutes, seconds = divmod(seconds, 60)
    return f"""{f"{days}d {hours}h {minutes}m {seconds}s"
    if days > 0 else f"{hours}h {minutes}m {seconds}s"
    if hours > 0 else f"{minutes}m {seconds}s"
    if minutes > 0 else f"{seconds}s"}"""