first commit
This commit is contained in:
commit
a859dcd3e8
GPG key ID:
3B7FBC22144E6398
1 changed files with 181 additions and 0 deletions
181
nitroplasm.py
Executable file
181
nitroplasm.py
Executable file
|
|
@ -0,0 +1,181 @@
|
|||
#!/usr/bin/python3
|
||||
|
||||
# nitroplasm: a lightweight desktop setter for KDE Plasma
|
||||
# by Derek Stevens <drkste@zoho.com>
|
||||
# written for use with the custom root menu by MatMoul:
|
||||
# https://github.com/MatMoul/plasma-containmentactions-customdesktopmenu
|
||||
# original script by pashazz:
|
||||
# https://github.com/pashazz/ksetwallpaper
|
||||
# license: GPLv3
|
||||
|
||||
from PyQt5 import QtGui, QtCore
|
||||
from PyQt5.QtWidgets import \
|
||||
QApplication, \
|
||||
QWidget, \
|
||||
QComboBox, \
|
||||
QVBoxLayout, \
|
||||
QHBoxLayout, \
|
||||
QPushButton, \
|
||||
QFileDialog , \
|
||||
QLabel, \
|
||||
QSizePolicy
|
||||
import sys
|
||||
from PyQt5.QtGui import QPixmap
|
||||
from Xlib import display
|
||||
import subprocess
|
||||
import dbus
|
||||
import argparse
|
||||
|
||||
class Window(QWidget):
|
||||
def __init__(self):
|
||||
super().__init__()
|
||||
|
||||
self.title = "Nitroplasm"
|
||||
self.top = 100
|
||||
self.left = 200
|
||||
self.width = 500
|
||||
self.height = 400
|
||||
|
||||
self.InitWindow()
|
||||
|
||||
def InitWindow(self):
|
||||
self.setWindowIcon(QtGui.QIcon(None))
|
||||
self.setWindowTitle(self.title)
|
||||
self.setGeometry(self.left, self.top, self.width, self.height)
|
||||
self.setFixedSize(self.width, self.height)
|
||||
self.setSizePolicy(QSizePolicy.Fixed, QSizePolicy.Fixed)
|
||||
self.setWindowFlags(QtCore.Qt.Dialog)
|
||||
|
||||
vbox = QVBoxLayout()
|
||||
hbox = QHBoxLayout()
|
||||
|
||||
self.image = None
|
||||
self.label = QLabel(self.image)
|
||||
self.label.setScaledContents(False)
|
||||
self.label.setAlignment(QtCore.Qt.AlignCenter)
|
||||
self.label.setSizePolicy(QSizePolicy.Ignored, QSizePolicy.Ignored)
|
||||
self.layout = 2
|
||||
self.monitor = -1
|
||||
|
||||
self.select_button = QPushButton("Select Image")
|
||||
self.select_button.clicked.connect(self.get_image)
|
||||
|
||||
layouts = {
|
||||
"Stretch": 0,
|
||||
"Scale": 1,
|
||||
"Zoom": 2,
|
||||
"Tile": 3,
|
||||
"Vertical Tile": 4,
|
||||
"Horizontal Tile": 5,
|
||||
"Center": 6
|
||||
}
|
||||
|
||||
self.layout_selector = QComboBox()
|
||||
for l in layouts:
|
||||
self.layout_selector.addItem(l, layouts[l])
|
||||
self.layout_selector.currentIndexChanged.connect(self.set_layout)
|
||||
self.layout_selector.setCurrentIndex(2)
|
||||
|
||||
self.monitor_selector = QComboBox()
|
||||
self.monitor_selector.addItem("All Monitors", -1)
|
||||
nDisplays = self.get_displays()
|
||||
print(nDisplays)
|
||||
if nDisplays > 1:
|
||||
for i in range(0, nDisplays):
|
||||
self.monitor_selector.addItem("Monitor" + str(i+1), i)
|
||||
self.monitor_selector.currentIndexChanged.connect(self.set_monitor)
|
||||
self.monitor_selector.setCurrentIndex(0)
|
||||
|
||||
self.apply_button = QPushButton("Apply Image")
|
||||
self.apply_button.setEnabled(False)
|
||||
self.apply_button.clicked.connect(self.apply_image)
|
||||
|
||||
vbox.addWidget(self.label)
|
||||
|
||||
hbox.addWidget(self.select_button)
|
||||
hbox.addWidget(self.layout_selector)
|
||||
hbox.addWidget(self.monitor_selector)
|
||||
hbox.addWidget(self.apply_button)
|
||||
|
||||
vbox.addLayout(hbox)
|
||||
self.setLayout(vbox)
|
||||
self.show()
|
||||
|
||||
def set_layout(self, index):
|
||||
self.layout = self.layout_selector.itemData(index)
|
||||
|
||||
def get_displays(self):
|
||||
if 'RANDR' in display.Display().list_extensions():
|
||||
try:
|
||||
output = [l for l in subprocess.check_output(["xrandr"]).decode("utf-8").splitlines()]
|
||||
return len([l.split()[0] for l in output if " connected " in l])
|
||||
except:
|
||||
return 1
|
||||
else:
|
||||
return 1
|
||||
|
||||
def set_monitor(self, index):
|
||||
self.monitor = self.monitor_selector.itemData(index)
|
||||
|
||||
def get_image(self):
|
||||
fname = QFileDialog.getOpenFileName(
|
||||
self,
|
||||
'Select Image File',
|
||||
'./',
|
||||
'Image files (*.png *.jpg *.gif *.jpeg)')
|
||||
|
||||
if fname[0]:
|
||||
self.image = fname[0]
|
||||
self.apply_button.setEnabled(True)
|
||||
|
||||
pixmap = QPixmap(self.image).scaled(
|
||||
self.label.width(),
|
||||
self.label.height(),
|
||||
QtCore.Qt.KeepAspectRatio,
|
||||
QtCore.Qt.SmoothTransformation)
|
||||
|
||||
#self.label.setMaximumSize(self.label.width(), self.label.height())
|
||||
self.label.setPixmap(pixmap)
|
||||
|
||||
def set_wallpaper(self, filepath, plugin = 'org.kde.image', layout = 2, desk = -1):
|
||||
if desk < 0:
|
||||
jscript = """
|
||||
var allDesktops = desktops();
|
||||
for (i=0;i<allDesktops.length;i++) {
|
||||
d = allDesktops[i];
|
||||
d.wallpaperPlugin = "%s";
|
||||
d.currentConfigGroup = Array("Wallpaper", "%s", "General");
|
||||
d.writeConfig("Image", "file://%s")
|
||||
d.writeConfig("FillMode", "%s")
|
||||
}
|
||||
"""
|
||||
else:
|
||||
jscript = """
|
||||
var d = desktops()[%s];
|
||||
d.wallpaperPlugin = "%s";
|
||||
d.currentConfigGroup = Array("Wallpaper", "%s", "General");
|
||||
d.writeConfig("Image", "file://%s")
|
||||
d.writeConfig("FillMode", "%s")
|
||||
"""
|
||||
|
||||
bus = dbus.SessionBus()
|
||||
plasma = dbus.Interface(
|
||||
bus.get_object(
|
||||
'org.kde.plasmashell',
|
||||
'/PlasmaShell'),
|
||||
dbus_interface='org.kde.PlasmaShell')
|
||||
if desk < 0:
|
||||
plasma.evaluateScript(jscript % (plugin, plugin, filepath, layout))
|
||||
else:
|
||||
plasma.evaluateScript(jscript % (desk, plugin, plugin, filepath, layout))
|
||||
|
||||
def apply_image(self):
|
||||
self.set_wallpaper(self.image, 'org.kde.image', self.layout, self.monitor)
|
||||
|
||||
if __name__ == '__main__':
|
||||
if len(sys.argv) == 1:
|
||||
App = QApplication(sys.argv)
|
||||
window = Window()
|
||||
sys.exit(App.exec())
|
||||
else:
|
||||
parser = argparse.ArgumentParser(description='KDE Wallpaper setter')
|
||||
Loading…
Reference in a new issue