aboutsummaryrefslogtreecommitdiff
path: root/single_shot.py
diff options
context:
space:
mode:
Diffstat (limited to 'single_shot.py')
-rwxr-xr-xsingle_shot.py103
1 files changed, 0 insertions, 103 deletions
diff --git a/single_shot.py b/single_shot.py
deleted file mode 100755
index 0c85947..0000000
--- a/single_shot.py
+++ /dev/null
@@ -1,103 +0,0 @@
-#!/usr/bin/env python3
-# Copyright (c) 2023 Florian Fischer. All rights reserved.
-#
-# This file is part of geldschieberbot.
-#
-# geldschieberbot is free software: you can redistribute it and/or modify it
-# under the terms of the GNU General Public License as published by the Free
-# Software Foundation, either version 3 of the License, or (at your option)
-# any later version.
-#
-# geldschieberbot is distributed in the hope that it will be useful, but WITHOUT
-# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
-# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
-#
-# You should have received a copy of the GNU General Public License along with
-# geldschieberbot found in the LICENSE file. If not,
-# see <https://www.gnu.org/licenses/>.
-"""Run geldschieberbot on all received messages, send replies, and exit"""
-
-import argparse
-from datetime import datetime
-import json
-import os
-import subprocess
-import sys
-import typing as T
-
-from geldschieberbot import Geldschieberbot
-from models import Reply, Quote
-
-
-def send(_cmd, msgs: list[Reply], quote: T.Optional[Quote] = None):
- """Send a message with optional attachment"""
- for msg in msgs:
- cmd = _cmd
- if msg.attachment:
- cmd += f' -a {msg.attachment}'
-
- if quote:
- cmd += f' --quote-timestamp={quote.timestamp} --quote-author={quote.author}'
-
- subprocess.run(cmd.split(' '), input=msg.msg.encode(), check=False)
-
-
-def die(msg: str, status=1):
- """Exit because an error ocurred"""
- print(msg, file=sys.stderr)
- sys.exit(status)
-
-
-def main():
- """Read messages from stdin and send the bot's replies back"""
- parser = argparse.ArgumentParser()
- parser.add_argument('-d',
- '--dry-run',
- help='do not persist changes',
- action='store_true')
- parser.add_argument('-f', '--state-file', help='the state file')
- parser.add_argument('-g', '--group-id', help='the group id to listen to')
- parser.add_argument('--send-cmd',
- help='the shell command used to send messages')
- parser.add_argument('-nq',
- '--no-quote',
- help='not quote the message causing the reply',
- action='store_true')
- args = parser.parse_args()
-
- if args.dry_run:
- print("Dry Run no changes will apply!")
-
- state_path = args.state_file or os.environ['GSB_STATE_FILE']
- if not state_path:
- die('A state path must be provided')
-
- send_cmd = args.send_cmd or os.environ['GSB_SEND_CMD']
- if not send_cmd:
- die('A send command must be provided')
-
- group_id = args.group_id or os.environ['GSB_GROUP_ID']
- if not group_id:
- die('A group id must be provided')
-
- with Geldschieberbot(state_path, group_id, dry_run=args.dry_run) as bot:
- # Read cmds from stdin
- for line in sys.stdin.read().splitlines():
- try:
- envelope = json.loads(line)['envelope']
- except json.JSONDecodeError:
- print(datetime.now(), line, "not valid json")
- continue
-
- quote = None
- if not args.no_quote:
- quote = Quote(timestamp=envelope['timestamp'],
- author=envelope['source'])
-
- send(send_cmd, bot.handle(envelope), quote)
-
- send(send_cmd, bot.run_scheduled_cmds())
-
-
-if __name__ == "__main__":
- main()