pretty_money_format.py - master - ilya.petrash/SimplePyScripts

pretty_money_format.py
65 строк2 KB
gil9red
Refactoring. Using f-strings
26 июн 2023, 12:46
26 июн 2023, 12:4665e8cfe
#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = "ipetrash"


import re


def pretty_money_format(money) -> str:
    money = str(money)
    money = re.sub(r"[^.,\d]", "", money)

    if money.count(".") + money.count(",") > 1:
        raise Exception(f'Invalid money format: "{money}".')

    money_sep = ""
    if money.count("."):
        money_sep = "."
    elif money.count(","):
        money_sep = ","

    if money_sep:
        major, minor = money.split(money_sep)
    else:
        major, minor = money, ""

    if len(major) > 3:
        new_major = list()

        i = 0
        for c in reversed(major):
            i += 1

            new_major.append(c)

            if i > 0 and i % 3 == 0:
                new_major.append(" ")

        major = "".join(reversed(new_major))

    money = major + money_sep + minor
    return money.strip()


if __name__ == "__main__":
    assert pretty_money_format(1000) == "1 000"
    assert pretty_money_format(1000.0) == "1 000.0"
    assert pretty_money_format(1000000) == "1 000 000"
    assert pretty_money_format(10000.56) == "10 000.56"
    assert pretty_money_format(1000000.0) == "1 000 000.0"
    assert pretty_money_format(100000) == "100 000"

    assert pretty_money_format("1000") == "1 000"
    assert pretty_money_format("1000.") == "1 000."
    assert pretty_money_format("1000000") == "1 000 000"
    assert pretty_money_format("10000.50") == "10 000.50"
    assert pretty_money_format("1000000.0") == "1 000 000.0"
    assert pretty_money_format("100000") == "100 000"

    assert pretty_money_format("10 00") == "1 000"
    assert pretty_money_format("100 00 00") == "1 000 000"
    assert pretty_money_format("1 0000.50$") == "10 000.50"
    assert pretty_money_format("100 00 00.0") == "1 000 000.0"
    assert pretty_money_format("100!000#") == "100 000"