python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2024-12-26 09:56:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Check session type using environment variable instead of loginctl

Browse Source
This commit is contained in:
TheNooB 2023-08-12 12:51:09 +08:00
parent 7b157b045a
commit 0b6ab79145
No known key found for this signature in database
GPG Key ID: B80F1826EB256C6E
1 changed files with 4 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/PIL/ImageGrab.py
View File

@ -140,27 +140,10 @@ def grabclipboard():
return BmpImagePlugin.DibImageFile(data)
return None
else:
if shutil.which("loginctl"):
try:
loginctl = subprocess.check_output("loginctl").decode().split("\n")
except subprocess.CalledProcessError:
loginctl = None
else:
loginctl = None
if loginctl is not None:
username = os.getlogin()
sessionid = [
line.split()[0] for line in loginctl if username in line.split()
][0]
sessiontype = (
subprocess.check_output(
["loginctl", "show-session", sessionid, "-p", "Type"]
)
.decode()
.strip("\n")
.split("=")[1]
)
if os.getenv("WAYLAND_DISPLAY"):
sessiontype = "wayland"
elif os.getenv("DISPLAY"):
sessiontype = "x11"
else: # Session type check failed
sessiontype = None