67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
from typing import TypeVar
|
|
|
|
import platform
|
|
import subprocess
|
|
import ctypes
|
|
import os
|
|
|
|
import logging
|
|
logger = logging.getLogger("desktopenv.getters.misc")
|
|
|
|
R = TypeVar("Rule")
|
|
def get_rule(env, config: R) -> R:
|
|
"""
|
|
Returns the rule as-is.
|
|
"""
|
|
return config["rules"]
|
|
|
|
|
|
def get_desktop_path():
|
|
username = os.getlogin() # Get the current username
|
|
if platform.system() == "Windows":
|
|
return os.path.join("C:", "Users", username, "Desktop")
|
|
elif platform.system() == "Darwin": # macOS is identified as 'Darwin'
|
|
return os.path.join("/Users", username, "Desktop")
|
|
elif platform.system() == "Linux":
|
|
return os.path.join("/home", username, "Desktop")
|
|
else:
|
|
raise Exception("Unsupported operating system")
|
|
|
|
|
|
def get_wallpaper():
|
|
def get_wallpaper_windows():
|
|
SPI_GETDESKWALLPAPER = 0x73
|
|
MAX_PATH = 260
|
|
buffer = ctypes.create_unicode_buffer(MAX_PATH)
|
|
ctypes.windll.user32.SystemParametersInfoW(SPI_GETDESKWALLPAPER, MAX_PATH, buffer, 0)
|
|
return buffer.value
|
|
|
|
def get_wallpaper_macos():
|
|
script = """
|
|
tell application "System Events" to tell every desktop to get picture
|
|
"""
|
|
process = subprocess.Popen(['osascript', '-e', script], stdout=subprocess.PIPE)
|
|
output, error = process.communicate()
|
|
if error:
|
|
logger.error("Error: %s", error)
|
|
else:
|
|
return output.strip().decode('utf-8')
|
|
|
|
def get_wallpaper_linux():
|
|
try:
|
|
output = subprocess.check_output(["gsettings", "get", "org.gnome.desktop.background", "picture-uri"])
|
|
return output.decode('utf-8').strip().replace('file://', '').replace("'", "")
|
|
except Exception as e:
|
|
logger.error("Error: %s", e)
|
|
return None
|
|
|
|
os_name = platform.system()
|
|
if os_name == 'Windows':
|
|
return get_wallpaper_windows()
|
|
elif os_name == 'Darwin':
|
|
return get_wallpaper_macos()
|
|
elif os_name == 'Linux':
|
|
return get_wallpaper_linux()
|
|
else:
|
|
return "Unsupported OS"
|