sources/uUtils.py - master - VolcanoByte/vbCinema

sources/uUtils.py
161 строка5 KB
Eugene Gavrilyuk
Update sync
13 июл 2026, 15:57
13 июл 2026, 15:5724fdc53
"""
Различные вспомогательные функции
"""
import os
import datetime as dt
from zipfile import ZipFile
from PyQt6.QtWidgets import (QMessageBox, QFileDialog, QDialog, QLabel,
                             QVBoxLayout)
from PyQt6.QtGui import QMovie
from PyQt6.QtCore import Qt
from PyQt6.QtWidgets import QTableView

from uConsts import AppConsts, FILE_FILTERS, ZIP_LEVEL, ZIP_COMPRESSION
from uGlobal import MessageLanguageConsts as mlc


class PopUp(QDialog):
    def __init__(self, parent):
        super().__init__(parent)

        self.setWindowFlag(Qt.WindowType.FramelessWindowHint)
        self.resize(406, 216)

        self.setLayout(QVBoxLayout())
        self.label = QLabel(self)
        self.layout().addWidget(self.label)

        # Интегрировать QMovie к метке и инициировать GIF
        self.movie = QMovie("sync.gif")
        self.label.setMovie(self.movie)
        self.movie.start()


def getDict(d: dict, key: str, default):
    """Функция нужна для того, что бы быть уверенным, что из словаря получим
    значение корректного типа"""
    _res = d.get(key, default)
    if _res is None or type(_res) != type(default):
        return default
    else:
        return _res


def getBool(value) -> bool | None:
    """Функция для QSettings"""
    if isinstance(value, str):
        return value == 'true'
    else:
        return value


def getInt(value) -> int | None:
    """Функция для QSettings"""
    if value:
        if isinstance(value, str):
            return int(value)
        else:
            return value
    else:
        return 0


def getFloat(value) -> int | None:
    """Функция для QSettings"""
    if isinstance(value, str):
        return float(value)
    else:
        return value


def timestamp2FileName(ts: float, syncId: str) -> str | None:
    """Преобразование timestamp в имя файла-бэкапа,
        типа 2025-12-20_16-28.Johnny_cinema"""
    if type(ts) == float and ts:
        _dt = dt.datetime.fromtimestamp(ts)
        return dt.datetime.strftime(_dt, f"%Y-%m-%d_%H-%M.{syncId}_"
                                         f"{AppConsts.CollectionKind}")


def zipDB(dbfilename: str, archive: str):
    """Упаковывает базу в архив. Возвращает True, если архив создан."""
    name = os.path.basename(dbfilename)
    with ZipFile(archive, 'w') as zipFile:
        zipFile.write(dbfilename, name, compress_type=ZIP_COMPRESSION,
                      compresslevel=ZIP_LEVEL)


def unzipDB(dbfilename: str, archive: str):
    """Распаковывает файлы из архива в указанную папку. Возвращает True,
    если папка существует."""
    with ZipFile(archive, 'r') as zipFile:
        zipFile.extractall(dbfilename)


def errorDlg(parent, message: str, caption: str = mlc.error):
    msg = QMessageBox(parent)
    msg.setWindowTitle(caption)
    msg.setText(message)
    msg.setIcon(QMessageBox.Icon.Critical)
    msg.exec()


def infoDlg(parent, message: str, caption: str = mlc.information):
    msg = QMessageBox(parent)
    msg.setWindowTitle(caption)
    msg.setText(message)
    msg.setIcon(QMessageBox.Icon.Information)
    msg.exec()


def confirmOkCancelDlg(parent, message: str, caption: str = mlc.confirm):
    msg = QMessageBox(parent)
    msg.setWindowTitle(caption)
    msg.setText(message)
    msg.setIcon(QMessageBox.Icon.Question)
    msg.setStandardButtons(QMessageBox.StandardButton.Ok |
                           QMessageBox.StandardButton.Cancel)
    msg.setDefaultButton(QMessageBox.StandardButton.Ok)
    return msg.exec()


def confirmYesNoCancelDlg(parent, message: str, caption: str = mlc.confirm):
    msg = QMessageBox(parent)
    msg.setWindowTitle(caption)
    msg.setText(message)
    msg.setIcon(QMessageBox.Icon.Question)
    msg.setStandardButtons(QMessageBox.StandardButton.Yes |
                           QMessageBox.StandardButton.No |
                           QMessageBox.StandardButton.Cancel)
    msg.setDefaultButton(QMessageBox.StandardButton.Ok)
    return msg.exec()


def getFileNameDialog(caption: str, accept: QFileDialog.AcceptMode,
                      mode: QFileDialog.FileMode) -> str | None:
    """Подготовка диалога и выбор файла"""
    dialog = QFileDialog()
    dialog.setWindowTitle(caption)
    dialog.setNameFilters(FILE_FILTERS)
    dialog.selectNameFilter(FILE_FILTERS[0])
    dialog.setAcceptMode(accept)
    dialog.setFileMode(mode)

    ok = dialog.exec()
    if ok:
        return dialog.selectedFiles()[0]


def getColumnsSize(table: QTableView) -> list:
    """Возвращает список размеров колонок"""
    result = list()
    for column in range(table.model().columnCount()):
        result.append(table.columnWidth(column))

    return result


def setColumnsSize(table: QTableView, columns):
    """Устанавливает размеры колонок"""
    for column in range(table.model().columnCount()):
        table.setColumnWidth(column, getInt(columns[column]))