shaba/openuds
1
0
Fork 1
mirror of https://github.com/dkmstr/openuds.git synced 2025-01-20 14:03:49 +03:00
Code Issues Releases Wiki Activity

Added screenshot capability

Browse Source
This commit is contained in:
Adolfo Gómez García 2019-12-10 20:20:43 +01:00
parent ff5601ce12
commit 0895d2feaf
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
actor/src/udsactor/client.py
View File

@ -34,6 +34,7 @@ import typing
import signal
from PyQt5.QtWidgets import QApplication, QMessageBox
from PyQt5.QtCore import QByteArray, QBuffer, QIODevice
from . import rest
from . import tools
@ -46,6 +47,7 @@ from .http import client
# Not imported at runtime, just for type checking
if typing.TYPE_CHECKING:
from . import types
from PyQt5.QtGui import QPixmap
class UDSActorClient(threading.Thread):
_running: bool
@ -111,7 +113,12 @@ class UDSActorClient(threading.Thread):
return 'ok'
def screenshot(self) -> typing.Any:
pass
pixmap: QPixmap = self._qApp.primaryScreen().grabWindow(0)
ba = QByteArray()
buffer = QBuffer(ba)
buffer.open(QIODevice.WriteOnly)
pixmap.save(buffer)
return bytes(ba.toBase64()).decode() # 'result' of JSON will contain base64 of screen
def script(self, script: str) -> typing.Any:
tools.ScriptExecutorThread(script).start()