homework.py - master - brootos/hw_python_oop

homework.py
172 строки6 KB
Дмитрий Клепиков
Фикс от 2023
04 окт 2023, 08:26
04 окт 2023, 08:26041c5dc
import datetime as dt
from typing import Optional, Union

DATE_FMT: str = '%d.%m.%Y'


class Record:
    """Класс для хранения записей о расходах или калориях."""

    def __init__(
        self,
        amount: int,
        comment: str,
        date: Optional[str] = None
    ) -> None:
        """
        Инициализация записи.

        :param amount: Сумма (количество потраченных денег или калорий).
        :param comment: Комментарий к записи.
        :param date: Дата записи в формате 'дд.мм.гггг' (по умолчанию - текущая дата).
        """

        self.amount = amount
        self.comment = comment
        if date is not None:
            self.date = dt.datetime.strptime(date, DATE_FMT).date()
        else:
            self.date = dt.datetime.now().date()


class Calculator:
    """Базовый класс калькулятора калорий и денег."""

    def __init__(self, limit: Union[float, int]):
        """
        Инициализация калькулятора.

        :param limit: Лимит (бюджет или калорийность) на день.
        """

        self.limit = limit
        self.records: list[Record] = []

    def add_record(self, record: Record) -> None:
        """
        Добавление записи в калькулятор.

        :param record: Запись о расходе или потреблении.
        """

        self.records.append(record)

    def get_today_stats(self) -> Union[int, float]:
        """
        Получение суммы расходов (или потребленных калорий) за текущий день.

        :return: Сумма расходов за текущий день.
        """

        return sum(
            record.amount for record in self.records
            if record.date == dt.date.today()
        )

    def get_week_stats(self) -> Union[int, float]:
        """
        Получение суммы расходов (или потребленных калорий) за последнюю неделю.

        :return: Сумма расходов за последние 7 дней.
        """

        today = dt.date.today()
        week_start = today - dt.timedelta(days=7)
        return sum(
            record.amount for record in self.records
            if week_start <= record.date <= today
        )

    def get_limit_today(self) -> Union[int, float]:
        """
        Получение остатка лимита (бюджета или калорийности) на текущий день.

        :return: Остаток лимита на текущий день.
        """

        return self.limit - self.get_today_stats()


class CaloriesCalculator(Calculator):
    """Дочерний класс калькулятора калорий."""

    def get_calories_remained(self) -> str:
        """
        Получение рекомендации по потреблению калорий на текущий день.

        :return: Рекомендация по калориям на текущий день.
        """

        return (
            f'Сегодня можно съесть что-нибудь ещё, но с общей калорийностью не более {limit_today} кКал'
            if (limit_today := self.get_limit_today()) > 0
            else 'Хватит есть!'
        )


class CashCalculator(Calculator):
    """Дочерний класс калькулятора денег."""

    USD_RATE: float = 92.0
    EURO_RATE: float = 102.0
    RUB_RATE: float = 1.0
    CALC_ACCURACY: int = 2

    def get_today_cash_remained(self, currency: str) -> str:
        """
        Получение рекомендации по расходам в день и оставшейся сумме в заданной валюте.

        :param currency: Валюта ('rub', 'usd' или 'eur').

        :return: Рекомендация по расходам и остаток средств в указанной валюте.
        """

        money: dict[str, tuple[float, str]] = {
            'rub': (self.RUB_RATE, 'руб'),
            'usd': (self.USD_RATE, 'USD'),
            'eur': (self.EURO_RATE, 'Euro')
        }

        if currency not in money:
            return '<выбрана неверная валюта>'

        limit_today = self.get_limit_today()

        if limit_today == 0:
            return 'Денег нет, держись'

        rate, currency_name = money[currency]

        cash_today = round(abs(limit_today) / rate, self.CALC_ACCURACY)

        if limit_today > 0:
            return f'На сегодня осталось {cash_today} {currency_name}'
        return f'Денег нет, держись: твой долг - {cash_today} {currency_name}'


if __name__ == "__main__":
    limit = 1000
    cash_calculator = CashCalculator(limit)
    calories_calculator = CaloriesCalculator(limit)

    # записи для денег
    r1 = Record(amount=145, comment='кофе')
    r2 = Record(amount=300, comment='Серёге за обед')
    r3 = Record(amount=3000, comment='Бар на Танин день рождения', date='08.11.2022')

    # записи для калорий
    r4 = Record(amount=118, comment='Кусок тортика. И ещё один.')
    r5 = Record(amount=84, comment='Йогурт.')
    r6 = Record(amount=1140, comment='Баночка чипсов.', date='24.02.2019')

    cash_calculator.add_record(r1)
    cash_calculator.add_record(r2)
    cash_calculator.add_record(r3)

    calories_calculator.add_record(r4)
    calories_calculator.add_record(r5)
    calories_calculator.add_record(r6)

    # вывод результатов
    print(cash_calculator.get_today_cash_remained('rub'))
    print(calories_calculator.get_calories_remained())