69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
"""System mode daemon."""
|
|
|
|
from argparse import ArgumentParser, Namespace
|
|
from logging import DEBUG, INFO, basicConfig, getLogger
|
|
from subprocess import Popen
|
|
from typing import Iterable
|
|
|
|
from tabletmode.config import load_config
|
|
|
|
|
|
DESCRIPTION = 'Setup system for laptop or tablet mode.'
|
|
EVTEST = '/usr/bin/evtest'
|
|
LOG_FORMAT = '[%(levelname)s] %(name)s: %(message)s'
|
|
LOGGER = getLogger('sysmoded')
|
|
|
|
|
|
def get_args() -> Namespace:
|
|
"""Parses the CLI arguments."""
|
|
|
|
parser = ArgumentParser(description=DESCRIPTION)
|
|
parser.add_argument(
|
|
'-v', '--verbose', action='store_true',
|
|
help='turn on verbose logging')
|
|
subparsers = parser.add_subparsers(dest='mode')
|
|
subparsers.add_parser('laptop', help='enable laptop mode')
|
|
subparsers.add_parser('tablet', help='enable tablet mode')
|
|
return parser.parse_args()
|
|
|
|
|
|
def disable_device(device: str) -> Popen:
|
|
"""Disables the respective device via evtest."""
|
|
|
|
return Popen((EVTEST, '--grab', device))
|
|
|
|
|
|
def disable_devices(devices: Iterable[str]) -> None:
|
|
"""Disables the given devices."""
|
|
|
|
subprocesses = []
|
|
|
|
for device in devices:
|
|
subprocess = disable_device(device)
|
|
subprocesses.append(subprocess)
|
|
|
|
for subprocess in subprocesses:
|
|
subprocess.wait()
|
|
|
|
|
|
def get_devices(mode: str) -> Iterable[str]:
|
|
"""Reads the device from the config file."""
|
|
|
|
config = load_config()
|
|
devices = config.get(mode) or ()
|
|
|
|
if not devices:
|
|
LOGGER.info('No devices configured to disable.')
|
|
|
|
return devices
|
|
|
|
|
|
def main():
|
|
"""Runs the main program."""
|
|
|
|
arguments = get_args()
|
|
level = DEBUG if arguments.verbose else INFO
|
|
basicConfig(level=level, format=LOG_FORMAT)
|
|
devices = get_devices(arguments.mode)
|
|
disable_devices(devices)
|