Update batch.py

Anihr · May 28, 2023 · 17a05b9 · 17a05b9
1 parent 5f2af1c
commit 17a05b9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/main/plugins/batch.py b/main/plugins/batch.py
@@ -26,6 +26,13 @@
 
 batch = []
 
+@Drone.on(events.NewMessage(incoming=True, pattern='/cancel'))
+async def cancel(event):
+    if not event.sender_id in batch:
+        return await event.reply("No batch active.")
+    batch.clear()
+    await event.reply("Done.")
+
 @Drone.on(events.NewMessage(incoming=True, from_users=AUTH, pattern='/batch'))
 async def _batch(event):
     if not event.is_private: