profitroll/TelegramPoster
profitroll/TelegramPoster
Archived
1
0
Fork 0
Code Issues Pull Requests Releases 3 Wiki Activity

Fixed caption being added explicitly

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
classes/pyroclient.py
View File

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

10
plugins/callbacks/submission.py
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])}) db_entry = await col_submitted.find_one({"_id": ObjectId(fullcallback[2])})
try: try:
submission = await app.submit_media(fullcallback[2]) submission = await app.submit_media(
fullcallback[2],
purge_caption=("caption" not in fullcallback),
)
except SubmissionUnavailableError: except SubmissionUnavailableError:
await callback.answer( await callback.answer(
text=app._("sub_msg_unavail", "callback", locale=user.locale), text=app._("sub_msg_unavail", "callback", locale=user.locale),
@ -120,6 +123,11 @@ async def callback_query_yes(app: PyroClient, callback: CallbackQuery):
fullcallback[2], fullcallback[2],
submission[1], 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]*")) @Client.on_callback_query(filters.regex("sub_no_[\s\S]*"))