Merge pull request #31 from floscha/schemeless-url

Accept URLs without scheme (also includes #18 and #28)
dvf · Jan 26, 2018 · 250a01c · 250a01c
2 parents 6650862 + 7603173
commit 250a01c
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/blockchain.py b/blockchain.py
@@ -25,7 +25,14 @@ def register_node(self, address):
         """
 
         parsed_url = urlparse(address)
-        self.nodes.add(parsed_url.netloc)
+        if parsed_url.netloc:
+            self.nodes.add(parsed_url.netloc)
+        elif parsed_url.path:
+            # Accepts an URL without scheme like '192.168.0.5:5000'.
+            self.nodes.add(parsed_url.path)
+        else:
+            raise ValueError('Invalid URL')
+
 
     def valid_chain(self, chain):
         """