feat: print response_dict when name not found

hlf20010508 · Jul 8, 2024 · 96cbb85 · 96cbb85
1 parent ae933c6
commit 96cbb85
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/modules/handlers/transfer.py b/modules/handlers/transfer.py
@@ -62,6 +62,8 @@ async def transfer_handler(event):
                 await status_message.finish(
                     path=os.path.join(last_remote_root_path, name), size=event.file.size
                 )
+        except KeyError as e:
+            await event.reply(logger("Error: %s\nresponse: %s" % (e, response_dict)))
         except Exception as e:
             await event.reply("Error: %s" % logger(e))
 
@@ -118,6 +120,10 @@ async def transfer_handler(event):
                             path=os.path.join(last_remote_root_path, name),
                             size=message.file.size,
                         )
+                except KeyError as e:
+                    await event.reply(
+                        logger("Error: %s\nresponse: %s" % (e, response_dict))
+                    )
                 except Exception as e:
                     await event.reply("Error: %s" % logger(e))
             else:

diff --git a/modules/handlers/url.py b/modules/handlers/url.py
@@ -70,6 +70,8 @@ async def url_handler(event):
             path=os.path.join(last_remote_root_path, response_dict["name"]),
             size=total_length,
         )
+    except KeyError as e:
+        await event.reply(logger("Error: %s\nresponse: %s" % (e, response_dict)))
     except Exception as e:
         logger(e)
         await status_message.report_error(e)