JezerM
4 years ago
11 changed files with 435 additions and 82 deletions
@ -0,0 +1,119 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# |
||||
# globals.py |
||||
# |
||||
# Copyright © 2017 Antergos |
||||
# |
||||
# This file is part of Web Greeter. |
||||
# |
||||
# Web Greeter is free software; you can redistribute it and/or modify |
||||
# it under the terms of the GNU General Public License as published by |
||||
# the Free Software Foundation; either version 3 of the License, or |
||||
# (at your option) any later version. |
||||
# |
||||
# Web Greeter is distributed in the hope that it will be useful, |
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
# GNU General Public License for more details. |
||||
# |
||||
# The following additional terms are in effect as per Section 7 of the license: |
||||
# |
||||
# The preservation of all legal notices and author attributions in |
||||
# the material or in the Appropriate Legal Notices displayed |
||||
# by works containing it is required. |
||||
# |
||||
# You should have received a copy of the GNU General Public License |
||||
# along with Web Greeter; If not, see <http://www.gnu.org/licenses/>. |
||||
|
||||
import sys |
||||
import yaml |
||||
import pkg_resources |
||||
import os |
||||
from typing import ( |
||||
ClassVar, |
||||
Type, |
||||
List, |
||||
Tuple, |
||||
) |
||||
|
||||
# 3rd-Party Libs |
||||
from whither.app import App |
||||
from whither.base.config_loader import ConfigLoader |
||||
from whither.bridge import BridgeObject |
||||
|
||||
# This Application |
||||
import resources |
||||
from bridge import ( |
||||
Config, |
||||
Greeter, |
||||
ThemeUtils, |
||||
) |
||||
|
||||
# Typing Helpers |
||||
BridgeObj = Type[BridgeObject] |
||||
|
||||
|
||||
BASE_DIR = os.path.dirname(os.path.realpath(__file__)) |
||||
CONFIG_FILE = os.path.join(BASE_DIR, 'whither.yml') |
||||
|
||||
class WebGreeter(App): |
||||
greeter = None # type: ClassVar[BridgeObj] | None |
||||
greeter_config = None # type: ClassVar[BridgeObj] | None |
||||
theme_utils = None # type: ClassVar[BridgeObj] | None |
||||
|
||||
def __init__(self, *args, **kwargs) -> None: |
||||
super().__init__('WebGreeter', *args, **kwargs) |
||||
self.logger.debug('Web Greeter started.') |
||||
self.greeter = Greeter(self.config) |
||||
self.greeter_config = Config(self.config) |
||||
self.theme_utils = ThemeUtils(self.greeter, self.config) |
||||
self._web_container.bridge_objects = (self.greeter, self.greeter_config, self.theme_utils) |
||||
|
||||
self._web_container.initialize_bridge_objects() |
||||
self._web_container.load_script(':/_greeter/js/bundle.js', 'Web Greeter Bundle') |
||||
self.load_theme() |
||||
|
||||
@classmethod |
||||
def __pre_init__(cls): |
||||
ConfigLoader.add_filter(cls.validate_greeter_config_data) |
||||
|
||||
def _before_web_container_init(self): |
||||
self.get_and_apply_user_config() |
||||
|
||||
@classmethod |
||||
def validate_greeter_config_data(cls, key: str, data: str) -> str: |
||||
if "'@" not in data: |
||||
return data |
||||
|
||||
if 'WebGreeter' == key: |
||||
path = '../build/web-greeter/whither.yml' |
||||
else: |
||||
path = '../build/dist/web-greeter.yml' |
||||
|
||||
return open(path, 'r').read() |
||||
|
||||
def get_and_apply_user_config(self): |
||||
self.logger.debug("Aplying config") |
||||
config_file = os.path.join(self.config.config_dir, 'web-greeter.yml') |
||||
branding_config = ConfigLoader('branding', config_file).config |
||||
greeter_config = ConfigLoader('greeter', config_file).config |
||||
features_config = ConfigLoader('features', config_file).config |
||||
|
||||
greeter_config.update(custom_config) |
||||
|
||||
self.config.branding.update(branding_config) |
||||
self.config.greeter.update(greeter_config) |
||||
self.config.features.update(features_config) |
||||
|
||||
self._config.debug_mode = greeter_config['debug_mode'] |
||||
self._config.allow_remote_urls = not greeter_config['secure_mode'] |
||||
|
||||
def load_theme(self): |
||||
self.logger.debug('Loading theme...') |
||||
theme_url = '/{0}/{1}/index.html'.format(self.config.themes_dir, self.config.greeter.theme) |
||||
self._web_container.load(theme_url) |
||||
|
||||
global custom_config |
||||
global greeter |
||||
custom_config = {} |
||||
|
||||
@ -0,0 +1,120 @@
|
||||
# -*- coding: utf-8 -*- |
||||
# |
||||
# config.py |
||||
# |
||||
# Copyright © 2021 JezerM |
||||
# |
||||
# This file is part of Web Greeter. |
||||
# |
||||
# Web Greeter is free software; you can redistribute it and/or modify |
||||
# it under the terms of the GNU General Public License as published by |
||||
# the Free Software Foundation; either version 3 of the License, or |
||||
# (at your option) any later version. |
||||
# |
||||
# Web Greeter is distributed in the hope that it will be useful, |
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||
# GNU General Public License for more details. |
||||
# |
||||
# The following additional terms are in effect as per Section 7 of the license: |
||||
# |
||||
# The preservation of all legal notices and author attributions in |
||||
# the material or in the Appropriate Legal Notices displayed |
||||
# by works containing it is required. |
||||
# |
||||
# You should have received a copy of the GNU General Public License |
||||
# along with Web Greeter; If not, see <http://www.gnu.org/licenses/>. |
||||
|
||||
from whither.toolkits.bootstrap import WebPage, MainWindow |
||||
|
||||
from PyQt5.QtCore import QUrl, pyqtSignal, Qt |
||||
from PyQt5.QtWidgets import QDialogButtonBox, QDialog, QVBoxLayout, QLabel, QPushButton, QAbstractButton |
||||
from PyQt5.QtGui import QKeyEvent |
||||
|
||||
import globals |
||||
|
||||
def javaScriptConsoleMessage(self, level: WebPage.JavaScriptConsoleMessageLevel, message: str, lineNumber: int, sourceID: str): |
||||
if sourceID == "": |
||||
sourceID = "console" |
||||
|
||||
error = False |
||||
typeLog = "" |
||||
if level == WebPage.JavaScriptConsoleMessageLevel.ErrorMessageLevel: |
||||
typeLog = "[ERROR]" |
||||
error = True |
||||
elif level == WebPage.JavaScriptConsoleMessageLevel.WarningMessageLevel: |
||||
typeLog = "[WARNING]" |
||||
elif level == WebPage.JavaScriptConsoleMessageLevel.InfoMessageLevel: |
||||
typeLog = "[LOG]" |
||||
else: |
||||
return |
||||
|
||||
logMessage = "{typ} {source} {line}: {msg}".format(typ = typeLog, msg = message, source = sourceID, line = lineNumber) |
||||
print(logMessage) |
||||
if error: |
||||
errorMessage = "{source} {line}: {msg}".format(source = sourceID, line = lineNumber, msg = message) |
||||
errorPrompt(errorMessage) |
||||
|
||||
class ErrorDialog(QDialog): |
||||
def __init__(self, parent=None, err=""): |
||||
super().__init__(parent) |
||||
|
||||
self.setWindowTitle("Error") |
||||
|
||||
self.buttonBox = QDialogButtonBox() |
||||
cancelBtn = QPushButton("Cancel") |
||||
defaultBtn = QPushButton("Set default theme") |
||||
reloadBtn = QPushButton("Reload theme") |
||||
|
||||
reloadBtn.clicked.connect(self.handle_reload) |
||||
|
||||
self.buttonBox.addButton(defaultBtn, QDialogButtonBox.ButtonRole.AcceptRole) |
||||
self.buttonBox.addButton(reloadBtn, QDialogButtonBox.ButtonRole.ResetRole) |
||||
self.buttonBox.addButton(cancelBtn, QDialogButtonBox.ButtonRole.RejectRole) |
||||
|
||||
self.buttonBox.accepted.connect(self.accept) |
||||
self.buttonBox.rejected.connect(self.reject) |
||||
|
||||
self.layout = QVBoxLayout() |
||||
message = QLabel("An error ocurred. Do you want to change to default theme?") |
||||
err = QLabel(err) |
||||
self.layout.addWidget(message) |
||||
self.layout.addWidget(err) |
||||
self.layout.addWidget(self.buttonBox) |
||||
self.setLayout(self.layout) |
||||
|
||||
def handle_reload(self, value: bool): |
||||
self.done(2) |
||||
|
||||
def errorPrompt(err): |
||||
print("ERROR PROMPT") |
||||
dia = ErrorDialog(globals.greeter._main_window.widget.centralWidget(), err) |
||||
|
||||
dia.exec() |
||||
result = dia.result() |
||||
|
||||
if result == 0: # Cancel |
||||
return |
||||
elif result == 1: # Default theme |
||||
globals.custom_config["theme"] = "default" |
||||
globals.greeter.get_and_apply_user_config() |
||||
globals.greeter.load_theme() |
||||
return |
||||
elif result == 2: # Reload |
||||
globals.greeter.load_theme() |
||||
return |
||||
|
||||
return |
||||
|
||||
def keyPressEvent(self, keyEvent: QKeyEvent): |
||||
super(MainWindow, self).keyPressEvent(keyEvent) |
||||
if (keyEvent.key() == Qt.Key.Key_MonBrightnessUp): |
||||
globals.greeter.greeter.brightnessIncrease(globals.greeter.config.features.backlight["value"]) |
||||
if (keyEvent.key() == Qt.Key.Key_MonBrightnessDown): |
||||
globals.greeter.greeter.brightnessDecrease(globals.greeter.config.features.backlight["value"]) |
||||
|
||||
|
||||
MainWindow.keyPressEvent = keyPressEvent |
||||
|
||||
WebPage.javaScriptConsoleMessage = javaScriptConsoleMessage # Yep, you can override functions like this!!! |
||||
|
||||
Loading…
Reference in new issue