Add /finish
This commit is contained in:
parent
111b101714
commit
00e24e3bd9
21
bot.py
21
bot.py
|
@ -3,6 +3,7 @@
|
||||||
import os
|
import os
|
||||||
import os.path as pt
|
import os.path as pt
|
||||||
import shutil
|
import shutil
|
||||||
|
import subprocess
|
||||||
|
|
||||||
from telegram.ext import Updater, CommandHandler, MessageHandler, filters
|
from telegram.ext import Updater, CommandHandler, MessageHandler, filters
|
||||||
import config
|
import config
|
||||||
|
@ -33,7 +34,23 @@ def on_img_received(upd, ctx):
|
||||||
|
|
||||||
|
|
||||||
def on_finish(upd, ctx):
|
def on_finish(upd, ctx):
|
||||||
pass
|
user_id = upd.effective_chat.id
|
||||||
|
path = f"data/{user_id}"
|
||||||
|
if not pt.isdir(path):
|
||||||
|
ctx.bot.send_message(upd.effective_chat.id, "No Images Available")
|
||||||
|
return
|
||||||
|
|
||||||
|
ctx.bot.send_message(upd.effective_chat.id, "Started Merging")
|
||||||
|
files = list(sorted(
|
||||||
|
os.listdir(path),
|
||||||
|
key=lambda file: int(file)
|
||||||
|
))
|
||||||
|
pdf_path = f"{path}/final.pdf"
|
||||||
|
subprocess.run(["convert", *[path + "/" + file for file in files], pdf_path])
|
||||||
|
with open(pdf_path, "rb") as f:
|
||||||
|
ctx.bot.send_document(upd.effective_chat.id, f)
|
||||||
|
shutil.rmtree(path)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def on_cancel(upd, ctx):
|
def on_cancel(upd, ctx):
|
||||||
|
@ -41,7 +58,7 @@ def on_cancel(upd, ctx):
|
||||||
path = f"data/{user_id}"
|
path = f"data/{user_id}"
|
||||||
if pt.isdir(path):
|
if pt.isdir(path):
|
||||||
shutil.rmtree(f"data/{user_id}")
|
shutil.rmtree(f"data/{user_id}")
|
||||||
ctx.bot.send_message(upd.effective_chat.id, f"Cleared images")
|
ctx.bot.send_message(upd.effective_chat.id, "Cleared images")
|
||||||
|
|
||||||
|
|
||||||
updater = Updater(token=config.API_KEY, use_context=True)
|
updater = Updater(token=config.API_KEY, use_context=True)
|
||||||
|
|
Loading…
Reference in New Issue