Fixed caption being added explicitly

This commit is contained in:
Profitroll 2024-04-30 20:57:51 +02:00
parent c99f22b1b8
commit 818afa0b74
Signed by: profitroll
GPG Key ID: FA35CAB49DACD3B2
2 changed files with 17 additions and 3 deletions

View File

@ -139,7 +139,7 @@ class PyroClient(PyroClient):
await super().stop()
async def submit_media(
self, id: str
self, id: str, purge_caption: bool = False
) -> Tuple[Union[Message, None], Union[str, None]]:
db_entry = await col_submitted.find_one({"_id": ObjectId(id)})
submission = None
@ -224,7 +224,13 @@ class PyroClient(PyroClient):
)
raise SubmissionDuplicatesError(str(filepath), duplicates)
await col_submitted.update_one({"_id": ObjectId(id)}, {"$set": {"done": True}})
db_update = (
{"$set": {"done": True, "caption": None}}
if purge_caption
else {"$set": {"done": True}}
)
await col_submitted.update_one({"_id": ObjectId(id)}, db_update)
try:
if db_entry["temp"]["uuid"] is not None:

View File

@ -28,7 +28,10 @@ async def callback_query_yes(app: PyroClient, callback: CallbackQuery):
db_entry = await col_submitted.find_one({"_id": ObjectId(fullcallback[2])})
try:
submission = await app.submit_media(fullcallback[2])
submission = await app.submit_media(
fullcallback[2],
purge_caption=("caption" not in fullcallback),
)
except SubmissionUnavailableError:
await callback.answer(
text=app._("sub_msg_unavail", "callback", locale=user.locale),
@ -120,6 +123,11 @@ async def callback_query_yes(app: PyroClient, callback: CallbackQuery):
fullcallback[2],
submission[1],
)
logger.info(
"Submission with ID '%s' accepted and uploaded with ID '%s'",
fullcallback[2],
submission[1],
)
@Client.on_callback_query(filters.regex("sub_no_[\s\S]*"))