From 5066b732ab2512bacbc6fadc0de51b7465929ec6 Mon Sep 17 00:00:00 2001
From: jMyles <justin@justinholmes.com>
Date: Tue, 5 Feb 2019 16:51:12 -0700
Subject: [PATCH] Simplifying get_nodes_via_rest - it doesn't need to look at
 which client its using anymore.

---
 nucypher/network/middleware.py | 25 +++++++++----------------
 nucypher/network/nodes.py      |  5 +----
 2 files changed, 10 insertions(+), 20 deletions(-)

diff --git a/nucypher/network/middleware.py b/nucypher/network/middleware.py
index d0218b167..9ea805167 100644
--- a/nucypher/network/middleware.py
+++ b/nucypher/network/middleware.py
@@ -171,11 +171,9 @@ class RestMiddleware:
         return response.content
 
     def get_nodes_via_rest(self,
-                           url,
-                           certificate_filepath,
+                           node,
                            announce_nodes=None,
                            nodes_i_need=None,
-                           client=requests,
                            fleet_checksum=None):
         if nodes_i_need:
             # TODO: This needs to actually do something.
@@ -188,21 +186,16 @@ class RestMiddleware:
         else:
             params = {}
 
-        req_kwargs = {}
-
-        if client is requests:
-            req_kwargs["verify"] = certificate_filepath
-            req_kwargs["timeout"] = 2
-            req_kwargs["params"] = params
-        else:
-            req_kwargs["query_string"] = params
-
         if announce_nodes:
             payload = bytes().join(bytes(VariableLengthBytestring(n)) for n in announce_nodes)
-            response = client.post("https://{}/node_metadata".format(url),
-                                   data=payload,
-                                   **req_kwargs)
+            response = self.client.post(node=node,
+                                        path="node_metadata",
+                                        params=params,
+                                        data=payload,
+                                        )
         else:
-            response = client.get("https://{}/node_metadata".format(url), **req_kwargs)
+            response = self.client.get(node=node,
+                                       path="node_metadata",
+                                       params=params)
 
         return response
diff --git a/nucypher/network/nodes.py b/nucypher/network/nodes.py
index e21128d40..50ab18347 100644
--- a/nucypher/network/nodes.py
+++ b/nucypher/network/nodes.py
@@ -673,8 +673,6 @@ class Learner:
             self.log.warn("Can't learn right now: {}".format(e.args[0]))
             return
 
-        teacher_uri = current_teacher.rest_interface
-
         if Teacher in self.__class__.__bases__:
             announce_nodes = [self]
         else:
@@ -685,10 +683,9 @@ class Learner:
             # TODO: Streamline path generation
             certificate_filepath = self.node_storage.generate_certificate_filepath(
                 checksum_address=current_teacher.checksum_public_address)
-            response = self.network_middleware.get_nodes_via_rest(url=teacher_uri,
+            response = self.network_middleware.get_nodes_via_rest(node=current_teacher,
                                                                   nodes_i_need=self._node_ids_to_learn_about_immediately,
                                                                   announce_nodes=announce_nodes,
-                                                                  certificate_filepath=certificate_filepath,
                                                                   fleet_checksum=self.known_nodes.checksum)
         except requests.exceptions.ConnectionError as e:
             unresponsive_nodes.add(current_teacher)