74 lines
2.6 KiB
Python
74 lines
2.6 KiB
Python
from typing import Generator
|
|
import json
|
|
from os import path, listdir
|
|
|
|
from kawai_focus.schemas import TimerTimeModel
|
|
from kawai_focus.utils.errors import ErrorMessage
|
|
|
|
|
|
class ReadJson:
|
|
"""Класс для чтения всех JSON-файлов в директории и объединения их в словарь."""
|
|
|
|
|
|
def __init__(self, folder_json: str):
|
|
self.folder_json = folder_json
|
|
self._json_data = self._read_all_json()
|
|
|
|
def _read_all_json(self) -> dict:
|
|
"""Читает все JSON-файлы в директории и возвращает объединённый словарь."""
|
|
|
|
if not path.exists(self.folder_json):
|
|
raise FileNotFoundError(ErrorMessage.FOLDER_NOT_FOUND.value)
|
|
|
|
# Получаем список всех файлов с расширением .json
|
|
json_files = [path.join(self.folder_json, file) for file in listdir(self.folder_json) if file.endswith('.json')]
|
|
|
|
if not json_files:
|
|
raise FileNotFoundError(ErrorMessage.JSON_NOT_FOUND.value)
|
|
|
|
data_dict = {}
|
|
|
|
for file_path in json_files:
|
|
with open(file_path, 'r', encoding='utf-8') as file:
|
|
# Обновляем словарь с помощью .update() для эффективности
|
|
data_dict.update(json.load(file))
|
|
|
|
return data_dict
|
|
|
|
def get_text(self, name: str) -> str:
|
|
"""Возвращает текст из словаря по ключу."""
|
|
|
|
return self._json_data.get(name, ErrorMessage.KEY_ERROR.value)
|
|
|
|
|
|
data_json = ReadJson(folder_json='json')
|
|
|
|
|
|
def custom_timer(mm_user: int) -> Generator[str, None, None]:
|
|
"""Функция отсчитывает время, установленное для таймера в формате
|
|
'hh:mm:ss'. Возвращает генератор, который возвращает текущее время
|
|
в формате 'hh:mm:ss'."""
|
|
|
|
total_seconds = mm_user * 60
|
|
|
|
for remaining in range(total_seconds, -1, -1):
|
|
yield f"{remaining // 3600:02d}:{(remaining % 3600) // 60:02d}:{remaining % 60:02d}"
|
|
|
|
|
|
def calculate_time(mm_user: int) -> str:
|
|
"""Функция для подсчёта часов, минут и секунд из минут"""
|
|
|
|
valid_data = None
|
|
hh = mm_user // 60
|
|
mm = mm_user % 60
|
|
|
|
if hh == 0:
|
|
valid_data = TimerTimeModel(mm=mm)
|
|
else:
|
|
valid_data = TimerTimeModel(hh=hh, mm=mm)
|
|
|
|
return (
|
|
f"{'0' + str(valid_data.hh) if valid_data.hh <= 9 else valid_data.hh}:"
|
|
f"{'0' + str(valid_data.mm) if valid_data.mm <= 9 else valid_data.mm}:00"
|
|
)
|