make Geldschieberbot a context manager

Previosuly there was the problem that the state was saved when the
bot object was destructed.
When there was an unhandled exception it was possible for the builtins
to be cleanedup before the bot's destructor was executed resulting
in yet another exception because the open builtin was no longer
available.

This is no longer possible when the bot is used as a context manager
python guarantees that the __exit__ code is executed before leaving
the with block.

This allows us prevent the state from beeing saved when there was an
exception to prevent invalid states.

1 files changed, 17 insertions, 18 deletions

diff --git a/single_shot.py b/single_shot.py
index 41d9849..0c85947 100755
--- a/single_shot.py
+++ b/single_shot.py
@@ -80,24 +80,23 @@ def main():
     if not group_id:
         die('A group id must be provided')
 
-    bot = Geldschieberbot(state_path, group_id, dry_run=args.dry_run)
-
-    # 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())
+    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__":