dron 0.1.20241008__py3-none-any.whl

dron/notify/common.py

@@ -0,0 +1,53 @@
+import argparse
+import platform
+import shlex
+import sys
+from subprocess import PIPE, STDOUT, Popen, check_output
+from typing import Iterator
+
+IS_SYSTEMD = platform.system() != 'Darwin'  # if not systemd it's launchd
+
+
+def get_parser() -> argparse.ArgumentParser:
+    p = argparse.ArgumentParser()
+    p.add_argument('--job', required=True)
+    p.add_argument('--stdin', action='store_true')
+    return p
+
+
+def get_stdin() -> Iterator[bytes]:
+    yield from sys.stdin.buffer
+
+
+def get_last_systemd_log(job: str) -> Iterator[bytes]:
+    # output unit status
+    cmd = ['systemctl', '--user', 'status', '--no-pager', job, '-o', 'cat']
+    yield b'$ ' + ' '.join(map(shlex.quote, cmd)).encode('utf8') + b'\n\n'
+    with Popen(cmd, stdout=PIPE, stderr=STDOUT) as po:
+        out = po.stdout
+        assert out is not None
+        yield from out
+    rc = po.poll()
+    assert rc in {
+        0,
+        3,  # 3 means failure due to job exit code
+    }, rc
+
+    # for logs, we used to use --lines 1000000 in systemctl status
+    # however, from around 2024 it stated consuming too much time
+    # (as if it actually retrieved 1000000 lines and only then tooks the ones relevant to the unit??)
+
+    cmd = ['systemctl', '--user', 'show', job, '-p', 'InvocationID', '--value']
+    invocation_id = check_output(cmd, text=True)
+    invocation_id = invocation_id.strip()  # for some reason dumps multiple lines?
+    assert len(invocation_id) > 0  # just in case, todo maybe make defensive?
+
+    yield b'\n'
+    cmd = ['journalctl', '--no-pager', f'_SYSTEMD_INVOCATION_ID={invocation_id}']
+    yield b'$ ' + ' '.join(map(shlex.quote, cmd)).encode('utf8') + b'\n\n'
+    with Popen(cmd, stdout=PIPE, stderr=STDOUT) as po:
+        out = po.stdout
+        assert out is not None
+        yield from out
+    rc = po.poll()
+    assert rc == 0

dron/notify/email.py

@@ -0,0 +1,43 @@
+import socket
+from subprocess import PIPE, Popen
+from typing import Iterator
+
+from .common import get_last_systemd_log, get_parser, get_stdin
+
+
+def send_payload(payload: Iterator[bytes]) -> None:
+    with Popen(['sendmail', '-t'], stdin=PIPE) as po:
+        stdin = po.stdin
+        assert stdin is not None
+        for line in payload:
+            stdin.write(line)
+        stdin.flush()
+    rc = po.poll()
+    assert rc == 0, rc
+
+
+def send_email(*, to: str, job: str, stdin: bool) -> None:
+    def payload() -> Iterator[bytes]:
+        hostname = socket.gethostname()
+        yield f'''
+To: {to}
+From: dron <root@{hostname}>
+Subject: {job}
+Content-Transfer-Encoding: 8bit
+Content-Type: text/plain; charset=UTF-8
+'''.lstrip().encode('utf8')
+        last_log = get_stdin() if stdin else get_last_systemd_log(job)
+        yield from last_log
+
+    send_payload(payload())
+
+
+def main() -> None:
+    p = get_parser()
+    p.add_argument('--to', required=True)
+    args = p.parse_args()
+    send_email(to=args.to, job=args.job, stdin=args.stdin)
+
+
+if __name__ == '__main__':
+    main()

dron/notify/ntfy_common.py

@@ -0,0 +1,24 @@
+"""
+uses https://github.com/dschep/ntfy
+"""
+
+import logging
+import socket
+import subprocess
+import sys
+from typing import NoReturn
+
+
+def run_ntfy(*, job: str, backend: str) -> NoReturn:
+    # TODO not sure what to do with --stdin arg here?
+    # could probably use last N lines of log or something
+    # TODO get last logs here?
+    title = f'dron[{socket.gethostname()}]: {job} failed'
+    body = title
+    try:
+        subprocess.check_call(['ntfy', '-b', backend, '-t', title, 'send', body])
+    except Exception as e:
+        logging.exception(e)
+        # TODO fallback on email?
+        sys.exit(1)
+    sys.exit(0)

dron/notify/ntfy_desktop.py

@@ -0,0 +1,15 @@
+from .common import IS_SYSTEMD, get_parser
+from .ntfy_common import run_ntfy
+
+BACKEND = 'linux' if IS_SYSTEMD else 'darwin'
+
+
+def main() -> None:
+    p = get_parser()
+    args = p.parse_args()
+
+    run_ntfy(job=args.job, backend=BACKEND)
+
+
+if __name__ == '__main__':
+    main()

dron/notify/ntfy_telegram.py

@@ -0,0 +1,12 @@
+from .common import get_parser
+from .ntfy_common import run_ntfy
+
+
+def main() -> None:
+    p = get_parser()
+    args = p.parse_args()
+    run_ntfy(job=args.job, backend='telegram')
+
+
+if __name__ == '__main__':
+    main()

dron/notify/telegram.py

@@ -0,0 +1,42 @@
+"""
+uses telegram-send for Telegram notifications
+make sure to run "telegram-send --configure" beforehand!
+"""
+
+import asyncio
+import logging
+import socket
+import sys
+
+from .common import get_last_systemd_log, get_parser, get_stdin
+
+
+def send(*, message: str) -> None:
+    import telegram_send  # type: ignore[import-untyped]
+
+    asyncio.run(telegram_send.send(messages=[message]))
+
+
+def main() -> None:
+    p = get_parser()
+    args = p.parse_args()
+
+    job: str = args.job
+    stdin: bool = args.stdin
+
+    body = f'dron[{socket.gethostname()}]: {job} failed'
+
+    last_log = get_stdin() if stdin else get_last_systemd_log(job)
+    body += '\n' + '\n'.join(l.decode('utf8') for l in last_log)
+
+    try:
+        send(message=body)
+    except Exception as e:
+        logging.exception(e)
+        # TODO fallback on email?
+        sys.exit(1)
+    sys.exit(0)
+
+
+if __name__ == '__main__':
+    main()