67 lines
1.8 KiB
Python
67 lines
1.8 KiB
Python
import os, logging
|
|
DEBUG = os.environ.get('DEBUG')
|
|
logging.basicConfig(
|
|
format='[%(asctime)s] %(levelname)s %(module)s/%(funcName)s - %(message)s',
|
|
level=logging.DEBUG if DEBUG else logging.INFO)
|
|
|
|
import subprocess
|
|
import socket
|
|
import asyncio
|
|
from aiomqtt import Client, TLSParameters
|
|
|
|
tls_params = TLSParameters(
|
|
ca_certs='/etc/ssl/certs/ISRG_Root_X1.pem',
|
|
)
|
|
|
|
HOSTNAME = socket.gethostname()
|
|
|
|
def lock_screen():
|
|
os.environ['DISPLAY'] = ':0'
|
|
subprocess.run(['loginctl', 'lock-session', '0'])
|
|
subprocess.run(['loginctl', 'lock-session', '1'])
|
|
subprocess.run(['loginctl', 'lock-session', '2'])
|
|
subprocess.run(['loginctl', 'lock-session', '3'])
|
|
|
|
async def process_lock_command(text):
|
|
target = text
|
|
|
|
logging.info('Lock target: %s', text)
|
|
|
|
if target == HOSTNAME or target == 'all':
|
|
logging.info('Locking screen...')
|
|
lock_screen()
|
|
|
|
async def process_mqtt(message):
|
|
text = message.payload.decode()
|
|
topic = message.topic.value
|
|
logging.debug('MQTT topic: %s, message: %s', topic, text)
|
|
|
|
if topic.startswith('lock'):
|
|
await process_lock_command(text)
|
|
else:
|
|
logging.debug('Invalid topic, returning')
|
|
return
|
|
|
|
async def fetch_mqtt():
|
|
await asyncio.sleep(3)
|
|
|
|
async with Client(
|
|
hostname='mqtt-dev.dns.t0.vc',
|
|
port=8081,
|
|
#username='reader',
|
|
#password=secrets.MQTT_READER_PASSWORD,
|
|
tls_params=tls_params,
|
|
) as client:
|
|
await client.subscribe('#')
|
|
async for message in client.messages:
|
|
loop = asyncio.get_event_loop()
|
|
loop.create_task(process_mqtt(message))
|
|
|
|
if __name__ == '__main__':
|
|
logging.info('')
|
|
logging.info('==========================')
|
|
logging.info('Booting up...')
|
|
|
|
loop = asyncio.get_event_loop()
|
|
loop.run_until_complete(fetch_mqtt())
|