From 7bd114ac591033e812ea232aac6de1d45f05521c Mon Sep 17 00:00:00 2001
From: Jason Robinson <mail@jasonrobinson.me>
Date: Fri, 25 Dec 2020 16:53:28 +0200
Subject: [PATCH] Make handle_send call a bit safer

---
 federation/outbound.py | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/federation/outbound.py b/federation/outbound.py
index e653562..f76f309 100644
--- a/federation/outbound.py
+++ b/federation/outbound.py
@@ -142,21 +142,18 @@ def handle_send(
             "headers": {},
             "payload": None,
             "urls": set(),
-            "method": None,
         },
         "diaspora": {
             "auth": None,
             "headers": {},
             "payload": None,
             "urls": set(),
-            "method": None,
         },
         "matrix": {
             "auth": None,
             "headers": {},
             "payload": None,
             "urls": set(),
-            "method": None,
         },
     }
     skip_ready_payload = {
@@ -356,9 +353,9 @@ def handle_send(
                 send_document(
                     url,
                     payload["payload"],
-                    auth=payload["auth"],
-                    headers=payload["headers"],
-                    method=payload["method"],
+                    auth=payload.get("auth"),
+                    headers=payload.get("headers"),
+                    method=payload.get("method"),
                 )
             except Exception as ex:
                 logger.error("handle_send - failed to send payload to %s: %s, payload: %s", url, ex, payload["payload"])