CreatBot/CreatBotKlipperScreen
6
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

functions: improve error handling while loading DPMS

Browse Source
This commit is contained in:
Alfredo Monclus 2024-06-07 15:36:13 -03:00
parent e8cc5cd591
commit ccd8fd9fcc
1 changed files with 22 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
ks_includes/functions.py
View File

@ -27,28 +27,34 @@ try:
def get_DPMS_state(display_name_in_byte_string=b':0'): def get_DPMS_state(display_name_in_byte_string=b':0'):
state = DPMS_State.Fail state = DPMS_State.Fail
if not isinstance(display_name_in_byte_string, bytes): if not isinstance(display_name_in_byte_string, bytes):
raise TypeError raise TypeError("display_name_in_byte_string must be of type bytes")
display_name = ctypes.c_char_p()
display_name.value = display_name_in_byte_string display_name = ctypes.c_char_p(display_name_in_byte_string)
libXext.XOpenDisplay.restype = ctypes.c_void_p libXext.XOpenDisplay.restype = ctypes.c_void_p
display = ctypes.c_void_p(libXext.XOpenDisplay(display_name)) display = ctypes.c_void_p(libXext.XOpenDisplay(display_name))
dummy1_i_p = ctypes.create_string_buffer(8)
dummy2_i_p = ctypes.create_string_buffer(8) major_opcode_p = ctypes.create_string_buffer(8)
first_event_p = ctypes.create_string_buffer(8)
if display.value: if display.value:
if libXext.DPMSQueryExtension(display, dummy1_i_p, dummy2_i_p) \ try:
and libXext.DPMSCapable(display): if libXext.DPMSQueryExtension(display, major_opcode_p, first_event_p) \
onoff_p = ctypes.create_string_buffer(1) and libXext.DPMSCapable(display):
state_p = ctypes.create_string_buffer(2) onoff_p = ctypes.create_string_buffer(1)
if libXext.DPMSInfo(display, state_p, onoff_p): state_p = ctypes.create_string_buffer(2)
onoff = struct.unpack('B', onoff_p.raw)[0] if libXext.DPMSInfo(display, state_p, onoff_p):
if onoff: onoff = struct.unpack('B', onoff_p.raw)[0]
state = struct.unpack('H', state_p.raw)[0] if onoff:
libXext.XCloseDisplay(display) state = struct.unpack('H', state_p.raw)[0]
finally:
libXext.XCloseDisplay(display)
return state return state
dpms_loaded = True dpms_loaded = True
except Exception as msg: except OSError as e:
logging.error(f"Couldn't load DPMS: {msg}") logging.error(f"Couldn't load DPMS library: {e}")
except Exception as e:
logging.error(f"An unexpected error occurred: {e}")
def get_software_version(): def get_software_version():