You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

356 lines
11 KiB

# -*- coding: utf-8 -*-
#
# Greeter.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/>.
# Standard Lib
import subprocess
import re
# 3rd-Party Libs
import gi
gi.require_version('LightDM', '1')
from gi.repository import LightDM
from whither.bridge import (
BridgeObject,
bridge,
Variant,
)
from PyQt5.QtCore import QTimer
# This Application
from . import (
language_to_dict,
layout_to_dict,
session_to_dict,
user_to_dict,
battery_to_dict,
debugLog
)
LightDMGreeter = LightDM.Greeter()
LightDMUsers = LightDM.UserList()
def changeBrightness(self, method: str, quantity: int):
if self._config.features.backlight["enabled"] != True:
return
try:
steps = self._config.features.backlight["steps"]
child = subprocess.run(["xbacklight", method, str(quantity), "-steps", str(steps)])
if child.returncode == 1:
raise ChildProcessError("xbacklight returned 1")
except Exception as err:
debugLog("Brightness: {}".format(err), 4)
else:
self.brightness_update.emit()
pass
def getBrightness(self):
if self._config.features.backlight["enabled"] != True:
return -1
try:
level = subprocess.run(["xbacklight", "-get"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
return int(level.stdout)
except Exception as err:
debugLog("Battery: {}".format(err), 4)
return -1
def updateBattery(self):
if self._config.features.battery != True:
return
try:
acpi = subprocess.run(["acpi", "-b"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True)
battery = acpi.stdout.split(": ")
data = re.sub("%|,", "", battery[1])
level = data.split(" ")[1]
self._battery = int(level)
self._acpi = acpi.stdout
except Exception as err:
debugLog("Battery: {}".format(err), 4)
else:
self.property_changed.emit()
class Greeter(BridgeObject):
# LightDM.Greeter Signals
authentication_complete = bridge.signal()
autologin_timer_expired = bridge.signal()
idle = bridge.signal()
reset = bridge.signal()
show_message = bridge.signal(str, LightDM.MessageType, arguments=('text', 'type'))
show_prompt = bridge.signal(str, LightDM.PromptType, arguments=('text', 'type'))
brightness_update = bridge.signal()
noop_signal = bridge.signal()
property_changed = bridge.signal()
_battery = -1
_acpi = ""
def __init__(self, config, *args, **kwargs):
super().__init__(name='LightDMGreeter', *args, **kwargs)
self._config = config
self._shared_data_directory = ''
self._themes_directory = config.themes_dir
LightDMGreeter.connect_to_daemon_sync()
self._connect_signals()
self._determine_shared_data_directory_path()
def _determine_shared_data_directory_path(self):
user = LightDMUsers.get_users()[0]
user_data_dir = LightDMGreeter.ensure_shared_data_dir_sync(user.get_name())
self._shared_data_directory = user_data_dir.rpartition('/')[0]
def _connect_signals(self):
LightDMGreeter.connect(
'authentication-complete',
lambda greeter: self._emit_signal(self.authentication_complete)
)
LightDMGreeter.connect(
'autologin-timer-expired',
lambda greeter: self._emit_signal(self.autologin_timer_expired)
)
LightDMGreeter.connect('idle', lambda greeter: self._emit_signal(self.idle))
LightDMGreeter.connect('reset', lambda greeter: self._emit_signal(self.reset))
LightDMGreeter.connect(
'show-message',
lambda greeter, msg, mtype: self._emit_signal(self.show_message, msg, mtype)
)
LightDMGreeter.connect(
'show-prompt',
lambda greeter, msg, mtype: self._emit_signal(self.show_prompt, msg, mtype)
)
def _emit_signal(self, _signal, *args):
self.property_changed.emit()
QTimer().singleShot(300, lambda: _signal.emit(*args))
@bridge.prop(str, notify=property_changed)
def authentication_user(self):
return LightDMGreeter.get_authentication_user() or ''
@bridge.prop(bool, notify=noop_signal)
def autologin_guest(self):
return LightDMGreeter.get_autologin_guest_hint()
@bridge.prop(int, notify=noop_signal)
def autologin_timeout(self):
return LightDMGreeter.get_autologin_timeout_hint()
@bridge.prop(str, notify=noop_signal)
def autologin_user(self):
return LightDMGreeter.get_autologin_user_hint()
@bridge.prop(Variant, notify=property_changed)
def batteryData(self):
return battery_to_dict(self._acpi)
@bridge.prop(int, notify=brightness_update)
def brightness(self):
return getBrightness(self)
@bridge.prop(bool, notify=noop_signal)
def can_hibernate(self):
return LightDM.get_can_hibernate()
@bridge.prop(bool, notify=noop_signal)
def can_restart(self):
return LightDM.get_can_restart()
@bridge.prop(bool, notify=noop_signal)
def can_shutdown(self):
return LightDM.get_can_shutdown()
@bridge.prop(bool, notify=noop_signal)
def can_suspend(self):
return LightDM.get_can_suspend()
@bridge.prop(bool, notify=noop_signal)
def can_access_brightness(self):
return self._config.features.backlight["enabled"]
@bridge.prop(bool, notify=noop_signal)
def can_access_battery(self):
return self._config.features.battery
@bridge.prop(str, notify=noop_signal)
def default_session(self):
return LightDMGreeter.get_default_session_hint()
@bridge.prop(bool, notify=noop_signal)
def has_guest_account(self):
return LightDMGreeter.get_has_guest_account_hint()
@bridge.prop(bool, notify=noop_signal)
def hide_users_hint(self):
return LightDMGreeter.get_hide_users_hint()
@bridge.prop(str, notify=noop_signal)
def hostname(self):
return LightDM.get_hostname()
@bridge.prop(bool, notify=property_changed)
def in_authentication(self):
return LightDMGreeter.get_in_authentication()
@bridge.prop(bool, notify=property_changed)
def is_authenticated(self):
return LightDMGreeter.get_is_authenticated()
@bridge.prop(Variant, notify=property_changed)
def language(self):
return language_to_dict(LightDM.get_language())
@bridge.prop(Variant, notify=noop_signal)
def languages(self):
return [language_to_dict(lang) for lang in LightDM.get_languages()]
@bridge.prop(Variant, notify=noop_signal)
def layout(self):
return layout_to_dict(LightDM.get_layout())
@bridge.prop(Variant, notify=noop_signal)
def layouts(self):
return [layout_to_dict(layout) for layout in LightDM.get_layouts()]
@bridge.prop(bool, notify=noop_signal)
def lock_hint(self):
return LightDMGreeter.get_lock_hint()
@bridge.prop(Variant, notify=property_changed)
def remote_sessions(self):
return [session_to_dict(session) for session in LightDM.get_remote_sessions()]
@bridge.prop(bool, notify=noop_signal)
def select_guest_hint(self):
return LightDMGreeter.get_select_guest_hint()
@bridge.prop(str, notify=noop_signal)
def select_user_hint(self):
return LightDMGreeter.get_select_user_hint() or ''
@bridge.prop(Variant, notify=noop_signal)
def sessions(self):
return [session_to_dict(session) for session in LightDM.get_sessions()]
@bridge.prop(str, notify=noop_signal)
def shared_data_directory(self):
return self._shared_data_directory
@bridge.prop(bool, notify=noop_signal)
def show_manual_login_hint(self):
return LightDMGreeter.get_show_manual_login_hint()
@bridge.prop(bool, notify=noop_signal)
def show_remote_login_hint(self):
return LightDMGreeter.get_show_remote_login_hint()
@bridge.prop(str, notify=noop_signal)
def themes_directory(self):
return self._themes_directory
@bridge.prop(Variant, notify=noop_signal)
def users(self):
return [user_to_dict(user) for user in LightDMUsers.get_users()]
@bridge.method(str)
def authenticate(self, username):
LightDMGreeter.authenticate(username)
self.property_changed.emit()
@bridge.method()
def authenticate_as_guest(self):
LightDMGreeter.authenticate_as_guest()
self.property_changed.emit()
@bridge.method()
def batteryUpdate(self):
return updateBattery(self)
@bridge.method(int)
def brightnessSet(self, quantity):
return changeBrightness(self, "-set", quantity)
@bridge.method(int)
def brightnessIncrease(self, quantity):
return changeBrightness(self, "-inc", quantity)
@bridge.method(int)
def brightnessDecrease(self, quantity):
return changeBrightness(self, "-dec", quantity)
@bridge.method()
def cancel_authentication(self):
LightDMGreeter.cancel_authentication()
self.property_changed.emit()
@bridge.method()
def cancel_autologin(self):
LightDMGreeter.cancel_autologin()
self.property_changed.emit()
@bridge.method(result=bool)
def hibernate(self):
return LightDM.hibernate()
@bridge.method(str)
def respond(self, response):
LightDMGreeter.respond(response)
self.property_changed.emit()
@bridge.method(result=bool)
def restart(self):
return LightDM.restart()
@bridge.method(str)
def set_language(self, lang):
if self.is_authenticated:
LightDMGreeter.set_language(lang)
self.property_changed.emit()
@bridge.method(result=bool)
def shutdown(self):
return LightDM.shutdown()
@bridge.method(str, result=bool)
def start_session(self, session):
if not session.strip():
return
return LightDMGreeter.start_session_sync(session)
@bridge.method(result=bool)
def suspend(self):
return LightDM.suspend()