+#!/usr/bin/env python3
+
+"""Wrapper functions for calling tplink.py"""
+
+import json
+import logging
+import os
+import re
+import subprocess
+import sys
+from typing import Dict, Optional
+
+import logging_utils
+from decorator_utils import timeout
+
+logger = logging.getLogger(__name__)
+
+
+@timeout(10.0, use_signals=False, error_message="Timed out waiting for tplink.py")
+def tplink_command(command: str) -> bool:
+ result = os.system(command)
+ signal = result & 0xFF
+ if signal != 0:
+ msg = f'{command} died with signal {signal}'
+ logger.warning(msg)
+ logging_utils.hlog(msg)
+ return False
+ else:
+ exit_value = result >> 8
+ if exit_value != 0:
+ msg = f'{command} failed, exited {exit_value}'
+ logger.warning(msg)
+ logging_utils.hlog(msg)
+ return False
+ logger.debug('%s succeeded.', command)
+ return True
+
+
+@timeout(10.0, use_signals=False, error_message="Timed out waiting for tplink.py")
+def tplink_get_info(cmd: str) -> Optional[Dict]:
+ logger.debug('Getting tplink device status via "%s"', cmd)
+ try:
+ out = subprocess.getoutput(cmd)
+ logger.debug('RAW OUT> %s', out)
+ out = re.sub("Sent:.*\n", "", out)
+ out = re.sub("Received: *", "", out)
+ info = json.loads(out)["system"]["get_sysinfo"]
+ logger.debug("%s", json.dumps(info, indent=4, sort_keys=True))
+ return info
+ except Exception as e:
+ logger.exception(e)
+ print(out, file=sys.stderr)
+ return None