confirm before delete_mine

rsbbs/bbs.py

@@ -165,18 +165,23 @@ class BBS():
 
     def delete_mine(self, args):
         '''Delete all messages addressed to user'''
-        with Session(self.engine) as session:
+        self.write_output("Delete all messages addressed to you? Y/N:")
-            try:
+        response = sys.stdin.readline().strip()
-                statement = delete(Message).where(Message.recipient == self.calling_station).returning(Message)
+        if response.lower() != "y":
-                results = session.execute(statement)
+            return
-                count = len(results.all())
+        else:
-                if count > 0:
+            with Session(self.engine) as session:
-                    self.write_output(f"Deleted {count} messages")
+                try:
-                    session.commit()
+                    statement = delete(Message).where(Message.recipient == self.calling_station).returning(Message)
-                else:
+                    results = session.execute(statement)
-                    self.write_output(f"No messages to delete.")
+                    count = len(results.all())
-            except Exception as e:
+                    if count > 0:
-                self.write_output(f"Unable to delete messages: {e}")
+                        self.write_output(f"Deleted {count} messages")
+                        session.commit()
+                    else:
+                        self.write_output(f"No messages to delete.")
+                except Exception as e:
+                    self.write_output(f"Unable to delete messages: {e}")
 
     def heard(self, args):
         '''Show heard stations log'''