ecc1bd8d5c
Execute the edit callback function in a thread. By executing it in a thread we don't have to wait for longer validation procedures to finish to continue updating the UI. This is most notable in the MoveGamePopUp which is heavy on disk IO. Because we cannot use special text formatting in a thread, the indicator messages have been reworked while also becoming extensible. A dictionary of extended reasons can be specified through the `IndicatorLineEdit.extend_reasons()` method. The dictionary has to follow the following format ``` python { MyIndicatorReasons.REASON: self.tr("Reason message") MyIndicatorReasons.OTHER_REASON: self.tr("Other reason message") } ``` In the above example `MyIndicatorReasons` is a subclass of `IndicatorReasons` which should be specified as follows ``` python MyIndicatorReasons(IndicatorReasons): REASON = auto() OTHER_REASON = auto() ```
215 lines
5.8 KiB
Python
215 lines
5.8 KiB
Python
import os
|
|
import platform
|
|
from logging import getLogger
|
|
from typing import List, Union, Type
|
|
|
|
import qtawesome
|
|
import requests
|
|
from PyQt5.QtCore import (
|
|
pyqtSignal,
|
|
QObject,
|
|
QRunnable,
|
|
QSettings,
|
|
QFile,
|
|
QDir,
|
|
Qt,
|
|
)
|
|
from PyQt5.QtGui import QPalette, QColor, QFontMetrics
|
|
from PyQt5.QtWidgets import qApp, QStyleFactory, QLabel
|
|
from PyQt5.sip import wrappertype
|
|
from legendary.core import LegendaryCore
|
|
from legendary.models.game import Game
|
|
from requests.exceptions import HTTPError
|
|
|
|
from rare.models.apiresults import ApiResults
|
|
from rare.utils.paths import resources_path
|
|
|
|
logger = getLogger("Utils")
|
|
settings = QSettings("Rare", "Rare")
|
|
|
|
color_role_map = {
|
|
0: "WindowText",
|
|
1: "Button",
|
|
2: "Light",
|
|
3: "Midlight",
|
|
4: "Dark",
|
|
5: "Mid",
|
|
6: "Text",
|
|
7: "BrightText",
|
|
8: "ButtonText",
|
|
9: "Base",
|
|
10: "Window",
|
|
11: "Shadow",
|
|
12: "Highlight",
|
|
13: "HighlightedText",
|
|
14: "Link",
|
|
15: "LinkVisited",
|
|
16: "AlternateBase",
|
|
# 17: "NoRole",
|
|
18: "ToolTipBase",
|
|
19: "ToolTipText",
|
|
20: "PlaceholderText",
|
|
# 21: "NColorRoles",
|
|
}
|
|
|
|
color_group_map = {
|
|
0: "Active",
|
|
1: "Disabled",
|
|
2: "Inactive",
|
|
}
|
|
|
|
|
|
def load_color_scheme(path: str) -> QPalette:
|
|
palette = QPalette()
|
|
scheme = QSettings(path, QSettings.IniFormat)
|
|
try:
|
|
scheme.beginGroup("ColorScheme")
|
|
for g in color_group_map:
|
|
scheme.beginGroup(color_group_map[g])
|
|
group = QPalette.ColorGroup(g)
|
|
for r in color_role_map:
|
|
role = QPalette.ColorRole(r)
|
|
color = scheme.value(color_role_map[r], None)
|
|
if color is not None:
|
|
palette.setColor(group, role, QColor(color))
|
|
else:
|
|
palette.setColor(group, role, palette.color(QPalette.Active, role))
|
|
scheme.endGroup()
|
|
scheme.endGroup()
|
|
except:
|
|
palette = None
|
|
return palette
|
|
|
|
|
|
def set_color_pallete(color_scheme: str):
|
|
if not color_scheme:
|
|
qApp.setStyle(QStyleFactory.create(qApp.property("rareDefaultQtStyle")))
|
|
qApp.setStyleSheet("")
|
|
qApp.setPalette(qApp.style().standardPalette())
|
|
return
|
|
qApp.setStyle(QStyleFactory.create("Fusion"))
|
|
custom_palette = load_color_scheme(f":/schemes/{color_scheme}")
|
|
if custom_palette is not None:
|
|
qApp.setPalette(custom_palette)
|
|
icon_color = qApp.palette().color(QPalette.Foreground).name()
|
|
qtawesome.set_defaults(color=icon_color)
|
|
|
|
|
|
def get_color_schemes() -> List[str]:
|
|
colors = []
|
|
for file in QDir(":/schemes"):
|
|
colors.append(file)
|
|
return colors
|
|
|
|
|
|
def set_style_sheet(style_sheet: str):
|
|
if not style_sheet:
|
|
qApp.setStyle(QStyleFactory.create(qApp.property("rareDefaultQtStyle")))
|
|
qApp.setStyleSheet("")
|
|
return
|
|
qApp.setStyle(QStyleFactory.create("Fusion"))
|
|
file = QFile(f":/stylesheets/{style_sheet}/stylesheet.qss")
|
|
file.open(QFile.ReadOnly)
|
|
stylesheet = file.readAll().data().decode("utf-8")
|
|
|
|
qApp.setStyleSheet(stylesheet)
|
|
icon_color = qApp.palette().color(QPalette.Text).name()
|
|
qtawesome.set_defaults(color="#eeeeee")
|
|
|
|
|
|
def get_style_sheets() -> List[str]:
|
|
styles = []
|
|
for file in QDir(":/stylesheets/"):
|
|
styles.append(file)
|
|
return styles
|
|
|
|
|
|
def get_translations():
|
|
langs = ["en"]
|
|
for i in os.listdir(os.path.join(resources_path, "languages")):
|
|
if i.endswith(".qm") and not i.startswith("qt_"):
|
|
langs.append(i.split(".")[0])
|
|
return langs
|
|
|
|
|
|
def get_latest_version():
|
|
try:
|
|
resp = requests.get(
|
|
"https://api.github.com/repos/Dummerle/Rare/releases/latest"
|
|
)
|
|
tag = resp.json()["tag_name"]
|
|
return tag
|
|
except requests.exceptions.ConnectionError:
|
|
return "0.0.0"
|
|
|
|
|
|
def get_size(b: Union[int, float]) -> str:
|
|
for s in ["", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei"]:
|
|
if b < 1024:
|
|
return f"{b:.2f} {s}B"
|
|
b /= 1024
|
|
|
|
|
|
class CloudWorker(QRunnable):
|
|
class Signals(QObject):
|
|
# List[SaveGameFile]
|
|
result_ready = pyqtSignal(list)
|
|
|
|
def __init__(self, core: LegendaryCore):
|
|
super(CloudWorker, self).__init__()
|
|
self.core = core
|
|
self.signals = CloudWorker.Signals()
|
|
self.setAutoDelete(True)
|
|
|
|
def run(self) -> None:
|
|
try:
|
|
result = self.core.get_save_games()
|
|
except HTTPError:
|
|
result = None
|
|
self.signals.result_ready.emit(result)
|
|
|
|
|
|
def get_raw_save_path(game: Game):
|
|
if game.supports_cloud_saves:
|
|
return (
|
|
game.metadata.get("customAttributes", {})
|
|
.get("CloudSaveFolder", {})
|
|
.get("value")
|
|
)
|
|
|
|
|
|
def get_default_platform(app_name, api_results: ApiResults):
|
|
if platform.system() != "Darwin" or app_name not in api_results.mac_games:
|
|
return "Windows"
|
|
else:
|
|
return "Mac"
|
|
|
|
|
|
def icon(icn_str: str, fallback: str = None, **kwargs):
|
|
try:
|
|
return qtawesome.icon(icn_str, **kwargs)
|
|
except Exception as e:
|
|
if not fallback:
|
|
logger.warning(f"{e} {icn_str}")
|
|
if fallback:
|
|
try:
|
|
return qtawesome.icon(fallback, **kwargs)
|
|
except Exception as e:
|
|
logger.error(str(e))
|
|
if kwargs.get("color"):
|
|
kwargs["color"] = "red"
|
|
return qtawesome.icon("ei.error", **kwargs)
|
|
|
|
|
|
def widget_object_name(widget: Union[wrappertype,QObject,Type], suffix: str) -> str:
|
|
if isinstance(widget, QObject):
|
|
return f"{type(widget).__name__}_{suffix}"
|
|
elif isinstance(widget, wrappertype):
|
|
return f"{widget.__name__}_{suffix}"
|
|
else:
|
|
raise RuntimeError(f"Argument {widget} not a QObject or type of QObject")
|
|
|
|
|
|
def elide_text(label: QLabel, text: str) -> str:
|
|
metrics = QFontMetrics(label.font())
|
|
return metrics.elidedText(text, Qt.ElideRight, label.sizeHint().width())
|