import subprocess

from Xlib import display
from Xlib.ext import dpms

_display = display.Display()


def is_screen_on():
    _display.sync()
    return _display.dpms_info().state == 1  # untested


def go_lock_screen():
    _display.dpms_force_level(dpms.DPMSModeOff)
    _display.sync()


def go_wake_screen():
    _display.dpms_force_level(dpms.DPMSModeOn)
    _display.sync()


def go_sleep_mode():
    subprocess.run(['systemctl', 'suspend'])