flibusta-apps/book_bot
1
0
Fork 0
mirror of https://github.com/flibusta-apps/book_bot.git synced 2025-12-06 15:35:35 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix

Browse Source
This commit is contained in:
Bulat Kurbanov
2023-06-07 22:07:35 +02:00
parent e6d302c25a
commit 50c3cf0718
1 changed files with 33 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
src/bots/approved_bot/modules/download.rs
View File

@@ -464,13 +464,32 @@ async fn download_archive(
.send() .send()
.await?; .await?;
let mut i = 15 * 60 / 5; tokio::spawn(async move {
let mut i = 15 * 60 / 5;
while task.status != TaskStatus::Complete && i >= 0 { while task.status != TaskStatus::Complete && i >= 0 {
task = match get_task(task.id).await { task = match get_task(task.id).await {
Ok(v) => v, Ok(v) => v,
Err(err) => { Err(err) => {
bot bot
.edit_message_text(message.chat.id, message.id, "Ошибка! Попробуйте позже :(")
.reply_markup(InlineKeyboardMarkup {
inline_keyboard: vec![],
})
.send()
.await?;
return Err(err);
},
};
sleep(Duration::from_secs(5)).await;
i = i - 1;
}
if task.status != TaskStatus::Complete {
bot
.edit_message_text(message.chat.id, message.id, "Ошибка! Попробуйте позже :(") .edit_message_text(message.chat.id, message.id, "Ошибка! Попробуйте позже :(")
.reply_markup(InlineKeyboardMarkup { .reply_markup(InlineKeyboardMarkup {
inline_keyboard: vec![], inline_keyboard: vec![],
@@ -478,34 +497,19 @@ async fn download_archive(
.send() .send()
.await?; .await?;
return Err(err); return Ok(());
}, }
};
sleep(Duration::from_secs(5)).await;
i = i - 1;
}
if task.status != TaskStatus::Complete {
bot bot
.edit_message_text(message.chat.id, message.id, "Ошибка! Попробуйте позже :(") .send_document(
.reply_markup(InlineKeyboardMarkup { message.chat.id,
inline_keyboard: vec![], InputFile::url(task.result_link.unwrap().parse().unwrap())
}) )
.send() .send()
.await?; .await?;
return Ok(()); Ok(())
} });
bot
.send_document(
message.chat.id,
InputFile::url(task.result_link.unwrap().parse().unwrap())
)
.send()
.await?;
Ok(()) Ok(())
} }